diff --git a/creyPY/fastapi/crud.py b/creyPY/fastapi/crud.py index 1f43a7f..b752797 100644 --- a/creyPY/fastapi/crud.py +++ b/creyPY/fastapi/crud.py @@ -54,14 +54,13 @@ def get_object_or_404( query = select(*selected_columns).where(getattr(db_class, lookup_column) == id) result = await db.execute(query) row = result.first() - + if row is None: raise HTTPException(status_code=404, detail="The object does not exist.") if hasattr(row, "_mapping"): obj_dict = dict(row._mapping) else: - obj_dict = {column.key: getattr(row, column.key) - for column in selected_columns} + obj_dict = {column.key: getattr(row, column.key) for column in selected_columns} else: query = select(db_class).where(getattr(db_class, lookup_column) == id) result = await db.execute(query)