debug - database functions

Author: AryaAhmadii

.gitignore

@@ -1,3 +1,4 @@
-/__pycache__/
+/__pycache__
+/app/__pycache__
 /venv
 .env

app/crud.py

@@ -4,4 +4,25 @@
 
 def create_user(db: Session, user: models.UserCreate):
 
-    db_user = models.User
+    db_user = models.User(name=user.name, email=user.email, password=user.password)
+    db.add(db_user)
+    db.commit()
+    db.refresh(db_user)
+    return db_user
+
+
+def get_users(db: Session):
+    
+    return db.query(models.User).all()
+
+
+
+def get_user(db: Session, user_id: int):
+    
+    return db.query(models.User).filter(models.User.id == user_id).first()
+    
+
+def delete_user(db: Session, user_id: int):
+    
+    db.query(models.User).filter(models.User.id == user_id).delete()
+    db.commit()

app/database.py

@@ -1,5 +1,7 @@
 from sqlalchemy import create_engine
 from dotenv import load_dotenv
+from sqlalchemy.orm import sessionmaker
+from sqlalchemy.ext.declarative import declarative_base
 import os
 
 load_dotenv()
@@ -13,3 +15,17 @@
 DATABASE_URL = f"postgresql://{DB_USER}:{DB_PASSWORD}@{DB_HOST}/{DB_NAME}"
 
 engine = create_engine(DATABASE_URL)
+
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+
+
+Base = declarative_base()
+
+
+def get_db():
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
+    

app/main.py

@@ -1,11 +1,11 @@
 from fastapi import FastAPI
-from .routers.user import user_router
+from .routers.user import router
 from .database import engine
 
 
 app = FastAPI()
 
-app.include_router(user_router)
+app.include_router(router)
 
 #===========================#
 

app/models.py

@@ -1,4 +1,6 @@
 from pydantic import BaseModel
+from .database import Base
+from sqlalchemy import Column, String
 
 class UserCreate(BaseModel):
     name: str
@@ -9,4 +11,13 @@ class UserCreate(BaseModel):
 class UserUpdate(BaseModel):
     name: str
     email: str
-    password: str
+    password: str
+    
+
+class User(Base):
+    __tablename__ = "users"
+    
+    id = Column(primary_key=True)
+    name = Column(String, nullable=False)
+    email = Column(String, nullable=False)
+    password = Column(String, nullable=False)

app/routers/user.py

@@ -3,14 +3,13 @@
 from sqlalchemy.ext.declarative import declarative_base
 from typing import List
 
-from ..database import engine
+from ..database import engine, get_db
 from ..models import UserCreate, UserUpdate
-from ..crud import create_user, get_user, update_user, elete_user
+from ..crud import create_user, get_user, update_user, delete_user, get_users
 
 
-Base = declarative_base()
-
-Base.metadata.create_all(bind=engine)
-
 router = APIRouter()
 
+@router.get("/users")
+async def get_all_users(db: Session = Depends(get_db)):
+    return get_users(db=db)