main.py

from fastapi import FastAPI , Depends , status , HTTPException , Query , Path , Body
from db.create_db import create_tables
from models.models_data import User , UserLog , UserTasks , AiRequest
from sqlalchemy.ext.asyncio import AsyncSession
from typing import Annotated , List
from db.data_manipulation import (create_user , 
                                  user_verification,
                                  create_user_task,
                                  get_users_task,
                                  delete_tasks,
                                  update_tasks,
                                  completed_tasks
                                  )
from Depends.create_session import get_session
from Depends.auth_scheme import encode_jwt , decode_jwt
from model_ai.ai_text import sending_a_reply
import json
app = FastAPI(
    title="To Do project📝"
)

@app.on_event("startup")
async def on_startup():
    await create_tables()

@app.get("/home/" ,
         summary="Главная страница" ,
         tags=["Home⬆"])
async def home () :
    return {
        "message" : "Главная страница ⬆"
    } 
    
@app.post("/auth/register/" ,
          summary="Регистрация ✅",
          tags=["Users🙍‍♂️"] ,
          status_code=status.HTTP_201_CREATED)

async def user_register (user : User , 
                         session : Annotated[AsyncSession , Depends(get_session)] ,
                         ) :
    result = await create_user(session , user)
    
    if result :
        
        return result
    
    raise HTTPException(status_code=status.HTTP_409_CONFLICT,
                        detail= "This user already exists in the database.")

@app.post("/auth/login/" , summary="Авторизация🔐" , tags=["Users🙍‍♂️"])
async def user_login (userlog : UserLog ,
                      session : Annotated[AsyncSession , Depends(get_session)]) :
    
    result = await user_verification(session , userlog)
    
    if result :
        res = encode_jwt(result)
        return {
            "token" : res  
        }
    raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST,
                        detail="Incorrect data or you are not registered")

@app.post("/tasks/" ,
          summary="Добавить задачу",
          tags=["Tasks📝"],
          status_code=status.HTTP_201_CREATED
          )
async def create_tasks (usertasks : List[ UserTasks] ,
                        user_id : Annotated[int , Depends(decode_jwt)],
                        session : Annotated[AsyncSession , Depends(get_session)],                       
                        ) :
    
   return await create_user_task(session , user_id , usertasks)
    

@app.get("/tasks/" ,
        summary="Получить задачу",
        tags=["Tasks📝"]
        )
async def get_task (session : Annotated[AsyncSession , Depends(get_session)],
                    user_id : Annotated[int , Depends(decode_jwt)],
                    filtering_by_title : Annotated[str | None, Query()] = None) :
    
    return await get_users_task(session , user_id , filtering_by_title)

@app.delete("/tasks/delete_task/{task_id}/",
            summary="Удолить задачу",
            tags=["Операции над задачами 🔍"]
            )
async def  delete_task (session : Annotated[AsyncSession , Depends(get_session)] ,
                        user_id : Annotated[int , Depends(decode_jwt)],
                        task_id : Annotated[int ,Path()]
                        ) :
    result = await  delete_tasks(session ,  user_id , task_id)
    if result :
        return result
    raise HTTPException(status_code=status.HTTP_404_NOT_FOUND)

@app.put("/tasks/update_tasks/{id}/" ,
        summary="Обновление задач" ,
        tags=["Операции над задачами 🔍"] 
        )
async def update_task (session : Annotated[AsyncSession , Depends(get_session)],
                       user_id : Annotated[int , Depends(decode_jwt)],
                       id : Annotated[int , Path()],
                       updated_task : UserTasks
                       ) :
    result = await update_tasks(session ,
                                user_id ,
                                id ,
                                updated_task)
    if result :
        return result
    
    raise HTTPException(status_code=status.HTTP_404_NOT_FOUND,
                        detail="Taks not found"
                        )
    
@app.put("/created/{id}/" , 
         summary="Выполнил задачу", 
         tags=["Операции над задачами 🔍"]
         )
async def created_task (session : Annotated[AsyncSession , Depends(get_session)],
                        id : Annotated[int , Path()],
                        user_id : Annotated[int , Depends(decode_jwt)]
                        ) :
    result = await completed_tasks(session , user_id , id)
    
    if result :
        return result
    
    raise HTTPException(status_code=status.HTTP_404_NOT_FOUND, 
                        detail="task not found"
                        )
    
@app.post("/task_ai/",
         summary= "Создание задач с помощью ИИ",
         tags=["AI 🤖"]
        )
async def creat_task_ai (
        user_prompt : AiRequest,
        _: Annotated[int , Depends(decode_jwt)]
        ) :
    
    result = await sending_a_reply(user_prompt.prompt)
    if result : 
        result_json = json.loads(result)
        return result_json 
    raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST)