.

Author: ProFastCode

app/api/anotations.py

@@ -1,5 +1,5 @@
 """
-Dependencies
+Dependency annotations
 """
 
 from typing import Annotated

app/core/db.py

@@ -2,8 +2,7 @@
 Database
 """
 
-from sqlalchemy.ext.asyncio import (AsyncEngine, async_sessionmaker,
-                                    create_async_engine)
+from sqlalchemy.ext.asyncio import AsyncEngine, async_sessionmaker, create_async_engine
 from sqlmodel.ext.asyncio.session import AsyncSession
 
 from app import repositories as repos

app/core/deps.py

@@ -8,7 +8,7 @@
 from fastapi.security import APIKeyHeader
 
 from app.core.db import Database, SessionLocal
-from app.core.logic import Logic
+from app.logic import Logic
 from app.core.security import Security
 from app.models.user import User
 
@@ -30,7 +30,7 @@ async def get_logic(
 
 
 async def get_current_user(
-    token: Annotated[str, Depends(APIKeyHeader(name='access-token'))],
+    token: Annotated[str, Depends(APIKeyHeader(name="access-token"))],
     logic: Annotated[Logic, Depends(get_logic)],
 ) -> User | None:
     return await logic.users.retrieve_by_token(token)

app/core/exps.py

@@ -5,12 +5,11 @@
 from fastapi import HTTPException, status
 
 # Users
-USER_EXISTS = HTTPException(status.HTTP_409_CONFLICT, 'User is already taken.')
-USER_NOT_FOUND = HTTPException(status.HTTP_404_NOT_FOUND, 'User not found.')
+USER_EXISTS = HTTPException(status.HTTP_409_CONFLICT, "User is already taken.")
+USER_NOT_FOUND = HTTPException(status.HTTP_404_NOT_FOUND, "User not found.")
 USER_IS_CORRECT = HTTPException(
-    status.HTTP_401_UNAUTHORIZED, 'User is correct'
-)
+    status.HTTP_401_UNAUTHORIZED, "User is correct.")
 
 # Tokens
-TOKEN_INVALID = HTTPException(status.HTTP_401_UNAUTHORIZED, 'Invalid token')
-TOKEN_EXPIRED = HTTPException(status.HTTP_401_UNAUTHORIZED, 'Token expired')
+TOKEN_INVALID = HTTPException(status.HTTP_401_UNAUTHORIZED, "Invalid token.")
+TOKEN_EXPIRED = HTTPException(status.HTTP_401_UNAUTHORIZED, "Token expired.")

app/core/logic.py

@@ -1,3 +1,12 @@
+from app.core.db import Database
+from app.core.security import Security
+
 from .users import Users
 
-__all__ = ['Users']
+
+class Logic:
+    def __init__(self, db: Database, security: Security):
+        self.users = Users(db, security)
+
+
+__all__ = ["Logic"]