using urllib

beautifier/__init__.py

@@ -1,5 +1,6 @@
 import re
 import os
+from urllib.parse import urlsplit, urlunsplit
 
 __version__ = "0.5.5"
 
@@ -54,14 +55,19 @@ def username(self):
         else:
             return {'msg': 'feature is currently available only with linkedin urls'}
 
+
+    @property
+    def scheme(self):
+        return urlsplit(self.cleanup).scheme
+
+    @property 
+    def path(self):
+        return urlsplit(self.cleanup).path
+
+
     @property
     def domain(self):
         """
         Return domain from the url
         """
-        remove_pac = self.cleanup.replace(
-            "https://", "").replace("http://", "").replace("www.", "")
-        try:
-            return remove_pac.split('/')[0]
-        except:
-            return None
+        return urlsplit(self.cleanup).netloc

beautifier/tests/tests.py

@@ -56,6 +56,17 @@ def test_url_domain(self):
         '''
         self.assertEqual(self.url.domain, 'in.linkedin.com')
 
+    def test_url_scheme(self):
+        '''
+        Checks the url scheme is returned
+        '''
+        self.assertEqual(self.url.scheme, 'https')
+
+    def test_url_path(self):
+        '''
+        Checks the url path is returned
+        '''
+        self.assertEqual(self.url.path, '/in/sachinphilip')
 
 if __name__ == '__main__':
     unittest.main()