def handle_send_group(self, group_id):
"""
Handles a send_group operation. This method uses the Flask request
to get the session_token that is passed in the Authentication header,
as well as the message to be sent. Assumes that the user is logged in.
:param group_id: The group to which the message will be sent.
:return: On success, JSON containing the group_id and the message (and code 200)
On failure, JSON containing the error code (as defined in rest_errors.py).
"""
session_token = request.authorization.password
try:
message = request.json['data']['message']
except:
return rest_errors.bad_request()
try:
self.send_message_to_group(session_token, message, group_id)
except GroupDoesNotExist:
return rest_errors.not_found()
except:
return rest_errors.internal_server_error()
return json.dumps({'data': {'group_id': group_id, 'message': message}}), 201
def handle_add_user_to_group(self, group_id):
"""
Handles an add_user_to_group operation. This uses the Flask request
to get the group_id that is passed in the HTTP request. Assumes
the user is logged in.
:return: On success, JSON containing the group_id and
the username (and code 200). On failure, JSON containing the error code
(as defined in rest_errors.py).
"""
try:
username = request.json['data']['username']
except:
return rest_errors.bad_request()
try:
self.add_user_to_group(username, group_id)
except GroupDoesNotExist:
return rest_errors.not_found()
except UsernameDoesNotExist:
return rest_errors.not_found()
except:
return rest_errors.internal_server_error()
return json.dumps({'data': {'group_id': group_id, 'username': username}}), 201
def handle_delete_user(self, user_id):
"""
Handles a delete account operation. This method uses uses the Flask request
to get the session_token passed in the Authentication headers. This checks
that the session_token matches the passed in :param user_id (see below).
Assumes that the user is logged in.
:param user_id: The user_id that must be deleted
:return: On success, JSON containing the username (and code 200).
On failure, JSON containing the error code (as defined in rest_errors.py).
"""
session_token = request.authorization.password
try:
session_username = self.username_for_session_token(session_token)
if (session_username != user_id):
return rest_errors.forbidden()
self.logout(user_id)
self.delete_account(user_id)
except UserKeyError:
return rest_errors.unauthorized()
except:
return rest_errors.internal_server_error()
return json.dumps({'data': {'user': {'username': user_id}}}), 200
def handle_logout(self):
"""
Handles a logout operation. There are no explicit parameters, but
this uses the Flask request to get the session_token.
Assumes that the user is logged in.
:return: On success, JSON containing the previous session_token (and code 200).
On failure, JSON containing the error code (as defined in rest_errors.py).
"""
session_token = request.authorization.password
try:
username = self.username_for_session_token(session_token)
self.logout(username)
except UserKeyError:
return rest_errors.internal_server_error()
except:
return rest_errors.internal_server_error()
return json.dumps({'data': {'user': {'session_token': session_token}}}), 200
def handle_get_groups(self):
"""
Handles a get_groups operation. There are no explicit parameters, but
this method uses the Flask request to get the wildcard passed in
as a query in the HTTP request. Assumes that the user is logged in.
:return: On success, JSON containing the list of matching groups (and code 200)
On failure, JSON containing the error code (as defined in rest_errors.py).
"""
wildcard = request.args.get('wildcard')
if wildcard is None:
wildcard = '.*'
try:
groups = [group['name'] for group in self.get_groups(wildcard)]
except:
rest_errors.internal_server_error()
return json.dumps({'data': {'groups': groups}})
def handle_fetch(self, user_id):
"""
Handles a fetch operation. Assumes that the user is logged in.
:param user_id: The user_id that required this fetch.
:return: On success, JSON containing a list of messages (and code 200)
On failure, JSON containing the error code (as defined in rest_errors.py).
"""
try:
messages = self.get_user_queued_messages(user_id)
self.clear_user_message_queue(user_id)
except UserKeyError:
return rest_errors.not_found()
except:
return rest_errors.internal_server_error()
return json.dumps({'data': {'messages': messages}})
def handle_create_group(self):
"""
Handles a create_group operation. There are no explicit parameters,
but this uses the Flask request to get the group_name that is
passed in the HTTP request.
:return: On success, JSON containing the group_name (and code 200).
On failure, JSON containing the error code (as defined in rest_errors.py).
"""
try:
group_name = request.json['data']['group_name']
except:
return rest_errors.bad_request()
try:
self.create_group(group_name)
except GroupExists:
return rest_errors.error(200, 'Group Name Taken: The group name chosen already exists')
except:
return rest_errors.internal_server_error()
return json.dumps({'data': {'group_id': group_name}}), 200
def handle_create_user(self):
"""
Handles a register operation. There are no explicit parameters, but
this uses the Flask request to get the username and password that are
passed in the HTTP request.
:return: On success, JSON containing the username (and code 200).
On failure, JSON containing the error code (as defined in rest_errors.py).
"""
try:
username = request.json['username']
password = request.json['password']
except:
return rest_errors.bad_request()
try:
self.create_account(username, password)
except UsernameExists:
return rest_errors.error(200, 'Username Taken: The username chosen already exists')
except:
return rest_errors.internal_server_error()
return json.dumps({'data': {'user': {'username': username}}}), 201