6

在Graphene Pythonschema.py中，当无法访问要设置 cookie 的对象时，应该如何设置HttpResponsecookie？

我当前的实现是通过覆盖 GraphQLView 的 dispatch 方法来设置 cookie data.operationName。这涉及对我需要设置 cookie 的操作名称/突变进行硬编码。

在views.py中：

class PrivateGraphQLView(GraphQLView):
    data = self.parse_body(request)
    operation_name = data.get('operationName')
    # hard-coding === not pretty.
    if operation_name in ['loginUser', 'createUser']:
        ...
        response.set_cookie(...)
    return response

有没有一种更简洁的方法来为特定的 Graphene Python 突变设置 cookie？

4

2 回答 2

4

最终通过中间件设置 cookie。

class CookieMiddleware(object):

    def resolve(self, next, root, args, context, info):
        """
        Set cookies based on the name/type of the GraphQL operation
        """

        # set cookie here and pass to dispatch method later to set in response
        ...

在自定义 graphql 视图中views.py，重写 dispatch 方法以读取 cookie 并设置它。

class MyCustomGraphQLView(GraphQLView):  

    def dispatch(self, request, *args, **kwargs):
        response = super(MyCustomGraphQLView, self).dispatch(request, *args, **kwargs)
        # Set response cookies defined in middleware
        if response.status_code == 200:
            try:
                response_cookies = getattr(request, CookieMiddleware.MIDDLEWARE_COOKIES)
            except:
                pass
            else:
                for cookie in response_cookies:
                    response.set_cookie(cookie.get('key'), cookie.get('value'), **cookie.get('kwargs'))
        return response
于 2017-09-22T12:22:47.540 回答
2

我有一个类似的问题。我需要能够在突变中设置语言 cookie，并最终将请求实例与自定义中间件结合使用。

这是简化的代码：

class SetLanguage(Mutation):
    class Arguments:
        code = String(required=True)

    ok = Field(Boolean)
    language = Field(LanguageType)

    def mutate(root, info, code):
        info.context.set_language_cookie = code
        return SetLanguage(ok=True, language=code)

突变无权访问响应，因此它临时将值存储在请求实例上。创建响应后，自定义中间件会检索它并设置 cookie：

class LanguageConfigMiddleware:
    def __init__(self, get_response):
        self.get_response = get_response

    def __call__(self, request):
        response = self.get_response(request)

        if code := getattr(request, "set_language_cookie", None):
            response.set_cookie(settings.LANGUAGE_COOKIE_NAME, code)

        return response
于 2020-02-11T01:51:42.470 回答