main.py

from email import message

from fastapi import FastAPI, Request, HTTPException, Response
from pydantic import BaseModel,field_validator
import sqlite3
import os
from dotenv import load_dotenv
import psycopg
app = FastAPI()

################################################################################
#DATABASE
load_dotenv()


DATABASE_URL = os.getenv("DATABASE_URL")

def get_db_connection():
    return psycopg.connect(DATABASE_URL)

def init_db():
    with get_db_connection() as con:
        with con.cursor() as cur:
            cur.execute("CREATE TABLE IF NOT EXISTS tasks(id SERIAL PRIMARY KEY, title TEXT, done BOOLEAN)")

            cur.execute("SELECT COUNT(*) FROM tasks")
            COUNT = cur.fetchone()[0]

            if COUNT ==0:
                cur.executemany("INSERT INTO tasks(title, done) VALUES(%s, %s)", [
                    ("Task 0", True),
                    ("Task 1", True),
                    ("Task 2", False)
                ])
        con.commit()

#####################################################################################
@app.on_event("startup")
def on_startup():
    init_db()
#####################################################################################
@app.get("/test")
async def test():
    return {}
###############################################################################
@app.get("/", summary="API information")
async def root():
    return {"name": "Task API", "version": "1.0", "endpoints": ["/tasks"] }

@app.get("/health",summary="Check API health")
async def health():
    return {"status": "ok"}
###############################################################################

###############################################################################
@app.get("/tasks", summary="Get all tasks")
async def get_tasks():
    with get_db_connection() as con:
        with con.cursor() as cur:
            cur.execute("SELECT * FROM tasks")
            taskx = cur.fetchall()
    return taskx


@app.get("/tasks/{id}", summary="Get task by ID")
async def get_task_by_id(id: int):
    with get_db_connection() as con:
        with con.cursor() as cur:
            id_valued = cur.execute("SELECT * FROM tasks WHERE id = %s", (id,))
            task = id_valued.fetchone()
            if task:
                    return task
            return {"error": f"Task {id} not found"}

###############################################################################
class Task(BaseModel):
    title: str

    @field_validator("title")
    def title_not_empty(cls, value):
        if value.strip() == "":
            raise ValueError("Title cannot be empty")

        return value


@app.post("/tasks", status_code=201)
async def create_task(task: Task):
    
    with get_db_connection() as con:
        with con.cursor() as cur:
            cur.execute(
                "INSERT INTO tasks(title, done) VALUES(%s, %s)",
                (task.title, False)
            )

            con.commit()

    return task

###############################################################################
from pydantic import BaseModel, field_validator


class UpdateTask(BaseModel):
    title: str
    done: bool

    @field_validator("title")
    def title_not_empty(cls, value):
        if value.strip() == "":
            raise ValueError("Title cannot be empty")
        return value


@app.put("/tasks/{id}", summary="Update a task")
async def update_task(id: int, req: UpdateTask):
    with get_db_connection() as con:
        with con.cursor() as cur:
            cur.execute(
                "SELECT * FROM tasks WHERE id = %s",
                (id,)
            )

            task = cur.fetchone()

            if task is None:
                raise HTTPException(
                    status_code=404,
                    detail=f"Task {id} not found"
                )

            cur.execute(
                """
                UPDATE tasks 
                SET title = %s, done = %s
                WHERE id = %s
                """,
                (req.title, req.done, id)
            )

            con.commit()

            return {
                "id": id,
                "title": req.title,
                "done": req.done
            }




@app.delete("/tasks/{id}", status_code=204, summary="Delete a task")
async def delete_task(id: int):
    with get_db_connection() as con:
        with con.cursor() as cur:
            cur.execute("DELETE FROM tasks WHERE id = %s", (id,))

            if cur.rowcount > 0:
                return Response(status_code=204)

            raise HTTPException(
                status_code=404,
                detail=f"Task {id} not found"
            )