romm/backend/handler/db_handler/db_platforms_handler.py

from decorators.database import begin_session
from handler.db_handler import DBHandler
from models.platform import Platform
from models.rom import Rom
from sqlalchemy import delete, func, or_, select
from sqlalchemy.orm import Session


class DBPlatformsHandler(DBHandler):
    @begin_session
    def add_platform(self, platform: Platform, session: Session = None):
        return session.merge(platform)

    @begin_session
    def get_platforms(self, id: int = None, session: Session = None):
        return (
            session.get(Platform, id)
            if id
            else (
                session.scalars(select(Platform).order_by(Platform.name.asc()))
                .unique()
                .all()
            )
        )

    @begin_session
    def get_platform_by_slug(self, fs_slug: str, session: Session = None):
        return session.scalars(
            select(Platform).filter_by(fs_slug=fs_slug).limit(1)
        ).first()

    @begin_session
    def delete_platform(self, id: int, session: Session = None):
        # Remove all roms from that platforms first
        session.execute(
            delete(Rom)
            .where(Rom.platform_id == id)
            .execution_options(synchronize_session="evaluate")
        )
        return session.execute(
            delete(Platform)
            .where(Platform.id == id)
            .execution_options(synchronize_session="evaluate")
        )

    @begin_session
    def get_rom_count(self, platform_id: int, session: Session = None):
        return session.scalar(
            select(func.count()).select_from(Rom).filter_by(platform_id=platform_id)
        )

    @begin_session
    def purge_platforms(self, fs_platforms: list[str], session: Session = None):
        return session.execute(
            delete(Platform)
            .where(or_(Platform.fs_slug.not_in(fs_platforms), Platform.slug.is_(None)))
            .where(
                select(func.count())
                .select_from(Rom)
                .filter_by(platform_id=Platform.id)
                .as_scalar()
                == 0
            )
            .execution_options(synchronize_session="fetch")
        )