def get(self, id):
logger.getLogger().debug("GET at /users/" + str(id))
if not "UserToken" in request.headers:
return ResponseMaker.response_error(constants.PARAMERR,
"Bad request - missing token")
token = request.headers['UserToken']
(valid, decoded) = TokenGenerator.validateToken(token)
print("GET at /user/id")
candidates = [user for user in self.users.find() if user['_id'] == id]
if len(candidates) == 0:
#If not available in local data-base, ask Shared-Server for user info.
(status, response) = ServerRequest.getUser(id)
if (status != constants.SUCCESS):
return ResponseMaker.response_error(status,
response["message"])
candidates = [response]
user = candidates[0]
self.users.insert_one(user)
if len(candidates) == 0:
logger.getLogger().error(
"Attempted to retrieve user with non-existent id.")
return ResponseMaker.response_error(constants.NOT_FOUND,
"User id not found:" + str(e))
return ResponseMaker.response_object(constants.SUCCESS, ['user'],
[User.UserJSON(candidates[0])])
def delete(self, id):
print("DELETE at /users/id")
logger.getLogger().debug("DELETE at /users/" + str(id))
if not "UserToken" in request.headers:
return ResponseMaker.response_error(constants.PARAMERR,
"Bad request - missing token")
token = request.headers['UserToken']
(valid, decoded) = TokenGenerator.validateToken(token)
print(decoded)
if not valid or (decoded['_id'] != id):
return ResponseMaker.response_error(constants.FORBIDDEN,
"Forbidden")
#Ask Shared-Server to delete this user
delete_success, status_code = ServerRequest.deleteUser(id)
if delete_success:
#Delete in local data-base
candidates = [
User.UserJSON(user) for user in self.users.find()
if user['_id'] == id
]
self.users.delete_many({"_id": id})
logger.getLogger().info("Successfully deleted user.")
else:
logger.getLogger().error(
"Attempted to delete user with non-existent id.")
return ResponseMaker.response_error(status_code, "Delete error")
return ResponseMaker.response_object(constants.DELETE_SUCCESS,
['user'], candidates)
def post(self):
print(request.json)
logger.getLogger().debug("POST at /users/logout")
logger.getLogger().debug(request.json)
if not "UserToken" in request.headers:
return ResponseMaker.response_error(constants.PARAMERR,
"Bad request - missing token")
token = request.headers['UserToken']
result = TokenGenerator.validateToken(token)
# (mongodb) Return all logged in users.
users_online = MongoController.getCollection("online")
user = [
User.UserJSON(user) for user in users_online.find()
if (user['username'] == result[1]['username'])
]
if (len(user) == 0):
return ResponseMaker.response_error(constants.NOT_FOUND,
"User not found")
#users_online.delete_many(user[0]);
users_online.update({"_id": user[0]["_id"]},
{"$set": {
"online": False
}})
return ResponseMaker.response_object(constants.SUCCESS, ['user'],
[user[0]])
def post(self):
logger.getLogger().debug("POST at /users/login")
logger.getLogger().debug(request.json)
try:
# (shared-server) First ask shared server for credentials validation
(valid, response) = ServerRequest.validateUser(request.json)
if not valid:
logger.getLogger().debug(
'Error 418: I\' m a teapot and your credentials are not valid!'
)
return ResponseMaker.response_error(response.status_code,
"Shared server error")
logger.getLogger().debug(
"Credentials are valid, server responsed with user")
user_js = User.UserJSON(response)
# (token-generation) Generate a new UserToken for that user
token = TokenGenerator.generateToken(response)
users_online = MongoController.getCollection("online")
# (mongodb) If credentials are valid, and user is not here, add it
for user in users_online.find():
if user_js["_id"] == user["_id"]:
# Found it! Checking refs!
logger.getLogger().debug(
"User trying to log in already found in the local db")
if user_js["_ref"] != user["_ref"]:
logger.getLogger().debug(
"Login of an user already in bdd. _ref is old")
user_js = User.UserUpdateDedicatedFields(user_js, user)
else:
user_js = user
break
user_js["online"] = True
users_online.update({"_id": user_js["_id"]}, user_js, upsert=True)
return ResponseMaker.response_object(constants.SUCCESS,
['user', 'token'],
[user_js, token])
except Exception as e:
logger.getLogger().exception(str(e))
print(str(e))
return ResponseMaker.response_error(500, "Internal Error")
def put(self, id):
print(id)
print("PUT at /user/" + str(id) + "/rating")
logger.getLogger().debug("PUT at /users/" + str(id) + "/rating")
if not "UserToken" in request.headers:
return ResponseMaker.response_error(constants.PARAMERR,
"Bad request - missing token")
token = request.headers['UserToken']
(valid, decoded) = TokenGenerator.validateToken(token)
if not valid or (decoded['_id'] != id):
return ResponseMaker.response_error(constants.FORBIDDEN,
"Forbidden")
try:
new_score = request.json["rate"]
rating_count = [
user["rating"]["rateCount"] for user in self.users.find()
if user['_id'] == id and user['type'] == "driver"
][0]
current_rate = [
user["rating"]["rate"] for user in self.users.find()
if user['_id'] == id and user['type'] == "driver"
][0]
if len(rating_count) == 0 and len(current_rate) == 0:
return ResponseMaker.response_error(constants.NOT_FOUND,
"User id not found.")
#Recalculate rating
rating_count += 1
current_rate = current_rate + new_score
#Update in local data-base
self.users.update({'_id': id}, {
"$set": {
"rating.rateCount": rating_count,
"rating.rate": current_rate
}
},
upsert=True)
logger.getLogger().info("Successfully updated driver rating")
driver = [
user for user in self.users.find()
if user['_id'] == id and user['type'] == "driver"
][0]
return ResponseMaker.response_object(constants.SUCCESS, ['rating'],
[User.RatingUserJSON(driver)])
except ValueError as e:
logger.getLogger().error(str(e))
return ResponseMaker.response_error(constants.UPDATE_CONFLICT,
"User update failed:" + str(e))
except requests.exceptions.Timeout as e:
logger.getLogger().error(str(e))
return ResponseMaker.response_error(constants.REQ_TIMEOUT,
"User update failed:" + str(e))
except Exception as e:
logger.getLogger().error(str(e))
return ResponseMaker.response_error(constants.ERROR, str(e))
def __init__(self, collection):
self.docs = []
self.source = []
if collection == "online":
for u in default_db:
self.source.append(User.UserJSON(u))
elif collection == "active_trips":
for t in trips_db:
self.source.append(t)
self.reset()
def post(self):
print(request.json)
logger.getLogger().debug("POST at /users")
logger.getLogger().debug(request.json)
# (shared-server) Send new user data to shared server.
(status, response) = ServerRequest.createUser(request.json)
if (status != constants.CREATE_SUCCESS):
return ResponseMaker.response_error(status, response['message'])
user_js = User.UserJSON(response)
users_online = MongoController.getCollection("online")
users_online.insert_one(user_js)
return ResponseMaker.response_object(status, ['user'], [user_js])
def get(self, id):
logger.getLogger().debug("GET at /users/" + str(id) + "/rating")
if not "UserToken" in request.headers:
return ResponseMaker.response_error(constants.PARAMERR,
"Bad request - missing token")
token = request.headers['UserToken']
(valid, decoded) = TokenGenerator.validateToken(token)
if not valid or (decoded['_id'] != id):
return ResponseMaker.response_error(constants.FORBIDDEN,
"Forbidden")
try:
print("GET at /users/" + str(id) + "/rating")
candidates = [
user for user in self.users.find() if user['_id'] == id
]
if len(candidates) == 0:
#If not available in local data-base, ask Shared-Server for user info.
(status, response) = ServerRequest.getUser(id)
if (status != constants.SUCCESS):
return ResponseMaker.response_error(
status, response["message"])
candidates = [response]
if len(candidates) == 0:
logger.getLogger().error(
"Attempted to retrieve user with non-existent id.")
return ResponseMaker.response_error(
constants.NOT_FOUND, "User id not found:" + str(e))
if candidates[0]["type"] != "driver":
return ResponseMaker.response_error(
constants.PARAMERR,
"The requested user is not a driver. Only drivers receive ratings."
)
return ResponseMaker.response_object(
constants.SUCCESS, ['rating'],
[User.RatingUserJSON(candidates[0])])
except Exception as e:
logger.getLogger().error(str(e))
return ResponseMaker.response_error(constants.ERROR, str(e))
def get(self):
logger.getLogger().debug("GET at /users")
# (validate-token) Validate user token
if not "UserToken" in request.headers:
return ResponseMaker.response_error(constants.PARAMERR,
"Bad request - missing token")
token = request.headers['UserToken']
(valid, response) = TokenGenerator.validateToken(token)
if not valid:
return ResponseMaker.response_error(constants.FORBIDDEN,
"Forbidden")
# (mongodb) Return all logged in users.
users_online = MongoController.getCollection("online")
aux = [User.UserJSON(user) for user in users_online.find()]
user = list(users_online.find({"_id": response["_id"]}))
if len(user) == 0:
return ResponseMaker.response_error(
constants.PARAMERR, "Bad request - you may not be loged in")
user = user[0]
## Aditional params (only supporting "sort")
params = request.args
if "filter" in params:
aux = [u for u in aux if u["type"] == params["filter"]]
if "sort" in params and params["sort"] == "near":
aux.sort(key=lambda x: Distances.computeDistance(
user["coord"], x["coord"]))
if "limit" in params:
limit = int(params["limit"])
if limit > 0:
aux = aux[:(limit)]
return ResponseMaker.response_object(constants.SUCCESS, ['users'],
[aux])
def put(self, id):
print(id)
print("PUT at /user/id")
logger.getLogger().debug("PUT at /users/" + str(id))
if not "UserToken" in request.headers:
return ResponseMaker.response_error(constants.PARAMERR,
"Bad request - missing token")
token = request.headers['UserToken']
(valid, decoded) = TokenGenerator.validateToken(token)
if not valid or (decoded['_id'] != id):
return ResponseMaker.response_error(constants.FORBIDDEN,
"Forbidden")
try:
#Ask Shared-Server to update this user
success, updated_user = ServerRequest.updateUser(request.json)
if success:
#Update in local data-base
self.users.update({'_id': updated_user['id']},
User.UserJSON(updated_user),
upsert=True)
logger.getLogger().info("Successfully updated user")
return ResponseMaker.response_error(
constants.SUCCESS, "User updated successfully!")
return ResponseMaker.response_error(constants.NOT_FOUND,
"User not found!")
except ValueError as e:
logger.getLogger().error(str(e))
return ResponseMaker.response_error(constants.UPDATE_CONFLICT,
"User update failed:" + str(e))
except requests.exceptions.Timeout as e:
logger.getLogger().error(str(e))
return ResponseMaker.response_error(constants.REQ_TIMEOUT,
"User update failed:" + str(e))
except Exception as e:
logger.getLogger().error(str(e))
return ResponseMaker.response_error(constants.FORBIDDEN,
"Forbidden: " + str(e))