dingdanquanliucheng/backend/app/api/files.py

29 lines
1.0 KiB
Python

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))