BIG refactors

src/main.py

@@ -9,22 +9,15 @@ from sqlalchemy.orm import Session
 
 import crud
 from database import SessionLocal, engine
-from enums import MinigameEnum, CourseEnum
+from enums import CourseEnum, MinigameEnum
 from models import Base
-from schemas import highscores, users, courseprogress
+from schemas import courseprogress, highscores, users
 
 app = FastAPI()
 
+
 Base.metadata.create_all(bind=engine)
 
-
-pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
-
-# JWT authentication setup
-jwt_secret = "secret_key"
-ALGORITHM = "HS256"
-ACCESS_TOKEN_EXPIRE_MINUTES = 44640 # One month
-
 bearer_scheme = HTTPBearer()
 
 
@@ -35,12 +28,11 @@ def get_db():
     finally:
         db.close()
 
-
-def get_current_user(
+def get_current_user_name(
     token: HTTPAuthorizationCredentials = Depends(bearer_scheme),
 ):
     try:
-        payload = jwt.decode(token.credentials, jwt_secret, algorithms=[ALGORITHM])
+        payload = jwt.decode(token.credentials, crud.jwt_secret, algorithms=[crud.ALGORITHM])
         username = payload.get("sub")
         if username is None:
             raise HTTPException(status_code=401, detail="Invalid JWT token")
@@ -60,45 +52,33 @@ async def read_users(db: Session = Depends(get_db)):
 
 
 @app.patch("/users")
-async def patch_current_user(user: users.UserCreate, current_user = Depends(get_current_user), db: Session = Depends(get_db)):
-    db_user = crud.get_us
-    return users
+async def patch_current_user(
+    user: users.UserCreate,
+    current_user_name = Depends(get_current_user_name),
+    db: Session = Depends(get_db),
+):
+    crud.patch_user(db, current_user_name, user)
 
 
 @app.post("/register", response_model=users.User)
 async def register(user: users.UserCreate, db: Session = Depends(get_db)):
-    db_user = crud.get_user_by_username(db, username=user.username)
-    if db_user:
-        raise HTTPException(status_code=400, detail="Username already registered")
-    return crud.create_user(
-        db=db, username=user.username, hashed_password=pwd_context.hash(user.password)
+    return crud.register(
+        db, user.username, user.password
     )
 
 
 @app.post("/login")
 async def login(user: users.UserCreate, db: Session = Depends(get_db)):
-    user = authenticate_user(user, db)
-    if not user:
-        raise HTTPException(status_code=401, detail="Invalid username or password")
-    access_token_expires = timedelta(minutes=ACCESS_TOKEN_EXPIRE_MINUTES)
-    access_token_payload = {
-        "sub": user.username,
-        "exp": datetime.utcnow() + access_token_expires,
-    }
-    access_token = jwt.encode(access_token_payload, jwt_secret, algorithm=ALGORITHM)
-    return {"access_token": access_token}
+    return crud.login(db, user)
 
 
 @app.get("/highscores", response_model=List[users.UserHighScore])
 async def read_high_scores(
     db: Session = Depends(get_db),
     minigame: Optional[MinigameEnum] = None,
-    n_highest: Optional[int] = None,
+    nr_highest: Optional[int] = None,
 ):
-    if n_highest < 1:
-        raise HTTPException(status_code=400, detail="Invalid number of high scores") 
-    high_scores = crud.get_high_scores(db, minigame, n_highest)
-    return high_scores
+    return crud.get_high_scores(db, minigame, nr_highest)
 
 
 @app.post("/highscores", response_model=highscores.HighScore)
@@ -111,22 +91,9 @@ async def create_high_score(
 #### TESTING!! DELETE LATER
 
 
-async def get_current_user(
-    token: HTTPAuthorizationCredentials = Depends(bearer_scheme),
-):
-    try:
-        payload = jwt.decode(token.credentials, jwt_secret, algorithms=[ALGORITHM])
-        username = payload.get("sub")
-        if username is None:
-            raise HTTPException(status_code=401, detail="Invalid JWT token")
-        return username
-    except jwt.exceptions.DecodeError:
-        raise HTTPException(status_code=401, detail="Invalid JWT token")
-
-
 @app.get("/protected")
-async def protected_route(current_user = Depends(get_current_user)):
-    return {"message": f"Hello, {current_user}!"}
+async def protected_route(current_user_name=Depends(get_current_user_name)):
+    return {"message": f"Hello, {current_user_name}!"}
 
 
 def authenticate_user(user: users.UserCreate, db: Session = Depends(get_db)):
@@ -139,10 +106,10 @@ def authenticate_user(user: users.UserCreate, db: Session = Depends(get_db)):
     return db_user
 
 
-
-
-
 @app.get("/courseprogress", response_model=List[courseprogress.CourseProgressBase])
-async def get_course_progress(course: Optional[CourseEnum] = CourseEnum.All, current_user = Depends(get_current_user), db: Session = Depends(get_db)):
-    user = crud.get_user_by_username(db, current_user)
-    return crud.get_course_progress(db = db, user = user, course = course)
+async def get_course_progress(
+    course: Optional[CourseEnum] = CourseEnum.All,
+    current_user_name=Depends(get_current_user_name),
+    db: Session = Depends(get_db),
+):
+    return crud.get_course_progress(db, current_user_name, course)