mirror of
https://github.com/a-mayb3/Kanban_clone_backend.git
synced 2026-03-21 10:05:38 +01:00
Started working on projects related endpoints
This commit is contained in:
parent
0b676688c2
commit
abd4d5e988
6 changed files with 93 additions and 63 deletions
64
main.py
64
main.py
|
|
@ -1,66 +1,10 @@
|
|||
from enum import Enum
|
||||
from fastapi import FastAPI, HTTPException, Depends
|
||||
from pydantic import BaseModel, ConfigDict
|
||||
from typing import List, Annotated, Optional
|
||||
|
||||
import models
|
||||
from database import SessionLocal, engine
|
||||
from sqlalchemy.orm import Session, joinedload
|
||||
|
||||
from misc.tasks import TaskBase
|
||||
from misc.users import UserBase
|
||||
from misc.projects import ProjectBase, ProjectCreate
|
||||
from routers.tasks import router as tasks_router
|
||||
from routers.projects import router as projects_router
|
||||
|
||||
app = FastAPI()
|
||||
|
||||
models.Base.metadata.create_all(bind=engine)
|
||||
|
||||
def get_db():
|
||||
db = SessionLocal()
|
||||
try:
|
||||
yield db
|
||||
finally:
|
||||
db.close()
|
||||
|
||||
db_dependency = Annotated[Session, Depends(get_db)]
|
||||
|
||||
@app.post("/projects/", response_model=ProjectBase)
|
||||
def create_project(project: ProjectCreate, db: db_dependency):
|
||||
db_project = models.Project(name=project.name, description=project.description)
|
||||
db.add(db_project)
|
||||
db.commit()
|
||||
db.refresh(db_project)
|
||||
|
||||
for task in project.tasks:
|
||||
db_task = models.Task(
|
||||
title=task.title,
|
||||
description=task.description,
|
||||
status=task.status.value,
|
||||
project_id=db_project.id
|
||||
)
|
||||
db.add(db_task)
|
||||
db.commit()
|
||||
|
||||
if project.user_ids:
|
||||
users = db.query(models.User).filter(models.User.id.in_(project.user_ids)).all()
|
||||
db_project.users.extend(users)
|
||||
db.commit()
|
||||
|
||||
db_project = db.query(models.Project).options(
|
||||
joinedload(models.Project.tasks),
|
||||
joinedload(models.Project.users)
|
||||
).filter(models.Project.id == db_project.id).first()
|
||||
|
||||
return db_project
|
||||
|
||||
@app.get("/projects/{project_id}", response_model=ProjectBase)
|
||||
def read_project(project_id: int, db: db_dependency):
|
||||
db_project = db.query(models.Project).options(
|
||||
joinedload(models.Project.tasks),
|
||||
joinedload(models.Project.users)
|
||||
).filter(models.Project.id == project_id).first()
|
||||
if db_project is None:
|
||||
raise HTTPException(status_code=404, detail="Project not found")
|
||||
return db_project
|
||||
|
||||
app.include_router(tasks_router)
|
||||
app.include_router(projects_router)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue