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()
