feat: add depency injection

Author: YuSUN

.github/workflows/main.yml

@@ -4,3 +4,7 @@ bump:
 	sed -i -e "s/__version__.*/__version__ = '${VERSION}'/g" gql/__init__.py
 	poetry version -s ${VERSION}
 
+publish:
+	rm -rf dist
+	flit build
+	flit publish --repository teletraan

Makefile

@@ -1,3 +1,6 @@
+"""
+Python schema-first GraphQL library based on GraphQL-core.
+"""
 from .enum import enum_type  # noqa
 from .execute import ExecutionContext  # noqa
 from .middleware import MiddlewareManager  # noqa
@@ -15,4 +18,4 @@
 from .schema_visitor import SchemaDirectiveVisitor  # noqa
 from .utils import gql  # noqa
 
-__version__ = '0.2.10'
+__version__ = '0.3.0'

gql/__init__.py

@@ -0,0 +1,35 @@
+from typing import Any, Callable
+
+from graphql import GraphQLResolveInfo
+
+
+class ResolverDepends:
+    def __init__(self, dependency: Callable[..., Any]) -> None:
+        self.dependency = dependency
+
+    def get_parameters(self, parent: Any, info: GraphQLResolveInfo) -> tuple:
+        """
+        Get parameters from parent and info.
+        """
+        return ()
+
+    def execute(self, parent: Any, info: GraphQLResolveInfo):
+        """
+        Execute depends to get real value.
+        """
+        return self.dependency(*self.get_parameters(parent, info))
+
+
+class InfoDepends(ResolverDepends):
+    def get_parameters(self, parent: Any, info: GraphQLResolveInfo):
+        return (info,)
+
+
+class ContextDepends(ResolverDepends):
+    def get_parameters(self, parent: Any, info: GraphQLResolveInfo):
+        return (info.context,)
+
+
+class RequestDepends(ResolverDepends):
+    def get_parameters(self, parent: Any, info: GraphQLResolveInfo):
+        return (info.context.get('request'),)

gql/depends.py

@@ -1,3 +1,4 @@
+import inspect
 import logging
 import traceback
 from collections import defaultdict
@@ -19,6 +20,7 @@
     is_union_type,
 )
 
+from .depends import ResolverDepends
 from .utils import execute_async_function, recursive_to_snake_case, to_camel_case, to_snake_case
 
 
@@ -44,15 +46,18 @@ def print_resolver_error(info: GraphQLResolveInfo):
 def reference_resolver(type_name: str):
     if type_name in reference_resolver_map:
         raise Exception(
-            f"{type_name} is already registered by "
-            f"{reference_resolver_map[type_name].__code__}"
+            f"{type_name} is already registered by " f"{reference_resolver_map[type_name].__code__}"
         )
 
     def wrap(func: ReferenceResolver):
         @wraps(func)
         def sync_resolver(parent, info, representation):
             try:
-                result = func(parent, info, representation)
+                kwargs = {}
+                for name, param in inspect.signature(func).parameters.items():
+                    if isinstance(param.default, ResolverDepends):
+                        kwargs[name] = param.default.execute(parent, info)
+                result = func(parent, info, representation, **kwargs)
                 return dict(result) if result is not None else result
             except Exception as exc:
                 print_resolver_error(info)
@@ -62,7 +67,11 @@ def sync_resolver(parent, info, representation):
         @wraps(func)
         async def async_resolver(parent, info, representation):
             try:
-                result = await execute_async_function(func, parent, info, representation)
+                kwargs = {}
+                for name, param in inspect.signature(func).parameters.items():
+                    if isinstance(param.default, ResolverDepends):
+                        kwargs[name] = param.default.execute(parent, info)
+                result = await execute_async_function(func, parent, info, representation, **kwargs)
                 return dict(result) if result is not None else result
             except Exception as exc:
                 print_resolver_error(info)
@@ -81,8 +90,7 @@ async def async_resolver(parent, info, representation):
 def type_resolver(type_name: str):
     if type_name in type_resolver_map:
         raise Exception(
-            f"{type_name} is already registered by "
-            f"{type_resolver_map[type_name].__code__}"
+            f"{type_name} is already registered by " f"{type_resolver_map[type_name].__code__}"
         )
 
     def wrap(func):
@@ -103,6 +111,11 @@ def wrap(func: GraphQLFieldResolver):
         def sync_resolver(parent, info, **kwargs):
             if snake_argument:
                 kwargs = recursive_to_snake_case(kwargs)
+
+            for name, param in inspect.signature(func).parameters.items():
+                if isinstance(param.default, ResolverDepends):
+                    kwargs[name] = param.default.execute(parent, info)
+
             if not print_exc:
                 return func(parent, info, **kwargs)
 
@@ -117,6 +130,11 @@ def sync_resolver(parent, info, **kwargs):
         async def async_resolver(parent, info, **kwargs):
             if snake_argument:
                 kwargs = recursive_to_snake_case(kwargs)
+
+            for name, param in inspect.signature(func).parameters.items():
+                if isinstance(param.default, ResolverDepends):
+                    kwargs[name] = param.default.execute(parent, info)
+
             if not print_exc:
                 return await execute_async_function(func, parent, info, **kwargs)