Thomas Peetz
102 lines
4.1 KiB
Python
102 lines
4.1 KiB
Python
from datetime import datetime
|
|
from typing import List
|
|
from uuid import uuid4, UUID
|
|
|
|
from fastapi import APIRouter, status, HTTPException
|
|
from sqlalchemy import select, Sequence
|
|
|
|
from src.models.media.file import MediaFileResponse, Link, get_file_details
|
|
from src.routers import SessionDep
|
|
from src.schema.media import MediaFile
|
|
|
|
router = APIRouter(
|
|
prefix="/media",
|
|
tags=["media"]
|
|
)
|
|
|
|
@router.get("/update-titles")
|
|
def update_titles(db: SessionDep) -> list[MediaFileResponse]:
|
|
results: list[MediaFileResponse] = []
|
|
files = db.query(MediaFile).filter(MediaFile.review == 1).all()
|
|
for mediafile in files:
|
|
mediafile.update_title()
|
|
db.add(mediafile)
|
|
response = MediaFileResponse(id=mediafile.id,
|
|
title=mediafile.title,
|
|
file_name=mediafile.file_name,
|
|
cloud_link=mediafile.cloud_link,
|
|
url=str(mediafile.url),
|
|
review=mediafile.review,
|
|
should_download=mediafile.should_download)
|
|
results.append(response)
|
|
db.commit()
|
|
return results
|
|
|
|
|
|
@router.get("/files", response_model=List[MediaFileResponse])
|
|
def get_all_files(db: SessionDep, review: bool = False, download: bool = False) -> List[MediaFileResponse]:
|
|
results: list[MediaFileResponse] = []
|
|
files: Sequence[MediaFile]
|
|
if review:
|
|
files = db.query(MediaFile).filter(MediaFile.review == review).all()
|
|
elif download:
|
|
files = db.query(MediaFile).filter(MediaFile.should_download == download).all()
|
|
else:
|
|
files = db.scalars(select(MediaFile)).all()
|
|
for mediafile in files:
|
|
response = MediaFileResponse(id=mediafile.id,
|
|
title=mediafile.title,
|
|
file_name=mediafile.file_name,
|
|
cloud_link=mediafile.cloud_link,
|
|
url=str(mediafile.url),
|
|
review=mediafile.review,
|
|
should_download=mediafile.should_download)
|
|
results.append(response)
|
|
return results
|
|
|
|
@router.get("/files/{file_id}", response_model=MediaFileResponse)
|
|
def get_file(file_id: UUID, db: SessionDep) -> MediaFileResponse:
|
|
mediafile = db.get(MediaFile, file_id)
|
|
if not mediafile:
|
|
raise HTTPException(status_code=404, detail="MediaFile could not be found")
|
|
response = get_file_details(mediafile)
|
|
return response
|
|
|
|
@router.put("/files/{file_id}", response_model=MediaFileResponse)
|
|
def update_file(file_id: UUID, db: SessionDep, info: MediaFileResponse) -> MediaFileResponse:
|
|
mediaFile = db.get(MediaFile, file_id)
|
|
if not mediaFile:
|
|
raise HTTPException(status_code=404, detail="MediaFile could not be found")
|
|
mediaFile.file_name = info.file_name
|
|
mediaFile.cloud_link = info.cloud_link
|
|
mediaFile.url = info.url
|
|
mediaFile.title = info.title
|
|
mediaFile.last_modified_date = datetime.now()
|
|
mediaFile.review = info.review
|
|
mediaFile.should_download = info.should_download
|
|
db.add(mediaFile)
|
|
db.commit()
|
|
return info
|
|
|
|
|
|
@router.post("/files", status_code=status.HTTP_201_CREATED)
|
|
def add_file(new_link: Link, db: SessionDep) -> MediaFileResponse:
|
|
print(new_link.url)
|
|
try:
|
|
mediaFile: MediaFile = MediaFile()
|
|
setattr(mediaFile, "url", new_link.url)
|
|
setattr(mediaFile, "review", True)
|
|
setattr(mediaFile, "should_download", True)
|
|
db.add(mediaFile)
|
|
db.commit()
|
|
except:
|
|
raise HTTPException(status_code=409, detail="Link duplicate")
|
|
response = MediaFileResponse(id=uuid4(),
|
|
title=mediaFile.title,
|
|
file_name=mediaFile.file_name,
|
|
cloud_link=mediaFile.cloud_link,
|
|
url=new_link.url,
|
|
review=mediaFile.review,
|
|
should_download=mediaFile.should_download)
|
|
return response
|