getToken added

app/api/endpoints/transcribe.py

@@ -2,7 +2,6 @@ from typing import Annotated, List, Union
 
 from fastapi import APIRouter, File, UploadFile, Request, Header, HTTPException
 from fastapi.background import BackgroundTasks
-from pydantic import BaseModel
 
 from app.core.database import SessionLocal
 
@@ -13,16 +12,12 @@ from app.utils.utils import (
     get_model_name,
 )
 from app.core.models import AuthTokenController, TranscribeController
+from app.api.models import Transcription
 
 router = APIRouter()
 database = SessionLocal()
 
 
-class Transcription(BaseModel):
-    text: str
-    filename: str
-
-
 @router.post("/", response_model=Transcription)
 async def post_audio(
     background_tasks: BackgroundTasks,

app/api/endpoints/users.py

@@ -3,7 +3,14 @@ from uuid import UUID
 from fastapi import APIRouter, status, Depends, HTTPException
 from sqlalchemy.orm import Session
 
-from app.api.models.user import User, UserResponse, PasswordUpdate, UserDeletedResponse
+from app.api.models.user import (
+    User,
+    UserResponse,
+    PasswordUpdate,
+    UserDeletedResponse,
+    User_GET_TOKEN,
+    Response_Token,
+)
 from app.core.database import get_db
 from app.core.models.User import UserController
 
@@ -33,6 +40,24 @@ async def create_user(user: User, db: Session = Depends(get_db)):
         ) from exc
 
 
+@router.post(
+    "/get_token", status_code=status.HTTP_200_OK, response_model=Response_Token
+)
+async def get_user_token(user: User_GET_TOKEN, db: Session = Depends(get_db)):
+    try:
+        USER = UserController(db)
+        token = USER.read_token(user.email, user.password)
+        return {"token": token}
+    except HTTPException as exc:
+        logger.error(exc)
+        raise exc
+    except Exception as exc:
+        logger.error(exc)
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST, detail=str(exc)
+        ) from exc
+
+
 @router.get("/{user_id}/", status_code=status.HTTP_200_OK, response_model=UserResponse)
 async def read_user(user_id: UUID, db: Session = Depends(get_db)):
     try:

app/api/models/__init__.py

@@ -1,2 +1,2 @@
 from .user import User
-from .transcribe import Transcribe
+from .transcribe import Transcription

app/api/models/ping.py

@@ -0,0 +1,5 @@
+from pydantic import BaseModel
+
+
+class PingResponse(BaseModel):
+    ping: str

app/api/models/transcribe.py

@@ -11,5 +11,6 @@ from sqlalchemy.orm import Mapped, mapped_column, relationship
 from app.core.database import Base
 
 
-class Transcribe(BaseModel):
-    ...
+class Transcription(BaseModel):
+    text: str
+    filename: str

app/api/models/user.py

@@ -31,6 +31,15 @@ class PasswordUpdate(BaseModel):
     new_password: str
 
 
+class User_GET_TOKEN(BaseModel):
+    email: str
+    password: str
+
+
+class Response_Token(BaseModel):
+    token: str
+
+
 class User(UserBase):
     password: str
 

app/core/models/AuthToken.py

@@ -39,6 +39,12 @@ class AuthTokenController:
             raise Exception("Invalid Token!")
         return user.user_id
 
+    def get_token_from_user_id(self, user_id: UUID) -> str:
+        token = self.db.query(AuthToken).filter(AuthToken.user_id == user_id).first()
+        if not token:
+            raise Exception("Invalid Token!")
+        self.auth_token = token
+
     def create(self, user_id: UUID):
         self.user_id = user_id
         self.token = self.create_token()

app/core/models/User.py

@@ -86,6 +86,14 @@ class UserController:
             return
         AuthTokenController(self.db).create(self.db_user.id)
 
+    def read_token(self, email: str, password: str):
+        self.read_by_email(email)
+        if not verify_password(password, self.db_user.hashed_password):
+            raise HTTPException(status_code=400, detail="Incorrect password")
+        TOKEN = AuthTokenController(self.db)
+        TOKEN.get_token_from_user_id(self.db_user.id)
+        return TOKEN.get_token()
+
     def CheckUserIsExistsByEmailAndUsername(self, email: str, username: str):
         db_user = (
             self.db.query(UserInDB)
@@ -96,6 +104,14 @@ class UserController:
             return True
         return False
 
+    def read_by_email(self, email: str):
+        self.db_user = self.db.query(UserInDB).filter(UserInDB.email == email).first()
+        if not self.db_user:
+            raise HTTPException(
+                status_code=status.HTTP_404_NOT_FOUND, detail="User not found"
+            )
+        self.user = self.db_user.data()
+
     def read(self, user_id: uuid.UUID):
         self.db_user = self.db.query(UserInDB).filter(UserInDB.id == user_id).first()
         if not self.db_user:

app/main.py

@@ -4,6 +4,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from app.api import api_router
 from app.core.config import settings
 from app.core.errors import error_handler
+from app.api.models.ping import PingResponse
 
 from app.utils import print_routes
 from app.utils.checks import run_checks
@@ -27,7 +28,7 @@ if settings.BACKEND_CORS_ORIGINS:
     )
 
 
-@app.get("/ping")
+@app.get("/ping", tags=["ping"], response_model=PingResponse)
 async def ping():
     return {"ping": "pong"}
 
@@ -36,7 +37,6 @@ async def ping():
 app.include_router(api_router, prefix=settings.API_V1_STR)
 
 # # Error handlers
-# app.add_exception_handler(422, error_handler)
 app.add_exception_handler(500, error_handler)
 
 # Print all routes