Add crud functionality

Author: aspects19

app/crud.py

@@ -1,6 +1,6 @@
 from sqlalchemy.orm import Session
 from fastapi import HTTPException
-from app import schemas, models
+import schemas, models
 
 
 def get_user(db: Session, user_id: int):

app/main.py

@@ -1,15 +1,46 @@
-from typing import Union
-
-from fastapi import FastAPI
+import uvicorn
+from fastapi import FastAPI, HTTPException, Depends
+from sqlalchemy.orm import Session
+import crud, models, schemas
+from database import engine, get_db, SessionLocal
 
 app = FastAPI()
 
+models.Base.metadata.create_all(bind=engine)
 
 @app.get("/")
 def read_root():
     return {"Hello": "World"}
 
+@app.post("/users/", response_model=schemas.User)
+def create_user(user: schemas.UserCreate, db: Session = Depends(get_db)):
+    return crud.create_user(db=db, user=user)
+
+@app.get("/users/", response_model=list[schemas.User])
+def read_users(skip:int=0, limit:int=100, db: Session = Depends(get_db)):
+    users = crud.get_users(db, skip=skip, limit=limit)
+    return users
+
+@app.get("/users/{user_id}/",response_model=schemas.User)
+def get_user(user_id:int, db:Session=Depends(get_db)):
+    db_user = crud.get_user(db,user_id =user_id )
+    if db_user is None:
+        raise HTTPException(status_code=404, detail="User not found")
+    return db_user
+
+@app.put("/users/{user_id}", response_model=schemas.User)
+def update_user(user_id: int, user: schemas.UserCreate, db: Session = Depends(get_db)):
+    db_user = crud.update_user(db, user_id=user_id, updated_user=user)
+    if db_user is None:
+        raise HTTPException(status_code=404, detail="User not found")
+    return db_user
+
+@app.delete("/users/{user_id}", response_model=schemas.User)
+def delete_user(user_id: int, db: Session = Depends(get_db)):
+    db_user = crud.delete_user(db, user_id=user_id)
+    if db_user is None:
+        raise HTTPException(status_code=404, detail="User not found")
+    return db_user
 
-@app.get("/items/{item_id}")
-def read_item(item_id: int, q: Union[str, None] = None):
-    return {"item_id": item_id, "q": q}
+if __name__ == "__main__":
+    uvicorn.run("main:app", reload=True)

app/models.py

@@ -1,6 +1,6 @@
 from sqlalchemy import Boolean, Column, Integer, String, DateTime
 from sqlalchemy.sql import func
-from app.database import Base
+from database import Base
 
 class User(Base):
     __tablename__ = "users"
@@ -9,4 +9,4 @@ class User(Base):
     premium_user = Column(Boolean)
     payment_id = Column(String(255))
     created_at = Column(DateTime(timezone=True), default=func.now() )
-    updated_at = Column(DateTime(timezone=True), default=func.now(), onupdate=func.now() )
+    updated_at = Column(DateTime(timezone=True), default=func.now(), onupdate=func.now())