contact working

app/utils/__init__.py

@@ -0,0 +1,50 @@
+
+from typing import Annotated
+
+from fastapi import Path, HTTPException
+from sqlalchemy import select
+
+
+def make_primary_annotation(name=str):
+    return Annotated[str, Path(description=f'{name}, either id (int) or name')]
+
+
+def get_single_record(session, cls, primary: str):
+    print(session)
+    print(cls)
+    print(primary)
+    result = session.get(cls, primary)
+    print(result)
+    if result is None:
+        result = session.scalar(select(cls).where(cls.name == primary))
+    if result is None:
+        raise HTTPException(
+            status_code=404, 
+            detail=f"{cls.__class__.__name__} {primary!r} not found.")
+    return result
+
+
+def set_item_status(session, current_user, item, status: str):
+    item.status_id = status
+    item._user__id = current_user.id
+    session.commit()
+    session.refresh(item)
+    return item
+
+
+def update_item(session, current_user, item, data):
+    for k, v in data.model_dump(exclude_unset=True).items():
+        setattr(item, k, v)
+    item._user__id = current_user.id
+    session.commit()
+    session.refresh(item)
+    return item
+
+
+def create_item(session, cls, current_user, data):
+    item = cls(**data.model_dump())
+    item._user__id = current_user.id
+    session.add(item)
+    session.commit()
+    session.refresh(item)
+    return item