class UserController(MasterController): _user = None def __init__(self, args=None): self._user = User() super(UserController, self).__init__() def login(self, args): password = self._user.return_pw(args['user_name']) if password['password'] == args['password']: self._user.load(args['user_name']) self._session.setCookie(self._user) print self._session._cookie self._user.attach(PostController) self._user.notify() def logout(self, args): self._session.clearCookie() print self._session._cookie self._user.attach(PostController) self._user.notify()
def auth(connection, email, password=None, token=None): user = User.load(email) if not user: params = {'result': 'Fail', 'errormsg': 'No such user'} connection.response({'cmd': 'auth', 'params': params}) return auth_succesful = False if token and token == user.token: auth_succesful = True if password and hashlib.md5(password).hexdigest() == user.password: auth_succesful = True if auth_succesful: user.update_token() params = {'result': 'Ok', 'token': user.token} connection.response({'cmd': 'auth', 'params': params}) connection.authenticate(user) else: params = {'result': 'Fail', 'errormsg': 'Authentication failed'} connection.response({'cmd': 'auth', 'params': params})
def auth(connection, email, password=None, token=None): user = User.load(email) if not user: params = {"result": "Fail", "errormsg": "No such user"} connection.response({"cmd": "auth", "params": params}) return auth_succesful = False if token and token == user.token: auth_succesful = True if password and hashlib.md5(password).hexdigest() == user.password: auth_succesful = True if auth_succesful: user.update_token() params = {"result": "Ok", "token": user.token} connection.response({"cmd": "auth", "params": params}) connection.authenticate(user) else: params = {"result": "Fail", "errormsg": "Authentication failed"} connection.response({"cmd": "auth", "params": params})
def test_search(self): actions.register(connection, email='*****@*****.**', password='******') actions.auth(connection, "*****@*****.**", password='******', token=None) dumb_socket2 = DumbSocket() connection2 = Connection(dumb_socket2) actions.register(connection2, email='*****@*****.**', password='******') actions.auth(connection2, "*****@*****.**", password='******', token=None) dumb_socket3 = DumbSocket() connection3 = Connection(dumb_socket3) actions.register(connection3, email='*****@*****.**', password='******') actions.auth(connection2, "*****@*****.**", password='******', token=None) actions.add_friend(connection, email="*****@*****.**") actions.add_friend(connection, email="*****@*****.**") user = User.load('*****@*****.**') print user.get_friends() actions.search(connection, need="kitten")
def test_load_user(self): user = User.load("*****@*****.**") self.assertEqual(user.email, "*****@*****.**") print user.id