diff --git a/server/app/apis/auth.py b/server/app/apis/auth.py index 49b5a3e573a4df0716f617ed4956b9db1c451fa1..e6a5345bd0346eb9959a94bb7d28f3a9e15f0097 100644 --- a/server/app/apis/auth.py +++ b/server/app/apis/auth.py @@ -11,7 +11,7 @@ from app.apis import item_response, protect_route, text_response from app.core import sockets from app.core.codes import verify_code from app.core.dto import AuthDTO -from app.database.models import Whitelist +from app.database.models import User, Whitelist from flask import current_app from flask_jwt_extended import create_access_token, get_jti, get_raw_jwt from flask_jwt_extended.utils import get_jti @@ -87,7 +87,7 @@ class AuthDelete(Resource): def delete(self, user_id): """ Deletes the specified user and adds their token to the blacklist. """ - item_user = dbc.get.user(user_id) + item_user = dbc.get.one(User, user_id) # Blacklist all the whitelisted tokens # in use for the user that will be deleted diff --git a/server/app/apis/users.py b/server/app/apis/users.py index 7cff57c8118a8b854944042a5c8ea6aad2fe11da..bf393a89abcc0f85d4885e8333afab1e0df8b56f 100644 --- a/server/app/apis/users.py +++ b/server/app/apis/users.py @@ -8,6 +8,7 @@ import app.database.controller as dbc from app.apis import item_response, list_response, protect_route from app.core.dto import UserDTO from app.core.parsers import search_parser, sentinel +from app.database.models import User from flask_jwt_extended import get_jwt_identity from flask_restx import Resource, inputs, reqparse @@ -49,7 +50,7 @@ class UsersList(Resource): def get(self): """ Gets all users. """ - item = dbc.get.user(get_jwt_identity()) + item = dbc.get.one(User, get_jwt_identity()) return item_response(schema.dump(item)) @protect_route(allowed_roles=["*"]) @@ -57,7 +58,7 @@ class UsersList(Resource): """ Posts a new user using the specified arguments. """ args = user_parser_edit.parse_args(strict=True) - item = dbc.get.user(get_jwt_identity()) + item = dbc.get.one(User, get_jwt_identity()) item = _edit_user(item, args) return item_response(schema.dump(item)) @@ -69,7 +70,7 @@ class Users(Resource): def get(self, ID): """ Gets the specified user. """ - item = dbc.get.user(ID) + item = dbc.get.one(User, ID) return item_response(schema.dump(item)) @protect_route(allowed_roles=["Admin"]) @@ -77,7 +78,7 @@ class Users(Resource): """ Edits the specified team using the provided arguments. """ args = user_parser_edit.parse_args(strict=True) - item = dbc.get.user(ID) + item = dbc.get.one(User, ID) item = _edit_user(item, args) return item_response(schema.dump(item)) diff --git a/server/app/database/controller/get.py b/server/app/database/controller/get.py index aafa7664d393693598fbeb4ee9cc9bb3ca34452c..abdb9e15d1d815d80a48d7e80256e6595c7d7620 100644 --- a/server/app/database/controller/get.py +++ b/server/app/database/controller/get.py @@ -60,12 +60,6 @@ def user_exists(email): return User.query.filter(User.email == email).count() > 0 -def user(user_id): - """ Gets the user object associated with the provided user. """ - - return User.query.filter(User.id == user_id).first_extended() - - def user_by_email(email): """ Gets the user object associated with the provided email. """