def test_decode_token(test_app, test_database, add_user): user = add_user("justatest", "*****@*****.**", "test") token = user.encode_token(user.id, "access") assert isinstance(token, bytes) assert User.decode_token(token) == user.id token = user.encode_token(user.id, "refresh") assert isinstance(token, bytes) assert User.decode_token(token) == user.id
def post(self): post_data = request.get_json() refresh_token = post_data.get("refresh_token") response_object = {} try: resp = User.decode_token(refresh_token) user = get_user_by_id(resp) if not user: auth_namespace.abort(401, "Invalid token") access_token = user.encode_token(user.id, "access") refresh_token = user.encode_token(user.id, "refresh") response_object = { "access_token": access_token.decode(), "refresh_token": refresh_token.decode(), } return response_object, 200 except jwt.ExpiredSignatureError: auth_namespace.abort(401, "Signature expired. Please log in again.") return "Signature expired. Please log in again." except jwt.InvalidTokenError: auth_namespace.abort(401, "Invalid token. Please log in again.")
def post(self): """Returns all deliveries of specified user.""" post_data = request.get_json() token = post_data.get("token") user_id = User.decode_token(token) user = get_user_by_id(user_id) if not user: deliveries_namespace.abort(404, f"User {user_id} does not exist") return user.deliveries, 200
def get(self): auth_header = request.headers.get("Authorization") if auth_header: try: access_token = auth_header.split(" ")[1] resp = User.decode_token(access_token) user = get_user_by_id(resp) if not user: auth_namespace.abort(401, "Invalid token") return user, 200 except jwt.ExpiredSignatureError: auth_namespace.abort(401, "Signature expired. Please log in again.") return "Signature expired. Please log in again." except jwt.InvalidTokenError: auth_namespace.abort(401, "Invalid token. Please log in again.") else: auth_namespace.abort(403, "Token required")
def post(self): """Creates a new delivery.""" post_data = request.get_json() token = post_data.get("token") user_id = User.decode_token(token) user = get_user_by_id(user_id) if not user: deliveries_namespace.abort(404, f"User {user_id} does not exist") end_date_time = post_data.get("end_date_time") end_date_time = datetime.strptime(post_data.get("end_date_time"), '%Y-%m-%dT%H:%M:%S.%fZ') lat = post_data.get("lat") lng = post_data.get("lng") response_object = {} delivery = add_delivery(user_id=user_id, end_date_time=end_date_time, lat=lat, lng=lng) user.deliveries.append(delivery) print(user.deliveries) response_object["message"] = f"{user_id}" return response_object, 201