telegram_files_server/fastapi_file_server/app/views.py

from fastapi import File, UploadFile, Depends
from starlette import status
from fastapi import APIRouter, HTTPException

from app.models import UploadedFile as UploadedFileDB
from app.serializers import UploadedFile, CreateUploadedFile
from app.services.file_uploader import FileUploader
from app.depends import check_token


router = APIRouter(
    prefix="/api/v1/files",
    dependencies=[Depends(check_token)],
    tags=["files"]
)


@router.get("/", response_model=list[UploadedFile])
async def get_files():
    return await UploadedFileDB.objects.all()


@router.get("/{file_id}", response_model=UploadedFile, responses={
    404: {},
})
async def get_file(file_id: int):
    uploaded_file = await UploadedFileDB.objects.get_or_none(id=file_id)

    if not uploaded_file:
        raise HTTPException(status.HTTP_404_NOT_FOUND)

    return uploaded_file


@router.post("/", response_model=UploadedFile)
async def create_file(data: CreateUploadedFile):
    return await UploadedFileDB.objects.create(
        **data.dict()
    )


@router.post("/upload/", response_model=UploadedFile)
async def upload_file(file: UploadFile = File({})):
    return await FileUploader.upload(file)


@router.delete("/{file_id}", response_model=UploadedFile, responses={
    400: {}
})
async def delete_file(file_id: int):
    uploaded_file = await UploadedFileDB.objects.get_or_none(id=file_id)

    if not uploaded_file:
        raise HTTPException(status.HTTP_400_BAD_REQUEST)

    await uploaded_file.delete()
    return uploaded_file