def update_location(header, data):
"""
Update geographical coordinates of a given user for the whole system.
:param header: Dictionary with `token` field
:param data: Dictionary with `location` field
:return: Empty dictionary in case of success or dictionary with `error` field otherwise
"""
try:
token = header['token']
except KeyError:
return _err_dict('Token is missing')
if not ldb.is_valid_token(token):
return _err_dict('Invalid token')
try:
location = data['location']
except KeyError:
return _err_dict('Required field is missing')
try:
latitude, longitude = map(float, location.split(';', maxsplit=1))
except ValueError:
raise _err_dict('Wrong location data format')
role_id = ldb.get_role_id_by_token(token)
if role_id != 7:
return _err_dict('Location of truck drivers only considered')
try:
vehicle_id = int(token) # TODO get_vehicle_id(token)
except ValueError:
return _err_dict('No vehicle ID found associated with given token')
try:
dc.try_update_location(vehicle_id, latitude, longitude)
except dc.DataClusterQueryFailure as e:
return _err_dict(str(e))
return {}
def get_support_phone_number(header):
"""
Get support service's phone number.
:param header: Dictionary with `token` field
:return: Dictionary with `number` field containing support service's phone number or `error`
"""
try:
token = header['token']
except KeyError:
return _err_dict('Token is missing')
if not ldb.is_valid_token(token):
return _err_dict('Invalid token')
return {'number': '+7 (800) 555-35-35'} # TODO actual data
def logout(header):
"""
Close the session associated with a given token.
:param header: Dictionary with `token` field
:return: Empty dictionary in case of success or dictionary with `error` field otherwise
"""
try:
token = header['token']
except KeyError:
return _err_dict('Token is missing')
if not ldb.is_valid_token(token):
return _err_dict('Invalid token')
try:
ldb.remove_token(token)
except ldb.LocalDBQueryFailure as e:
return _err_dict(e)
return {}
def get_available_orders(header):
"""
Get a list of all unassigned orders.
:param header: Dictionary with `token` field
:return: List of dictionaries of all unassigned orders (reduced order info) or `error`
"""
try:
token = header['token']
except KeyError:
return _err_dict('Token is missing')
if not ldb.is_valid_token(token):
return _err_dict('Invalid token')
role_id = ldb.get_role_id_by_token(token)
try:
orders = dc.get_available_orders(role_id)
except dc.DataClusterQueryFailure as e:
return _err_dict(str(e))
return orders
def get_assigned_orders(header):
"""
Get a list of all orders assigned to the specified user.
:param header: Dictionary with `token` field
:return: List of dictionaries of all orders of a given user (reduced order info) or `error`
"""
try:
token = header['token']
except KeyError:
return _err_dict('Token is missing')
if not ldb.is_valid_token(token):
return _err_dict('Invalid token')
try:
login = ldb.get_login_by_token(token)
except ldb.LocalDBQueryFailure as e:
return _err_dict(str(e))
try:
orders = dc.get_assigned_orders(login)
except dc.DataClusterQueryFailure as e:
return _err_dict(str(e))
return orders
def get_order_details(header, data):
"""
Get all the details about the specified order.
:param header: Dictionary with `token` field
:param data: Dictionary with `order_id` field
:return: Dictionary with full info about the given order or `error`
"""
try:
token = header['token']
except KeyError:
return _err_dict('Token is missing')
if not ldb.is_valid_token(token):
return _err_dict('Invalid token')
try:
order_id = data['order_id']
except KeyError:
return _err_dict('Required field is missing')
try:
order = dc.get_order_details(order_id)
except dc.DataClusterQueryFailure as e:
return _err_dict(str(e))
return order
def cancel_order(header, data):
"""
Set the state of a given order as cancelled.
:param header: Dictionary with `token` field
:param data: Dictionary with `order_id` field
:return: Empty dictionary in case of success or dictionary with `error` field otherwise
"""
try:
token = header['token']
except KeyError:
return _err_dict('Token is missing')
if not ldb.is_valid_token(token):
return _err_dict('Invalid token')
try:
order_id = data['order_id']
except KeyError:
return _err_dict('Required field is missing')
try:
dc.try_cancel_order(order_id)
except dc.DataClusterQueryFailure as e:
return _err_dict(str(e))
return {}
def validate_customer(header, data):
"""
Make request for customer validation key creation while trying to end delivery.
:param header: Dictionary with `token` field
:param data: Dictionary with `order_id` field
:return: Empty dictionary in case of success or dictionary with `error` field otherwise
"""
try:
token = header['token']
except KeyError:
return _err_dict('Token is missing')
if not ldb.is_valid_token(token):
return _err_dict('Invalid token')
try:
order_id = data['order_id']
except KeyError:
return _err_dict('Required field is missing')
try:
dc.try_validate_customer(order_id)
except dc.DataClusterQueryFailure as e:
return _err_dict(str(e))
return {}
def deliver_order(header, data):
"""
Set the state of a given order as delivered using given validation key.
:param header: Dictionary with `token` field
:param data: Dictionary with `order_id` and `key` fields
:return: Empty dictionary in case of success or dictionary with `error` field otherwise
"""
try:
token = header['token']
except KeyError:
return _err_dict('Token is missing')
if not ldb.is_valid_token(token):
return _err_dict('Invalid token')
try:
order_id = data['order_id']
key = data['key']
except KeyError:
return _err_dict('Required field(s) is missing')
try:
dc.try_deliver_order(order_id, key)
except dc.DataClusterQueryFailure as e:
return _err_dict(str(e))
return {}