def delete(cls, user_id): user = UserModel.find_by_id(user_id) if not user: return {'message': 'User not found'}, 404 user.delete_from_db() return {'message': 'User deleted successfully.'}, 200
def test_crud(self): with self.app_context(): user = UserModel('test_username', 'A12345678') self.assertIsNone( UserModel.find_by_username('test_username'), "Found an user with name 'test_username' before save_to_db") self.assertIsNone(UserModel.find_by_id(1), "Found an user with id '1' before save_to_db") user.save_to_db() self.assertIsNotNone( UserModel.find_by_username('test_username'), "Did not find an user with name 'test_username' after save_to_db" ) self.assertIsNotNone( UserModel.find_by_id(1), "Did not find an user with id '1' after save_to_db")
def get(cls, _id: int): user = UserModel.find_by_id(_id) if user: return { "username": user.username, "email": user.email, "balance": user.balance }, 200 return { "message": response_quote("user_id_not_found").format(_id) }, 404
def delete(cls, _id: int): current_user = get_jwt_identity() user = UserModel.find_by_id(_id) if user: if user.session_key != current_user: return {"message": response_quote("code_401")}, 401 user.delete_from_db() return {"message": response_quote("user_deleted")}, 201 return { "message": response_quote("user_id_not_found").format(_id) }, 404
def delete(cls, _id: int): current_user = get_jwt_identity() user = UserModel.find_by_id(_id) if user: if user.sha_private != current_user: return {"message": response_quote("code_401")}, 401 user.delete_from_db() # TODO: удалить все jwt токены. return {"message": response_quote("user_deleted")}, 201 return { "message": response_quote("user_id_not_found").format(_id) }, 404
def get(cls, user_id) -> tuple: """ Get the user by an id. :param user_id: Integer of userid. :return: User or .json message """ user = UserModel.find_by_id(id_=user_id) if not user: return {'message': 'User not found!'}, 404 return user.json(), 200
def delete(cls, user_id) -> tuple: """ Delete a user from the db. Requires a fresh JWT. :param user_id: Int of user id. :return: .json message + status code. """ user = UserModel.find_by_id(id_=user_id) if not user: return {'message': 'User not found!'}, 404 user.delete_from_db() return {'message': 'User deleted.'}, 200
def get(cls, _id: int): user = UserModel.find_by_id(_id) if user: return { "username": user.username, "name": user.name, "surname": user.surname, "locality": user.locality, "profile_pic": user.profile_pic, "second_fa_enabled": user.second_fa_enabled, "balance": user.balance }, 200 return { "message": response_quote("user_id_not_found").format(_id) }, 404
def put(cls, _id: int): data = request.get_json() current_user = get_jwt_identity() user = UserModel.find_by_id(_id) if user: if user.sha_private != current_user: return {"message": response_quote("code_401")}, 401 user.username = data["username"] user.name = data["name"] user.surname = data["surname"] user.locality = data["locality"] user.balance = data["balance"] user.profile_pic = data["profile_pic"] user.second_fa_enabled = data["second_fa_enabled"] user.save_to_db() return {"message": response_quote("user_data_changed")}, 201 return { "message": response_quote("user_id_not_found").format(_id) }, 404
def delete(self, id): user = UserModel.find_by_id(id) if user: user.delete() return '', 204
def get(self, id): user = UserModel.find_by_id(id) if not user: ns.abort(404) return user.as_dict()
def identity(payload): user_id = payload['identity'] return UserModel.find_by_id(user_id)
def delete(cls, user_id: int): user = UserModel.find_by_id(user_id) if not user: return {'message': 'User Not Found'}, 404 user.delete_from_db() return {'message': 'User deleted.'}, 200
def get(cls, user_id): user = UserModel.find_by_id(user_id) if not user: return {'message': 'User Not Found'}, 404 return user.json(), 200
def identity(payload): _LOGGER.debug("Identity..") user_id = payload['identity'] return UserModel.find_by_id(user_id)
def get(self, user_id): user = UserModel.find_by_id(user_id) if not user: return {'message': 'User not found'}, 404 return user.json()