fix: Response model (#41)

* feat: Filter response fields

* changes

* changes

* import fix

* changes

.github/workflows/ci.yml

@@ -38,7 +38,7 @@ jobs:
       - uses: actions/checkout@v4
       - uses: actions/setup-python@v5
         with:
-          python-version: '3.12'
+          python-version: '3.13'
       - run: python -m pip install --upgrade pip
       - run: |
           python -m pip install -r requirements.txt
@@ -93,7 +93,7 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v5
         with:
-          python-version: '3.12'
+          python-version: '3.13'
 
       - name: Install dependencies
         run: |

creyPY/fastapi/crud.py

@@ -1,4 +1,4 @@
-from typing import Type, TypeVar, overload
+from typing import Type, TypeVar, overload, List
 from uuid import UUID
 
 from fastapi import HTTPException
@@ -19,13 +19,19 @@ async def get_object_or_404(
     db: AsyncSession,
     expunge: bool = False,
     lookup_column: str = "id",
+    response_fields: List[str] = [],
 ) -> T:
     pass
 
 
 @overload
 def get_object_or_404(
-    db_class: Type[T], id: UUID | str, db: Session, expunge: bool = False, lookup_column: str = "id"
+    db_class: Type[T],
+    id: UUID | str,
+    db: Session,
+    expunge: bool = False,
+    lookup_column: str = "id",
+    response_fields: List[str] = [],
 ) -> T:
     pass
 
@@ -36,10 +42,17 @@ def get_object_or_404(
     db: Session | AsyncSession,
     expunge: bool = False,
     lookup_column: str = "id",
+    response_fields: List[str] = [],
 ) -> T:
 
     async def _get_async_object() -> T:
-        query = select(db_class).filter(getattr(db_class, lookup_column) == id)
+        if response_fields:
+            selected_columns = [
+                getattr(db_class, field) for field in response_fields if hasattr(db_class, field)
+            ]
+            query = select(*selected_columns).select_from(db_class)
+        else:
+            query = select(db_class).filter(getattr(db_class, lookup_column) == id)
         result = await db.execute(query)
         obj = result.scalar_one_or_none()
         if obj is None:
@@ -49,7 +62,14 @@ def get_object_or_404(
         return obj
 
     def _get_sync_object() -> T:
-        obj = db.query(db_class).filter(getattr(db_class, lookup_column) == id).one_or_none()
+        if response_fields:
+            selected_columns = [
+                getattr(db_class, field) for field in response_fields if hasattr(db_class, field)
+            ]
+            query = db.query(*selected_columns).filter(getattr(db_class, lookup_column) == id)
+        else:
+            query = db.query(db_class).filter(getattr(db_class, lookup_column) == id)
+        obj = query.one_or_none()
         if obj is None:
             raise HTTPException(status_code=404, detail="The object does not exist.")  # type: ignore
         if expunge:

creyPY/fastapi/schemas/__init__.py

@@ -1 +1,2 @@
 from .base import *  # noqa
+from .response_schema import * #noqa

creyPY/fastapi/schemas/response_schema.py

@@ -0,0 +1,40 @@
+from typing import List, Optional, Type
+from pydantic import BaseModel, create_model
+from fastapi import Query
+
+
+class ResponseModelDependency:
+    def __init__(self, model_class: Type[BaseModel]):
+        self.model_class = model_class
+
+    def __call__(self, response_fields: Optional[List[str]] = Query(None)) -> Type[BaseModel]:
+        def process_result(result, fields=None):
+            if not fields:
+                return result
+
+            if hasattr(result, "_fields"):
+                row_fields = result._fields
+                return dict(zip(row_fields, result))
+            elif isinstance(result, tuple):
+                return dict(zip(fields, result))
+            elif isinstance(result, dict):
+                return result
+            else:
+                return {field: getattr(result, field) for field in fields if hasattr(result, field)}
+
+        if not response_fields:
+            return self.model_class, None, process_result
+
+        all_annotations = {}
+        for cls in self.model_class.__mro__:
+            if hasattr(cls, "__annotations__"):
+                all_annotations.update(cls.__annotations__)
+
+        fields = {}
+        for field in response_fields:
+            if field in all_annotations:
+                fields[field] = (all_annotations[field], None)
+
+        dynamic_model = create_model(f"Dynamic{self.model_class.__name__}", **fields)
+
+        return dynamic_model, response_fields, process_result

creyPY/helpers.py

@@ -1,5 +1,7 @@
 import secrets
 import string
+import csv
+from pathlib import Path
 
 
 def create_random_password(length: int = 12) -> str:
@@ -14,3 +16,11 @@ def create_random_password(length: int = 12) -> str:
     password += [secrets.choice(all_characters) for _ in range(length - 4)]
     secrets.SystemRandom().shuffle(password)
     return "".join(password)
+
+
+def data_to_csv(file: Path, data: list) -> None:
+
+    with file.open(mode="w", newline="", encoding="utf-8") as f:
+        writer = csv.DictWriter(f, fieldnames=data[0].keys(), delimiter=";")
+        writer.writeheader()
+        writer.writerows(data)

creyPY/services/auth0/utils.py

@@ -120,6 +120,16 @@ def create_user_invite(email: str, company_id: str) -> dict:
     return re.json()
 
 
+def delete_user_invite(user_id: str) -> None:
+    re = requests.delete(
+        f"https://{AUTH0_DOMAIN}/api/v2/users/{user_id}",
+        headers={"Authorization": f"Bearer {get_management_token()}"},
+        timeout=5,
+    )
+    if re.status_code != 204:
+        raise HTTPException(re.status_code, re.json())
+
+
 def password_change_mail(email: str) -> bool:
     re = requests.post(
         f"https://{AUTH0_DOMAIN}/dbconnections/change_password",

requirements.stripe.txt

@@ -1 +1 @@
-stripe==11.5.0 # Stripe
+stripe==11.6.0 # Stripe