Spaces:
Sleeping
Sleeping
File size: 3,104 Bytes
7190fd0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 | from fastapi import APIRouter, Depends, HTTPException
from sqlalchemy import select, func
from sqlalchemy.ext.asyncio import AsyncSession
from app.database import get_db
from app.models.user import User
from app.models.project import Project
from app.schemas.project import ProjectCreate, ProjectUpdate, ProjectResponse, ProjectListResponse
from app.services.auth import get_current_user
router = APIRouter(prefix="/api/projects", tags=["projects"])
@router.get("", response_model=ProjectListResponse)
async def list_projects(
skip: int = 0,
limit: int = 20,
user: User = Depends(get_current_user),
db: AsyncSession = Depends(get_db),
):
result = await db.execute(
select(Project)
.where(Project.user_id == user.id)
.order_by(Project.updated_at.desc())
.offset(skip).limit(limit)
)
projects = list(result.scalars().all())
count_result = await db.execute(
select(func.count()).select_from(Project).where(Project.user_id == user.id)
)
total = count_result.scalar() or 0
return ProjectListResponse(projects=projects, total=total)
@router.post("", response_model=ProjectResponse, status_code=201)
async def create_project(
data: ProjectCreate,
user: User = Depends(get_current_user),
db: AsyncSession = Depends(get_db),
):
project = Project(user_id=user.id, **data.model_dump(exclude_none=True))
db.add(project)
await db.commit()
await db.refresh(project)
return project
@router.get("/{project_id}", response_model=ProjectResponse)
async def get_project(
project_id: int,
user: User = Depends(get_current_user),
db: AsyncSession = Depends(get_db),
):
result = await db.execute(
select(Project).where(Project.id == project_id, Project.user_id == user.id)
)
project = result.scalar_one_or_none()
if not project:
raise HTTPException(status_code=404, detail="Project not found")
return project
@router.patch("/{project_id}", response_model=ProjectResponse)
async def update_project(
project_id: int,
data: ProjectUpdate,
user: User = Depends(get_current_user),
db: AsyncSession = Depends(get_db),
):
result = await db.execute(
select(Project).where(Project.id == project_id, Project.user_id == user.id)
)
project = result.scalar_one_or_none()
if not project:
raise HTTPException(status_code=404, detail="Project not found")
for key, value in data.model_dump(exclude_none=True).items():
setattr(project, key, value)
await db.commit()
await db.refresh(project)
return project
@router.delete("/{project_id}", status_code=204)
async def delete_project(
project_id: int,
user: User = Depends(get_current_user),
db: AsyncSession = Depends(get_db),
):
result = await db.execute(
select(Project).where(Project.id == project_id, Project.user_id == user.id)
)
project = result.scalar_one_or_none()
if not project:
raise HTTPException(status_code=404, detail="Project not found")
await db.delete(project)
await db.commit()
|