Resolve "evaluate uv"

2025-04-21 17:39:20 +02:00
parent bb049441b4
commit 934ef826c9
366 changed files with 849 additions and 160 deletions
@@ -0,0 +1,100 @@
+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.schema import MediaFile, SessionDep
+
+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