miniuri.py: a fully functional uri class written completely in 99 lines python (including comments)!

miniuri.py

@@ -0,0 +1,99 @@
+class Uri( object ):
+    """
+    >>> from miniuri import Uri
+    >>> u = Uri( "http://www.foxhop.net/samsung/HL-T5087SA/red-LED-failure" )
+    >>> u.uri = "https://fox:pass@www.foxhop.net:8080/path/filename.jpg?p=2#comment"
+
+    foo://username:password@example.com:8042/over/there/index.php?pet=cat&name=bam#nose
+    \_/   \_______________/ \_________/ \__/            \___/ \_/ \______________/ \__/
+     |           |               |       |                |    |         |           |
+     |       userinfo        hostname  port               |    |       query   fragment
+     |    \________________________________/\_____________|____|/
+    scheme                |                        |      |    |
+                      authority                   path    | extension
+                                                          |    
+                                                      filename                          
+    """
+    scheme = username = password = hostname = port = None
+    path = filename = query = fragment = None 
+
+    def __init__( self, uri = None ):
+        if uri: self.uri = uri # invoke uri.setter
+
+    @property
+    def uri( self ):
+        """build and return uri from attributes"""
+        scheme = path = filename = query = fragment = ''
+        if self.path: path = '/'.join( self.path.split('/')[:-1] )
+        if self.scheme: scheme = self.scheme + '://'
+        if self.filename: filename = '/' + self.filename
+        if self.query: query = '?' + self.query
+        if self.fragment: fragment = '#' + self.fragment
+        return ''.join( [scheme, self.authority, path, filename, query, fragment] )
+
+    @uri.setter
+    def uri( self, uri ):
+        """parse and set all uri attributes"""
+        if '://' in uri: # attempt to parse scheme
+            self.scheme, uri = uri.split( '://' )
+
+        # invoke authority setter
+        self.authority = uri.split( '/' )[0]
+
+        uri = uri[ len( self.authority ): ] 
+
+        if '#' in uri: # set fragment
+            uri, self.fragment = uri.split( '#' )  
+        
+        if '?' in uri: # set path and possibly query
+            self.path, self.query = uri.split( '?' )
+        else: self.path = uri
+        
+        self.filename = self.path.split( '/' )[-1]
+        
+    @property
+    def authority( self ):
+        """return a authority string from attributes"""
+        a = ''
+        if self.username:
+            a += self.username
+            if self.password: a += ':' + self.password
+            a += '@'
+        a += self.hostname
+        if self.port: a += ':' + self.port
+        return a
+
+    @authority.setter
+    def authority( self, a ):
+        """set all the attribute that makeup a authority"""
+        self.username = self.password = self.port = None
+        self.hostname = a
+        if '@' in a:
+            self.userinfo, self.hostname = a.split( '@' ) # userinfo setter
+        if ':' in self.hostname:
+            self.hostname, self.port = self.hostname.split( ':' )
+
+    @property
+    def userinfo( self ):
+        """return username:password"""
+        return self.username + ':' + self.password
+
+    @userinfo.setter
+    def userinfo( self, info ):
+        """set username and password"""
+        self.username, self.password = info, None 
+        if ':' in self.username: 
+            self.username, self.password = self.username.split( ':' )
+
+    @property
+    def extension( self ):
+        """return extension"""
+        if '.' in self.filename: return self.filename.split('.')[1]
+        else: return None 
+
+    @property
+    def parts( self ):
+        """remove scheme then return a list of uri parts splitting on / """
+        p = self.uri
+        if self.scheme: p = p[ len(self.scheme) + 3: ]
+        return p.split( '/' )