def test_crud(self): with self.app_context(): user = UserModel("ipo", "kk") self.assertIsNone(UserModel.find_by_username("ipo")) self.assertIsNone(UserModel.find_by_id(1)) user.save_to_db() self.assertIsNotNone(UserModel.find_by_username("ipo")) self.assertIsNotNone(UserModel.find_by_id(1))
def delete(self, user_id): claims = get_jwt_claims() if not claims['is_admin']: return {'message': 'Admin Privileges required for this operation'} user = UserModel.find_by_id(user_id) if (not user): return {'message': 'User not found'}, 404 elif user.id == 1: return {'message': 'Can not delete admin'}, 500 user.delete_from_db() return {'message': 'User {} deleted'.format(user.username)}, 200
def post(self, user_id): data = self.parser.parse_args() if UserModel.find_by_id(user_id): favorite_games = FavoriteGamesModel.find_favorite_games_by_user_id(user_id) for favorite_game in favorite_games: if favorite_game.game_url_id == data['game_url_id']: return {"message": "This game ID is already in your list of favorites"} new_favorite_game = FavoriteGamesModel(user_id, **data) new_favorite_game.save_favorite_game_id() return {"message": f"Favorite game was save to user_id: {user_id}"} return{"message": "User do not exist"}
def delete(self, user_id): game_url_id = request.args.get('game_url_id') if not game_url_id: return {"message": "Please provide a (game_url_id = ? ) as query parameter to remove it from the list"} if UserModel.find_by_id(user_id): favorite_games = FavoriteGamesModel.find_favorite_games_by_user_id(user_id) for favorite_game in favorite_games: if favorite_game.game_url_id == int(game_url_id): favorite_game.remove_favorite_game() return {"message":"Game was removed from the list of favorites"},202 return {"message":"Game is not in the list of favorites"} return{"message": "User do not exist"}
def test_register_user(self): with self.app() as client: with self.app_context(): response = client.post("/register", data={ "username": "******", "password": "******" }) self.assertEqual(response.status_code, 201) self.assertIsNotNone(UserModel.find_by_username("ipo")) self.assertIsNotNone(UserModel.find_by_id(1)) self.assertDictEqual( {"message": "the account register successfully"}, json.loads(response.data))
def put(self): parser = reqparse.RequestParser() parser.add_argument("username", type=str, required=True, help="Username required to update the account") parser.add_argument("password", type=str, required=True, help="Password required to update the account") data = parser.parse_args() user_id = get_jwt_identity() user = UserModel.find_by_id(user_id) if safe_str_cmp(user.password, data['password']): return {"message": "You are using the same password please change it"} user.username = data['username'] user.password = data['password'] user.save_to_db() return {"message": "User name and password have been changed"}
def get(cls, user_id: int): user = UserModel.find_by_id(user_id) if not user: return {'message': 'User Not Found'}, 404 return user.json(), 200
def get(self, user_id): user = UserModel.find_by_id(user_id) if (not user): return {'message': 'User not found'}, 404 return user.json()
def delete(self): user_id = get_jwt_identity() user = UserModel.find_by_id(user_id) user.delete_from_db() return {"message": "User was deleted"}
def get(self): user_id = get_jwt_identity() user = UserModel.find_by_id(user_id) return user.json()
def get(cls, user_id): user = UserModel.find_by_id(user_id) if user: return user.json() return {"message": "User not found"}, 404
def security(payload): """ Parses the payload data for the user Identity """ user_id = payload['identity'] return UserModel.find_by_id(user_id)
def delete(cls, user_id): user = UserModel.find_by_id(user_id) if user: user.delete_from_db() return {"message": "User deleted successfully"}, 200 return {"message": "User not found"}, 404
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 identity(payload): userid = payload['identity'] return UserModel.find_by_id(userid)
def identity(payload): id_user = payload['identity'] return UserModel.find_by_id(id_user)
def identity(payload): """if the user is authenticated, Flask-JWT is verified their authorization header is correct""" user_id = payload["identity"] return UserModel.find_by_id(user_id)