Question

在Graphene Python中，当无法访问schema.py对象以设置Cookie时，如何在HttpResponse中设置Cookie？

我目前的实现是通过捕获data.operationName来覆盖GraphQLView的调度方法来设置cookie。这涉及到我需要设置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？

Answer 1

通过中间件来设置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中，覆盖调度方法以读取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

Answer 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

如何在Graphene Python变异中设置cookie？

2 个答案: