mirror of
https://github.com/flibusta-apps/telegram_files_server.git
synced 2025-12-06 12:35:39 +01:00
58 lines
1.5 KiB
Python
58 lines
1.5 KiB
Python
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
|