main.py

from fastapi import Depends,FastAPI
from fastapi.middleware.cors import CORSMiddleware
from models import product
from database import session,engine
import database_models
from sqlalchemy.orm import Session

app=FastAPI()

app.add_middleware(
    CORSMiddleware,
    allow_origins=["http://localhost:3000"],
    allow_methods=["*"]
)
database_models.Base.metadata.create_all(bind=engine)

@app.get("/")
def greet():
    return "welcome to telusko trac"

products=[
    product(id=1,name="Phone",description="a smartphone",price=699.99,quantity=50),
    product(id=2,name="laptop",description="a powerful laptop",price=999.99,quantity=30),
    product(id=5,name="Pen",description="a blue ink pen",price=1.99,quantity=100),
    product(id=6,name="table",description="a wooden table",price=199.99,quantity=20),
]

def get_db():
    db=session()
    try: 
        yield db
    finally: 
        db.close()

def init_db():
    db=session()
    count=db.query(database_models.product).count

    if count==0:
        for product in products:
            db.add(database_models.product(**product.model_dump()))

        db.commit()

init_db()

@app.get("/products")

def get_all_products(db:Session=Depends(get_db)):
    db_products=db.query(database_models.product).all()

    return db_products

@app.get("/products/{id}")

def get_product_by_id(id: int ,db:Session=Depends(get_db)):
    db_product=db.query(database_models.product).filter(database_models.product.id==id).first()
    if db_product:
        return db_product
    return "Product not found"

@app.post("/products")

def add_product(product:product,db:Session=Depends(get_db)):
    db.add(database_models.product(**product.model_dump()))
    db.commit()
    return product

@app.put("/products/{id}")
def update_product(id:int ,product:product,db:Session=Depends(get_db)):
   db_product=db.query(database_models.product).filter(database_models.product.id==id).first()
   if db_product:
       db_product.name=product.name
       db_product.description=product.description
       db_product.price=product.price
       db_product.quantity=product.quantity
       db.commit()
       return "product updated"
   else:
       return "No product found"

@app.delete("/products/{id}")
def delete_product(id: int,db:Session=Depends(get_db)):

    db_product=db.query(database_models.product).filter(database_models.product.id==id).first()
    if db_product:
        db.delete(db_product)
        db.commit()
    else:
        return "no product found"