add login logout and verify powered by couchdb

Author: leesper

couchdb/client.py

@@ -231,11 +231,53 @@ def register(self, name, password):
         """Register regular user in authentication database.
         :param name: name of regular user, normally user id
         :param password: password of regular user
+        :return: (id, rev) tuple of the registered user
+        :rtype: `tuple`
         """
         user_db = self['_users']
         doc = {'_id': 'org.couchdb.user:'+name, 'name': name, 'password': password, 'roles': [], 'type': 'user'}
         return user_db.save(doc)
 
+    def login(self, name, password):
+        """Login regular user in couch db
+        :param name: name of regular user, normally user id
+        :param password: password of regular user
+        :return: (status, token) tuple of the login user
+        :rtype: `tuple`
+        """
+        data = {'name': name, 'password': password}
+        status, headers, data = self.resource.post_json('_session', data)
+        if status != 200:
+            return status, None
+        cookie = headers.headers[0].split(';')[0]
+        pos = cookie.find('=')
+        token = cookie[pos+1:]
+        return status, token
+
+    def logout(self, token):
+        """Logout regular user in couch db
+        :param token: token of login user
+        :return: True if successfully logout
+        :rtype: bool
+        """
+        header = {'Accept': 'application/json', 'Cookie': 'AuthSession='+token}
+        status, headers, data = self.resource.delete_json('_session', headers=header)
+        if status != 200:
+            return False
+        return True
+
+    def verify(self, token):
+        """Verify whether token is ok
+        :param token: token of login user
+        :return: True if token is ok
+        :rtype: bool
+        """
+        header = {'Accept': 'application/json', 'Cookie': 'AuthSession='+token}
+        status, headers, data = self.resource.get_json('_session', headers=header)
+        if status != 200:
+            return False
+        return True
+
 
 class Database(object):
     """Representation of a database on a CouchDB server.