bootgraph 1.9.0.dev24128__tar.gz → 1.9.0.dev24129__tar.gz

{bootgraph-1.9.0.dev24128 → bootgraph-1.9.0.dev24129}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: bootgraph
-Version: 1.9.0.dev24128
+Version: 1.9.0.dev24129
 Summary: A Python library for integrating SQLModel and Strawberry, providing a seamless GraphQL integration with FastAPI and advanced features for database interactions.
 Home-page: https://github.com/MDoreto/graphemy
 License: MIT

{bootgraph-1.9.0.dev24128 → bootgraph-1.9.0.dev24129}/bootgraph/dl.py

@@ -69,6 +69,8 @@ class Dl:
 
     source: str | list[str]
     target: str | list[str]
+    extract_from_nested_dl: str | None = None
+    return_class: str | None = None
     foreign_key: bool | None = None
 
     def __init__(
@@ -76,6 +78,8 @@ class Dl:
         source: str | list[str],
         target: str | list[str],
         foreign_key: bool = None,
+        extract_from_nested_dl: str | None = None,
+        return_class: str | None = None,
     ):
         if type(source) != type(target):
             raise "source and target must have same type"
@@ -90,6 +94,8 @@ class Dl:
         self.source = source
         self.target = target
         self.foreign_key = foreign_key
+        self.extract_from_nested_dl = extract_from_nested_dl
+        self.return_class = return_class
 
 
 class GraphemyDataLoader(DataLoader):

{bootgraph-1.9.0.dev24128 → bootgraph-1.9.0.dev24129}/bootgraph/schemas/generators.py

@@ -7,7 +7,8 @@ from typing import (
     TypeVar,
     Union,
     get_args,
-    get_origin, Type,
+    get_origin,
+    Type
 )
 
 from sqlalchemy.orm import Mapped
@@ -322,7 +323,7 @@ def get_dl_function(
 
     # Define the return type using Strawberry's lazy type resolution
     return_type = Annotated[
-        f"{class_type}", # Schema
+        f"{class_type}" if not field_value.return_class else field_value.return_class,
         strawberry.lazy("bootgraph.router"),
     ]
     if is_list:
@@ -355,7 +356,15 @@ def get_dl_function(
             if isinstance(field_value.source, list)
             else getattr(self, field_value.source)
         )
-        return await info.context[dl_name].load(source_value, {"filters": filter_args})
+        resp = await info.context[dl_name].load(source_value, {"filters": filter_args})
+        if field_value.extract_from_nested_dl:
+            if isinstance(resp, list):
+                # If the response is a list, apply getattr to each item
+                return [await getattr(item, field_value.extract_from_nested_dl)(info) for item in resp]
+            else:
+                # Single object
+                return await getattr(resp, field_value.extract_from_nested_dl)(info)
+        return resp
 
     # Customize the function attributes for introspection or other purposes
     loader_func.__name__ = field_name

{bootgraph-1.9.0.dev24128 → bootgraph-1.9.0.dev24129}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "bootgraph"
-version = "v1.9.0.dev24128"
+version = "v1.9.0.dev24129"
 description = "A Python library for integrating SQLModel and Strawberry, providing a seamless GraphQL integration with FastAPI and advanced features for database interactions."
 authors = ["Matheus Doreto <matheusdoreto.md@gmail.com>", "Pavel Mulin <mulin.pasha@gmail.com>"]
 readme = "README.md"