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
