mirror of
https://github.com/rommapp/romm.git
synced 2026-02-18 00:27:41 +01:00
62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
from decorators.database import begin_session
|
|
from models.assets import Screenshot
|
|
from sqlalchemy import delete, select, update
|
|
from sqlalchemy.orm import Session
|
|
|
|
from .base_handler import DBBaseHandler
|
|
|
|
|
|
class DBScreenshotsHandler(DBBaseHandler):
|
|
@begin_session
|
|
def add_screenshot(
|
|
self, screenshot: Screenshot, session: Session = None
|
|
) -> Screenshot:
|
|
return session.merge(screenshot)
|
|
|
|
@begin_session
|
|
def get_screenshot(self, id, session: Session = None) -> Screenshot:
|
|
return session.get(Screenshot, id)
|
|
|
|
@begin_session
|
|
def get_screenshot_by_filename(
|
|
self, rom_id: int, user_id: int, file_name: str, session: Session = None
|
|
) -> Screenshot | None:
|
|
return session.scalars(
|
|
select(Screenshot)
|
|
.filter_by(rom_id=rom_id, user_id=user_id, file_name=file_name)
|
|
.limit(1)
|
|
).first()
|
|
|
|
@begin_session
|
|
def update_screenshot(
|
|
self, id: int, data: dict, session: Session = None
|
|
) -> Screenshot:
|
|
return session.execute(
|
|
update(Screenshot)
|
|
.where(Screenshot.id == id)
|
|
.values(**data)
|
|
.execution_options(synchronize_session="evaluate")
|
|
)
|
|
|
|
@begin_session
|
|
def delete_screenshot(self, id: int, session: Session = None) -> None:
|
|
return session.execute(
|
|
delete(Screenshot)
|
|
.where(Screenshot.id == id)
|
|
.execution_options(synchronize_session="evaluate")
|
|
)
|
|
|
|
@begin_session
|
|
def purge_screenshots(
|
|
self, rom_id: int, user_id: int, screenshots: list[str], session: Session = None
|
|
) -> None:
|
|
return session.execute(
|
|
delete(Screenshot)
|
|
.where(
|
|
Screenshot.rom_id == rom_id,
|
|
Screenshot.user_id == user_id,
|
|
Screenshot.file_name.not_in(screenshots),
|
|
)
|
|
.execution_options(synchronize_session="evaluate")
|
|
)
|