mirror of
https://github.com/creyD/creyPY.git
synced 2026-04-12 19:30:30 +02:00
fix: get_object alter for async response filter (#44)
This commit is contained in:
@@ -50,16 +50,30 @@ def get_object_or_404(
|
|||||||
selected_columns = [
|
selected_columns = [
|
||||||
getattr(db_class, field) for field in response_fields if hasattr(db_class, field)
|
getattr(db_class, field) for field in response_fields if hasattr(db_class, field)
|
||||||
]
|
]
|
||||||
query = select(*selected_columns).select_from(db_class)
|
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}
|
||||||
else:
|
else:
|
||||||
query = select(db_class).filter(getattr(db_class, lookup_column) == id)
|
query = select(db_class).where(getattr(db_class, lookup_column) == id)
|
||||||
result = await db.execute(query)
|
result = await db.execute(query)
|
||||||
obj = result.scalar_one_or_none()
|
row = result.scalar_one_or_none()
|
||||||
if obj is None:
|
|
||||||
raise HTTPException(status_code=404, detail="The object does not exist.") # type: ignore
|
if row is None:
|
||||||
|
raise HTTPException(status_code=404, detail="The object does not exist.")
|
||||||
|
|
||||||
|
obj_dict = {k: v for k, v in row.__dict__.items()
|
||||||
|
if not k.startswith('_')}
|
||||||
if expunge:
|
if expunge:
|
||||||
await db.expunge(obj)
|
await db.expunge(obj_dict)
|
||||||
return obj
|
return obj_dict
|
||||||
|
|
||||||
def _get_sync_object() -> T:
|
def _get_sync_object() -> T:
|
||||||
if response_fields:
|
if response_fields:
|
||||||
|
|||||||
Reference in New Issue
Block a user