mirror of
https://github.com/rommapp/romm.git
synced 2026-02-19 07:50:57 +01:00
72 lines
2.4 KiB
Python
72 lines
2.4 KiB
Python
from decorators.auth import protected_route
|
|
from endpoints.responses.assets import ScreenshotSchema, UploadedScreenshotsResponse
|
|
from exceptions.endpoint_exceptions import RomNotFoundInDatabaseException
|
|
from fastapi import File, Request, UploadFile
|
|
from handler.auth.base_handler import Scope
|
|
from handler.database import db_rom_handler, db_screenshot_handler
|
|
from handler.filesystem import fs_asset_handler
|
|
from handler.scan_handler import scan_screenshot
|
|
from logger.logger import log
|
|
from utils.router import APIRouter
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@protected_route(router.post, "/screenshots", [Scope.ASSETS_WRITE])
|
|
def add_screenshots(
|
|
request: Request,
|
|
rom_id: int,
|
|
screenshots: list[UploadFile] = File(...), # noqa: B008
|
|
) -> UploadedScreenshotsResponse:
|
|
rom = db_rom_handler.get_rom(rom_id)
|
|
if not rom:
|
|
raise RomNotFoundInDatabaseException(rom_id)
|
|
|
|
current_user = request.user
|
|
log.info(f"Uploading screenshots to {rom.name}")
|
|
|
|
screenshots_path = fs_asset_handler.build_screenshots_file_path(
|
|
user=request.user, platform_fs_slug=rom.platform_slug
|
|
)
|
|
|
|
for screenshot in screenshots:
|
|
if not screenshot.filename:
|
|
log.warning("Skipping empty screenshot")
|
|
continue
|
|
|
|
fs_asset_handler.write_file(file=screenshot, path=screenshots_path)
|
|
|
|
# Scan or update screenshot
|
|
scanned_screenshot = scan_screenshot(
|
|
file_name=screenshot.filename,
|
|
user=request.user,
|
|
platform_fs_slug=rom.platform_slug,
|
|
)
|
|
db_screenshot = db_screenshot_handler.get_screenshot_by_filename(
|
|
rom_id=rom.id, user_id=current_user.id, file_name=screenshot.filename
|
|
)
|
|
if db_screenshot:
|
|
db_screenshot_handler.update_screenshot(
|
|
db_screenshot.id,
|
|
{"file_size_bytes": scanned_screenshot.file_size_bytes},
|
|
)
|
|
continue
|
|
|
|
scanned_screenshot.rom_id = rom.id
|
|
scanned_screenshot.user_id = current_user.id
|
|
db_screenshot_handler.add_screenshot(scanned_screenshot)
|
|
|
|
rom = db_rom_handler.get_rom(rom_id)
|
|
if not rom:
|
|
raise RomNotFoundInDatabaseException(rom_id)
|
|
|
|
return {
|
|
"uploaded": len(screenshots),
|
|
"screenshots": [
|
|
ScreenshotSchema.model_validate(s)
|
|
for s in rom.screenshots
|
|
if s.user_id == current_user.id
|
|
],
|
|
"merged_screenshots": rom.merged_screenshots,
|
|
}
|