diff --git a/backend/handler/database/platforms_handler.py b/backend/handler/database/platforms_handler.py index 9f0cf378e..d198b1011 100644 --- a/backend/handler/database/platforms_handler.py +++ b/backend/handler/database/platforms_handler.py @@ -30,7 +30,7 @@ class DBPlatformsHandler(DBBaseHandler): def add_platform( self, platform: Platform, query: Query = None, session: Session = None ) -> Platform | None: - session.merge(platform) + platform = session.merge(platform) session.flush() return session.scalar(query.filter_by(id=platform.id).limit(1)) diff --git a/backend/handler/database/roms_handler.py b/backend/handler/database/roms_handler.py index 7acfd65dd..5e2c27400 100644 --- a/backend/handler/database/roms_handler.py +++ b/backend/handler/database/roms_handler.py @@ -53,8 +53,8 @@ def _order(self, data: Select[Rom], order_by: str, order_dir: str): @begin_session @with_assets - def add_rom(self, rom: Rom, query: Query = None, session: Session = None): - session.merge(rom) + def add_rom(self, rom: Rom, query: Query = None, session: Session = None) -> Rom: + rom = session.merge(rom) session.flush() return session.scalar(query.filter_by(id=rom.id).limit(1)) @@ -70,7 +70,7 @@ def get_roms( order_dir: str = "asc", query: Query = None, session: Session = None, - ): + ) -> list[Rom] | Rom | None: return ( session.scalar(query.filter_by(id=id).limit(1)) if id @@ -89,7 +89,7 @@ def get_rom_by_filename( file_name: str, query: Query = None, session: Session = None, - ): + ) -> Rom | None: return session.scalar( query.filter_by(platform_id=platform_id, file_name=file_name).limit(1) ) @@ -98,7 +98,7 @@ def get_rom_by_filename( @with_assets def get_rom_by_filename_no_tags( self, file_name_no_tags: str, query: Query = None, session: Session = None - ): + ) -> Rom | None: return session.scalar( query.filter_by(file_name_no_tags=file_name_no_tags).limit(1) ) @@ -107,13 +107,13 @@ def get_rom_by_filename_no_tags( @with_assets def get_rom_by_filename_no_ext( self, file_name_no_ext: str, query: Query = None, session: Session = None - ): + ) -> Rom | None: return session.scalar( query.filter_by(file_name_no_ext=file_name_no_ext).limit(1) ) @begin_session - def update_rom(self, id: int, data: dict, session: Session = None): + def update_rom(self, id: int, data: dict, session: Session = None) -> Rom: return session.execute( update(Rom) .where(Rom.id == id) @@ -122,7 +122,7 @@ def update_rom(self, id: int, data: dict, session: Session = None): ) @begin_session - def delete_rom(self, id: int, session: Session = None): + def delete_rom(self, id: int, session: Session = None) -> Rom: return session.execute( delete(Rom) .where(Rom.id == id) @@ -130,7 +130,9 @@ def delete_rom(self, id: int, session: Session = None): ) @begin_session - def purge_roms(self, platform_id: int, roms: list[str], session: Session = None): + def purge_roms( + self, platform_id: int, roms: list[str], session: Session = None + ) -> int: return session.execute( delete(Rom) .where(and_(Rom.platform_id == platform_id, Rom.file_name.not_in(roms))) @@ -138,17 +140,21 @@ def purge_roms(self, platform_id: int, roms: list[str], session: Session = None) ) @begin_session - def get_rom_note(self, rom_id: int, user_id: int, session: Session = None): - return session.scalars( + def get_rom_note( + self, rom_id: int, user_id: int, session: Session = None + ) -> RomNote | None: + return session.scalar( select(RomNote).filter_by(rom_id=rom_id, user_id=user_id).limit(1) - ).first() + ) @begin_session - def add_rom_note(self, rom_id: int, user_id: int, session: Session = None): + def add_rom_note( + self, rom_id: int, user_id: int, session: Session = None + ) -> RomNote: return session.merge(RomNote(rom_id=rom_id, user_id=user_id)) @begin_session - def update_rom_note(self, id: int, data: dict, session: Session = None): + def update_rom_note(self, id: int, data: dict, session: Session = None) -> RomNote: return session.execute( update(RomNote) .where(RomNote.id == id) diff --git a/backend/logger/logger.py b/backend/logger/logger.py index 6280e4b2d..53adae87d 100644 --- a/backend/logger/logger.py +++ b/backend/logger/logger.py @@ -8,14 +8,14 @@ log.setLevel(logging.DEBUG) # Set up sqlachemy logger -sql_log = logging.getLogger("sqlalchemy.engine") -sql_log.setLevel(logging.DEBUG) +# sql_log = logging.getLogger("sqlalchemy.engine") +# sql_log.setLevel(logging.DEBUG) # Define stdout handler stdout_handler = logging.StreamHandler(sys.stdout) stdout_handler.setFormatter(StdoutFormatter()) log.addHandler(stdout_handler) -sql_log.addHandler(stdout_handler) +# sql_log.addHandler(stdout_handler) # Hush passlib warnings logging.getLogger("passlib").setLevel(logging.ERROR) diff --git a/frontend/src/App.vue b/frontend/src/App.vue index 7f3e6bd9c..7457c7439 100644 --- a/frontend/src/App.vue +++ b/frontend/src/App.vue @@ -44,7 +44,7 @@ socket.on( socket.on("scan:scanning_rom", (rom: Rom) => { scanningStore.set(true); - if (romsStore.platform.name === rom.platform_name) { + if (romsStore.platformID === rom.platform_id) { romsStore.add([rom]); romsStore.setFiltered( isFiltered ? romsStore.filteredRoms : romsStore.allRoms, diff --git a/frontend/src/components/Gallery/AppBar/AdminMenu.vue b/frontend/src/components/Gallery/AppBar/AdminMenu.vue index c1bda8842..449f45d3c 100644 --- a/frontend/src/components/Gallery/AppBar/AdminMenu.vue +++ b/frontend/src/components/Gallery/AppBar/AdminMenu.vue @@ -7,8 +7,8 @@ import DeleteBtn from "@/components/Gallery/AppBar/DeleteBtn.vue";