romm/backend/handler/db_handler/db_states_handler.py

from decorators.database import begin_session
from handler.db_handler import DBHandler
from models.assets import State
from sqlalchemy import and_, delete, select, update
from sqlalchemy.orm import Session


class DBStatesHandler(DBHandler):
    @begin_session
    def add_state(self, state: State, session: Session = None):
        return session.merge(state)

    @begin_session
    def get_state(self, id: int, session: Session = None):
        return session.get(State, id)

    @begin_session
    def get_state_by_filename(
        self, rom_id: str, file_name: str, session: Session = None
    ):
        return session.scalars(
            select(State).filter_by(rom_id=rom_id, file_name=file_name).limit(1)
        ).first()

    @begin_session
    def update_state(self, id: int, data: dict, session: Session = None):
        session.execute(
            update(State)
            .where(State.id == id)
            .values(**data)
            .execution_options(synchronize_session="evaluate")
        )

    @begin_session
    def delete_state(self, id: int, session: Session = None):
        return session.execute(
            delete(State)
            .where(State.id == id)
            .execution_options(synchronize_session="evaluate")
        )

    @begin_session
    def purge_states(self, rom_id: int, states: list[str], session: Session = None):
        return session.execute(
            delete(State)
            .where(and_(State.rom_id == rom_id, State.file_name.not_in(states)))
            .execution_options(synchronize_session="evaluate")
        )