Initial working FastAPI app with PostGreSQL

Author: sathya

.gitignore

@@ -0,0 +1,24 @@
+# Python
+__pycache__/
+*.py[cod]
+*.pyo
+
+# Environments
+env/
+venv/
+.venv/
+
+# dotenv
+.env
+.env.*
+
+# IDE
+.idea/
+.vscode/
+
+# macOS
+.DS_Store
+
+# DB backups / misc
+*.sqlite3
+*.log

crud.py

@@ -0,0 +1,38 @@
+from models import Item
+from sqlalchemy.orm import Session
+from schemas import ItemCreate
+
+# CREATE
+def create_item(db: Session, item: ItemCreate):
+    db_item = Item(**item.model_dump())
+    db.add(db_item)
+    db.commit()
+    db.refresh(db_item)
+    return db_item
+
+# READ
+def get_items(db: Session):
+    return db.query(Item).all()
+
+# READ ONLY ONE
+def get_item(db: Session, item_id: int):
+    return db.query(Item).filter(Item.id == item_id).first()
+
+def update_item(db: Session, item_id: int, updated_data: ItemCreate):
+    item = db.query(Item).filter(Item.id == item_id).first()
+    if item:
+        item.name = updated_data.name
+        item.description = updated_data.description
+        db.commit()
+        db.refresh(item)
+
+        return item
+    return None
+
+def delete_item(db: Session, item_id: int):
+    item = db.query(Item).filter(Item.id == item_id).first()
+    if item:
+        db.delete(item)
+        db.commit()
+        return item
+    return None

database.py

@@ -0,0 +1,14 @@
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker, declarative_base
+from dotenv import load_dotenv
+import os
+
+load_dotenv()
+
+DATABASE_URL= os.getenv("DATABASE_URL")
+
+engine = create_engine(DATABASE_URL)
+
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+
+Base = declarative_base()

main.py

@@ -0,0 +1,47 @@
+from database import SessionLocal, engine, Base
+import models, schemas, crud
+from fastapi import FastAPI, Depends, HTTPException
+from sqlalchemy.orm import Session
+
+models.Base.metadata.create_all(bind=engine)
+print("Table is created")
+
+def get_db():
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
+
+# Creating Routes
+app = FastAPI()
+
+@app.post("/items/", response_model= schemas.Item)
+def create_item(item: schemas.ItemCreate, db: Session = Depends(get_db)):
+    return crud.create_item(db, item)
+
+@app.get("/items/", response_model=list[schemas.Item])
+def read_items(db: Session = Depends(get_db)):
+    return crud.get_items(db)
+
+@app.get("/items/{item_id}", response_model=schemas.Item)
+def read_item(item_id: int, db: Session = Depends(get_db)):
+    item= crud.get_item(db, item_id)
+    if not item:
+        raise HTTPException(status_code=404, detail="Item NOT FOUND")
+    return item
+
+@app.put("/items/{item_id}", response_model= schemas.Item)
+def update_item(item_id: int, item: schemas.ItemCreate, db: Session= Depends(get_db)):
+    updated_item = crud.update_item(db, item_id, item)
+    if not updated_item:
+        raise HTTPException(status_code=404, detail="Item NOT FOUND")
+    return updated_item
+
+@app.delete("/items/{item_id}")
+def delete_item(item_id: int, db: Session= Depends(get_db)):
+    deleted_item = crud.delete_item(db, item_id)
+    if not deleted_item:
+        raise HTTPException(status_code=404, detail="Item NOT FOUND")
+    return {"details": "Deleted"}
+    

models.py

@@ -0,0 +1,9 @@
+from database import Base
+from sqlalchemy import Column, Integer, String
+
+class Item(Base):
+    __tablename__ = "items"
+
+    id = Column(Integer, primary_key=True, index=True)
+    name = Column(String, index=True)
+    description = Column(String)

schemas.py

@@ -0,0 +1,15 @@
+from pydantic import BaseModel
+# For data validation while performing CRUD operations
+
+class ItemBase(BaseModel):
+    name: str
+    description: str
+
+class ItemCreate(ItemBase):
+    pass
+
+class Item(ItemBase):
+    id: int
+    model_config = {
+        "from_attributes": True
+    }