mirror of
https://github.com/rommapp/romm.git
synced 2026-02-19 07:50:57 +01:00
8281d5679b
The current implementation for some of the database handlers, where the same method is used to retrieve either a single entity (when an `id` is passed), a list of entities, or `None`, makes the typing and overall design more complex. This change simplifies database handlers, by having two separate methods where appropiate: * A method that receives an `id`, and returns either an entity, or `None`. * A method that optionally receives filters, and returns (depending on the current handler implementation) a list of entities, or a `Select` object that allows chaining more SQLAlchemy operations.
77 lines
2.2 KiB
Python
77 lines
2.2 KiB
Python
from decorators.database import begin_session
|
|
from models.firmware import Firmware
|
|
from sqlalchemy import and_, delete, select, update
|
|
from sqlalchemy.orm import Session
|
|
|
|
from .base_handler import DBBaseHandler
|
|
|
|
|
|
class DBFirmwareHandler(DBBaseHandler):
|
|
@begin_session
|
|
def add_firmware(self, firmware: Firmware, session: Session = None) -> Firmware:
|
|
return session.merge(firmware)
|
|
|
|
@begin_session
|
|
def get_firmware(
|
|
self,
|
|
id: int,
|
|
*,
|
|
session: Session = None,
|
|
) -> Firmware | None:
|
|
return session.scalar(select(Firmware).filter_by(id=id).limit(1))
|
|
|
|
@begin_session
|
|
def list_firmware(
|
|
self,
|
|
*,
|
|
platform_id: int | None = None,
|
|
session: Session = None,
|
|
) -> list[Firmware]:
|
|
return session.scalars(
|
|
select(Firmware)
|
|
.filter_by(platform_id=platform_id)
|
|
.order_by(Firmware.file_name.asc())
|
|
).all()
|
|
|
|
@begin_session
|
|
def get_firmware_by_filename(
|
|
self, platform_id: int, file_name: str, session: Session = None
|
|
):
|
|
return session.scalar(
|
|
select(Firmware)
|
|
.filter_by(platform_id=platform_id, file_name=file_name)
|
|
.limit(1)
|
|
)
|
|
|
|
@begin_session
|
|
def update_firmware(self, id: int, data: dict, session: Session = None) -> Firmware:
|
|
return session.execute(
|
|
update(Firmware)
|
|
.where(Firmware.id == id)
|
|
.values(**data)
|
|
.execution_options(synchronize_session="evaluate")
|
|
)
|
|
|
|
@begin_session
|
|
def delete_firmware(self, id: int, session: Session = None) -> None:
|
|
return session.execute(
|
|
delete(Firmware)
|
|
.where(Firmware.id == id)
|
|
.execution_options(synchronize_session="evaluate")
|
|
)
|
|
|
|
@begin_session
|
|
def purge_firmware(
|
|
self, platform_id: int, firmware: list[str], session: Session = None
|
|
) -> None:
|
|
return session.execute(
|
|
delete(Firmware)
|
|
.where(
|
|
and_(
|
|
Firmware.platform_id == platform_id,
|
|
Firmware.file_name.not_in(firmware),
|
|
)
|
|
)
|
|
.execution_options(synchronize_session="evaluate")
|
|
)
|