dingdanquanliucheng/backend/app/api/files.py

from fastapi import APIRouter, Depends
from sqlalchemy.orm import Session

from backend.app.api.deps import get_current_user
from backend.app.db import get_db_session
from backend.app.schemas.common import success_payload
from backend.app.schemas.files import CreateUploadTokenRequest, SaveAttachmentRequest
from backend.app.services.file_service import file_service

router = APIRouter(prefix="/api/files", tags=["files"])


@router.post("/upload-token")
def create_upload_token(
    payload: CreateUploadTokenRequest,
    session: Session = Depends(get_db_session),
    current_user: dict = Depends(get_current_user),
) -> dict:
    return success_payload(file_service.create_upload_token(payload.model_dump(), session))


@router.post("/attachments")
def save_attachment(
    payload: SaveAttachmentRequest,
    session: Session = Depends(get_db_session),
    current_user: dict = Depends(get_current_user),
) -> dict:
    return success_payload(file_service.save_attachment({**payload.model_dump(), "created_by": current_user.get("user_id")}, session))