使用functools.wraps函数与一些不同的参数列表

Question

我想做这样的事情：

def need_session(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        session = SessionLocal()
        try:
            func(session, *args, **kwargs)
        finally:
            session.close()

我正在使用wraps函数，因为我正在使用草莓，取决于参数类型。

如您所见，给出了func一个额外的session参数。

我该如何完成这项工作？

如果我尝试使用具有上面装饰器装饰的功能的草莓运行GraphQl Server，则会给我这个错误：

TypeError：无法解决突变场。意外的类型'＆lt; class'sqlalchemy.orm.session.session'＆gt;'

如果我将装饰器更改为：

def need_session(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        session = SessionLocal()
        kwargs['session'] = session
        try:
            func(*args, **kwargs)
        finally:
            session.close()

并将装饰的函数的参数列表更改为：

def some_func(some_arg: SomeClass, **kwargs):
    ...

我仍然会遇到此错误：

strawberry.exceptions.missingargumentsAnnotationsError：缺少参数的注释“ Kwargs”字段“ login_user”，您是否忘记了添加它？

Answer 1

为什么错误？

您会得到此例外，因为草莓不支持** kwargs，每个strawberry.field.field必须为所有参数提供注释。它们由strawberry.type装饰器处理，因此您无法修改功能签名。

可能的解决方案1：

您可以通过编写架构扩展名来提供草莓信息内部的会话。

示例：

from strawberry.extensions import Extension
from mydb import get_db_session


class DbSessionMiddleWare(Extension):
    def on_request_start(self):
        self.execution_context.context["db"] = get_db_session()

    def on_request_end(self):
        self.execution_context.context["db"].close()

schema = strawberry.Schema(
    query=Queries,
    mutation=Mutations,
    extensions=[
        DbSessionMiddleWare,
    ],
)

一些解析器：

from strawberry.types import Info


def resolve_name(info: Info):
    name = info.context['db'].execute("SELECT ...")

---

可能的解决方案2：

您可以用自己的装饰器包裹草莓装饰器，并根据需要注入一些注释。
请记住，您应该清除任何*args或** kwargs从功能签名的其他方式，草莓会抱怨这一点。

例子：

from strawberry.arguments import StrawberryArgument, StrawberryAnnotation, UNSET
import inspect

def create_strawberry_argument(
    python_name: str, graphql_name: str, type_, default=None
):
    return StrawberryArgument(
        python_name=python_name,
        graphql_name=graphql_name,
        type_annotation=StrawberryAnnotation(type_),
        default=default or UNSET,
    )

def hide_args_kwargs(field):

    sig = inspect.signature(field)
    cleared = tuple(
        p for p in sig.parameters.values() if p.name not in ("kwargs", "args")
    )
    field.__signature__ = inspect.signature(field).replace(parameters=(cleared))
    return field

def inject_mutation(field, arguments: tuple[str, type]):
    field = hide_args_kwargs(field)
    field = strawberry.mutation(field, description=field.__doc__)
    for arg_tuple in arguments:
        arg = create_strawberry_argument(arg_tuple[0], arg_tuple[0], arg_tuple[1])
        field.arguments.append(arg)
    return field

@inject_mutation(('arg2', int))
def some_resolver(info, arg1, **kwargs) -> SomeOutputType:
    f = kwargs.get('arg2', None)