Add Config and Exception Handlers

backend/core/ConfigEnv.py

@@ -0,0 +1,32 @@
+"""Config class for handling env variables.
+"""
+from functools import lru_cache
+
+from pydantic import BaseSettings
+
+
+class Settings(BaseSettings):
+    HOSTNAME: str
+    DATABASE: str
+    UID: str
+    PASSWORD: str
+    ALGORITHM:str
+    JWT_SECRET_KEY:str
+    JWT_REFRESH_SECRET_KEY:str
+    # OPENAI_KEY:str
+    APP_ID:str
+    USER_ID:str
+    MODEL_ID:str
+    CLARIFAI_PAT:str
+    MODEL_VERSION_ID:str
+    
+    class Config:
+        env_file = ".env"
+
+
+@lru_cache()
+def get_settings():
+    return Settings()
+
+
+config = get_settings()

backend/core/ExceptionHandlers.py

@@ -0,0 +1,27 @@
+from docguptea import app
+from .Exceptions import *
+
+from fastapi.responses import JSONResponse
+from fastapi.requests import Request
+from fastapi import status
+
+
+
+@app.exception_handler(ExistingUserException)
+async def handle_existing_user_found(request: Request, exec: ExistingUserException):
+    return JSONResponse(status_code=status.HTTP_226_IM_USED,
+                        content=repr(exec)
+                        )
+
+
+@app.exception_handler(InvalidCredentialsException)
+async def handle_login_failed(request: Request, exec: InvalidCredentialsException):
+    return JSONResponse(status_code=status.HTTP_403_FORBIDDEN,
+                        content=repr(exec)
+                        )
+
+@app.exception_handler(InfoNotFoundException)
+async def handle_info_not_found(request: Request, exec: InfoNotFoundException):
+    return JSONResponse(status_code=status.HTTP_401_UNAUTHORIZED,
+                        content=repr(exec)
+                        )