From 284317d2a8de010bb0fb3d1537fd91639d272d54 Mon Sep 17 00:00:00 2001 From: Pierre <397503+bemble@users.noreply.github.com> Date: Fri, 17 May 2024 14:54:21 +0200 Subject: [PATCH] fix(server): fix get model --- server/holerr/api/routers/downloads.py | 2 +- server/holerr/database/repositories.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/server/holerr/api/routers/downloads.py b/server/holerr/api/routers/downloads.py index 0563146..3662588 100644 --- a/server/holerr/api/routers/downloads.py +++ b/server/holerr/api/routers/downloads.py @@ -17,7 +17,7 @@ async def list_downloads(): @router.delete("/{download_id}", response_model=Download, tags=["Downloads"]) async def delete_download(download_id: str): session = db.new_session() - download = DownloadRepository(session).get_model(download_id) + download = DownloadRepository(session).get_model(download_id.encode('UTF-8')) if download is None: raise HTTPException(status_code=404, detail=f"Download {download_id} not found") download.to_delete = True diff --git a/server/holerr/database/repositories.py b/server/holerr/database/repositories.py index b0e6e51..7878a72 100644 --- a/server/holerr/database/repositories.py +++ b/server/holerr/database/repositories.py @@ -24,8 +24,8 @@ def __init__(self, session: Session, entity: Base): self.entity = entity def get_model(self, id: any) -> Base | None: - res = self.session.scalars(select(self.entity).where(self.entity.id == id)) - return res.one_or_none() + res = self.get_all_models(self.entity.id == id) + return res[0] if len(res) > 0 else None def get_all_models(self, conditions=True, options=None) -> list[Base]: query = select(self.entity).where(conditions)