Use Flask-Login for login handling

1 files changed, 13 insertions, 1 deletions

diff --git a/accounts/models.py b/accounts/models.py
index 9e0c45e..f967180 100644
--- a/accounts/models.py
+++ b/accounts/models.py
@@ -1,6 +1,11 @@
 # -*- coding: utf-8 -*-
+from __future__ import absolute_import
+from flask.ext.login import UserMixin
 
-class Account(object):
+from accounts.utils.login import create_userid
+
+
+class Account(UserMixin):
     """
     An Account represents a complex ldap tree entry for spline users.
     For each service a spline user can have a different password.
@@ -62,6 +67,13 @@ class Account(object):
         raise AttributeError("'%s' object has no attribute '%s'" %
                              (self.__class__.__name__, name))
 
+    def get_id(self):
+        """
+        This is for flask-login. The returned string is saved inside
+        the cookie and used to identify the user.
+        """
+        return create_userid(self.uid, self.password)
+
 
 class Service(object):
     def __init__(self, id, name, url):