mirror of
https://github.com/velocitatem/cvfs.git
synced 2026-05-31 16:53:38 +00:00
53 lines
1.8 KiB
Python
53 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
from fastapi import APIRouter, Depends, File, Form, HTTPException, UploadFile
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.api.deps import get_current_user, get_db
|
|
from app.schemas import DocumentListResponse, DocumentResponse
|
|
from app.services.documents import create_document, get_document, list_documents
|
|
from dlib.auth import AuthenticatedUser
|
|
|
|
|
|
router = APIRouter(prefix="/documents", tags=["documents"])
|
|
|
|
|
|
@router.get("/", response_model=DocumentListResponse)
|
|
async def list_user_documents(
|
|
session: AsyncSession = Depends(get_db),
|
|
user: AuthenticatedUser = Depends(get_current_user),
|
|
):
|
|
documents = await list_documents(session, owner_id=user.sub)
|
|
payload = [DocumentResponse.model_validate(doc) for doc in documents]
|
|
return DocumentListResponse(items=payload)
|
|
|
|
|
|
@router.get("/{document_id}", response_model=DocumentResponse)
|
|
async def get_user_document(
|
|
document_id: str,
|
|
session: AsyncSession = Depends(get_db),
|
|
user: AuthenticatedUser = Depends(get_current_user),
|
|
):
|
|
document = await get_document(session, owner_id=user.sub, document_id=document_id)
|
|
if not document:
|
|
raise HTTPException(status_code=404, detail="Document not found")
|
|
return DocumentResponse.model_validate(document)
|
|
|
|
|
|
@router.post("/", response_model=DocumentResponse)
|
|
async def upload_document(
|
|
title: str = Form(...),
|
|
description: str | None = Form(default=None),
|
|
file: UploadFile = File(...),
|
|
session: AsyncSession = Depends(get_db),
|
|
user: AuthenticatedUser = Depends(get_current_user),
|
|
):
|
|
document = await create_document(
|
|
session,
|
|
owner_id=user.sub,
|
|
title=title,
|
|
description=description,
|
|
upload=file,
|
|
)
|
|
return DocumentResponse.model_validate(document)
|