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

fix from_orm_with_request

Browse Source
This commit is contained in:
Georges-Antoine Assi
2024-02-08 13:11:36 -05:00
parent 0f3a2f5705
commit 01c3054e76
1 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
backend/endpoints/responses/rom.py
View File

@@ -99,13 +99,19 @@ class RomSchema(BaseModel):
rom = cls.model_validate(db_rom)
user_id = request.user.id
rom.sibling_roms = db_rom.get_sibling_roms()
rom.user_saves = [save for save in db_rom.saves if save.user_id == user_id]
rom.user_states = [state for state in db_rom.states if state.user_id == user_id]
rom.sibling_roms = [
RomSchema.model_validate(r) for r in db_rom.get_sibling_roms()
]
rom.user_saves = [
SaveSchema.model_validate(s) for s in db_rom.saves if s.user_id == user_id
]
rom.user_states = [
StateSchema.model_validate(s) for s in db_rom.states if s.user_id == user_id
]
rom.user_screenshots = [
screenshot
for screenshot in db_rom.screenshots
if screenshot.user_id == user_id
ScreenshotSchema.model_validate(s)
for s in db_rom.screenshots
if s.user_id == user_id
]
return rom