def unregister():
email = from_params_or_json(request, 'email')
email = email if is_email(email) else ''
password = from_params_or_json(request, 'password')
try:
User(email=email, password=password).unregister()
except OAuth2Error:
pass
response.status = 204
def register_or_login():
""" Registers or logs in the user, always returning access_token on success """
email = from_params_or_json(request, 'email')
email = email if is_email(email) else ''
password = from_params_or_json(request, 'password')
grant_type = from_params_or_json(request, 'grant_type') or 'password'
# meta = from_params_or_json(request, 'meta')
register_resp = register(email, password)
if register_resp.get('error_description') == "Email already registered": # the lookup first approach fails
response.status = 200
return login(email, password, grant_type)
elif 'access_token' in register_resp:
return register_resp
return error(response, 'server_error', "Registration failed")
def login(email=None, password=None, grant_type=None):
email = from_params_or_json(request, 'email') or email
email = email if is_email(email) else ''
password = from_params_or_json(request, 'password') or password
grant_type = from_params_or_json(request, 'grant_type') or grant_type
try:
if email and password and grant_type == 'password':
login_resp = User(email=email, password=password).login()
if login_resp and 'access_token' in login_resp:
return login_resp
return error(response, 'server_error', "Login failed")
else:
return error(response, 'invalid_request', "`email`, `password` and `grant_type='password'` required")
except OAuth2Error as e:
message = dict(e.message)
response.status = message.pop('status_code')
return message
def register(email=None, password=None):
email = from_params_or_json(request, 'email') or email
email = email if is_email(email) else ''
password = from_params_or_json(request, 'password') or password
if not email or not password:
return error(response, 'invalid_request', "`email` and `password` required")
try:
registered = User(email=email, password=password).register()
if registered:
tok = AccessToken(user=registered).generate()
return {'access_token': tok.token,
'expires_in': randint(0, 200)}
except NotUniqueError:
return error(response, 'access_denied', "Email already registered")
except ValidationError as e:
return error(response, 'invalid_request', e.message)
except OAuth2Error as e:
message = dict(e.message)
response.status = message.pop('status_code')
return message
return error(response, 'server_error', "Registration failed")
def test_is_email_failure(self):
self.assertFalse(is_email('foo@bar'))
self.assertFalse(is_email('f@b'))
self.assertFalse(is_email(''))
def test_is_email_success(self):
self.assertTrue(is_email('*****@*****.**'))
self.assertTrue(is_email('*****@*****.**'))
self.assertTrue(is_email('*****@*****.**'))