rommapp/romm
1
0
Fork 0
mirror of https://github.com/rommapp/romm.git synced 2026-02-19 07:50:57 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix tests

Browse Source
This commit is contained in:
Georges-Antoine Assi
2024-05-19 13:06:11 -04:00
parent 6735be4ce8
commit fffc954750
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
backend/handler/database/platforms_handler.py
View File

@@ -29,17 +29,17 @@ class DBPlatformsHandler(DBBaseHandler):
@with_query
def add_platform(
self, platform: Platform, query: Query = None, session: Session = None
):
) -> Platform | None:
session.merge(platform)
session.flush()
return query.filter(Platform.id == platform.id).one()
return query.filter(Platform.id == platform.id).first()
@begin_session
@with_query
def get_platforms(
self, id: int = None, query: Query = None, session: Session = None
):
) -> list[Platform] | Platform | None:
return (
query.get(Platform, id)
if id
@@ -54,13 +54,13 @@ class DBPlatformsHandler(DBBaseHandler):
@with_query
def get_platform_by_fs_slug(
self, fs_slug: str, query: Query = None, session: Session = None
):
) -> Platform | None:
return session.scalars(
query.filter_by(fs_slug=fs_slug).limit(1)
).first()
@begin_session
def delete_platform(self, id: int, session: Session = None):
def delete_platform(self, id: int, session: Session = None) -> int:
# Remove all roms from that platforms first
session.execute(
delete(Rom)
@@ -76,7 +76,7 @@ class DBPlatformsHandler(DBBaseHandler):
@begin_session
def purge_platforms(
self, fs_platforms: list[str], session: Session = None
):
) -> int:
return session.execute(
delete(Platform)
.where(or_(Platform.fs_slug.not_in(fs_platforms), Platform.slug.is_(None)))