45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
from fastapi import APIRouter, Depends, File, Form, UploadFile
|
|
from sqlalchemy.orm import Session
|
|
|
|
from app.common.db.session import get_db
|
|
from app.common.responses.api_response import success_response
|
|
from app.common.security.deps import get_current_user
|
|
from app.models.entities import User
|
|
from app.modules.assets.service import AssetsService
|
|
|
|
|
|
router = APIRouter(prefix="/api/v1/assets", tags=["assets"])
|
|
|
|
|
|
@router.post("/upload-token")
|
|
def create_upload_token(mediaType: str = "image", db: Session = Depends(get_db)):
|
|
return success_response(AssetsService(db).create_upload_token(mediaType))
|
|
|
|
|
|
@router.post("")
|
|
def upload_asset(
|
|
file: UploadFile = File(...),
|
|
mediaType: str = Form("image"),
|
|
current_user: User = Depends(get_current_user),
|
|
db: Session = Depends(get_db),
|
|
):
|
|
return success_response(AssetsService(db).save_asset(current_user.id, file, mediaType))
|
|
|
|
|
|
@router.get("")
|
|
def list_assets(
|
|
current_user: User = Depends(get_current_user),
|
|
db: Session = Depends(get_db),
|
|
):
|
|
return success_response(AssetsService(db).list_assets(current_user.id))
|
|
|
|
|
|
@router.delete("/{asset_id}")
|
|
def delete_asset(
|
|
asset_id: int,
|
|
current_user: User = Depends(get_current_user),
|
|
db: Session = Depends(get_db),
|
|
):
|
|
return success_response(AssetsService(db).delete_asset(current_user.id, asset_id))
|
|
|