bootgraph 1.10.0.dev24599__tar.gz → 1.11.0.dev24752__tar.gz

{bootgraph-1.10.0.dev24599 → bootgraph-1.11.0.dev24752}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: bootgraph
-Version: 1.10.0.dev24599
+Version: 1.11.0.dev24752
 Summary: A Python library for integrating SQLModel and Strawberry, providing a seamless GraphQL integration with FastAPI and advanced features for database interactions.
 License: MIT
 Author: Matheus Doreto

{bootgraph-1.10.0.dev24599 → bootgraph-1.11.0.dev24752}/bootgraph/router.py

@@ -11,6 +11,7 @@ from fastapi import Request, Response
 from graphql.error import GraphQLError
 from graphql.error.graphql_error import format_error as format_graphql_error
 from sqlalchemy.engine.base import Engine
+from sqlalchemy.ext.asyncio import AsyncEngine
 from strawberry.fastapi import GraphQLRouter
 from strawberry.http import GraphQLHTTPResponse
 from strawberry.types import ExecutionResult
@@ -112,7 +113,7 @@ class GraphemyRouter(GraphQLRouter):
         permission_getter: Callable | None = None,
         dl_filter: Callable | None = None,
         query_filter: Callable | None = None,
-        engine: Engine | Dict[str, Engine] = None,
+        engine: AsyncEngine | Engine | Dict[str, Engine] = None,
         extensions: list = None,
         enable_queries: bool = True,
         enable_put_mutations: bool = False,
@@ -169,11 +170,16 @@ class GraphemyRouter(GraphQLRouter):
             # model_mutation_fields = {}
             if cls_mutations:
                 # Create a resolver that returns an instance of ModelMutations
-                def model_mutations_resolver() -> cls_mutations:
-                    return cls_mutations()
+                def make_mutation_resolver(mutations):
+                    def resolver() -> mutations:
+                        return mutations()
+
+                    return resolver
+
+                resolver = make_mutation_resolver(cls_mutations)
 
                 # Create a Strawberry field for this query
-                mutation_field = strawberry.field(description=f"Mutations for {cls.__queryname__}")(model_mutations_resolver)
+                mutation_field = strawberry.field(description=f"Mutations for {cls.__queryname__}")(resolver)
 
                 need_mutation = False
                 setattr(mutation, cls.__queryname__, mutation_field)
@@ -185,9 +191,15 @@ class GraphemyRouter(GraphQLRouter):
             )
 
         async def get_context(request: Request, response: Response) -> dict:
-            context = {"engine": engine}
-            if context_session:
-                context = {"session": context_session(engine)}
+            if type(engine) == AsyncEngine:
+                context = {"engine": engine}
+                if context_session:
+                    session_generator = context_session()
+                    context = {"session": await anext(session_generator)}
+            else:
+                context = {"engine": engine}
+                if context_session:
+                    context = {"session": context_session(engine)}
             if context_getter:
                 context = await context_getter(request, response, context)
             for k, (func, return_class) in functions.items():

{bootgraph-1.10.0.dev24599 → bootgraph-1.11.0.dev24752}/bootgraph/schemas/generators.py

@@ -10,7 +10,8 @@ from typing import (
     get_origin,
     Type
 )
-
+from sqlalchemy.orm import sessionmaker
+from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy.orm import Mapped
 from sqlalchemy import ForeignKeyConstraint, func
 from sqlalchemy.inspection import inspect
@@ -895,6 +896,7 @@ def get_mutations(cls: "Graphemy"):
     """
     fields = {}
 
+    @strawberry.type
     class ModelMutations:
         pass
 

{bootgraph-1.10.0.dev24599 → bootgraph-1.11.0.dev24752}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "bootgraph"
-version = "v1.10.0.dev24599"
+version = "v1.11.0.dev24752"
 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"
@@ -51,6 +51,8 @@ jinja2 = "^3.1.2"
 mdx-include = "^1.4.2"
 mkdocstrings = "^0.26.1"
 mkdocstrings-python = "^1.11.1"
+asyncpg = ">=0.29.0"
+greenlet = ">=3.0.3"
 
 [build-system]
 requires = ["poetry-core"]