Different contexts getters depending on the query or mutation about strawberry

[patrick91]

changing the context based on the resolver instead of setting only in the main router.

do you have a more concrete example of this? 😊

[Focadecombate]

Something like this:

src.main.py

import strawberry
from fastapi import FastAPI
from strawberry.fastapi import GraphQLRouter
from strawberry.tools import merge_types

from src.a import QueryA
from src.b import QueryB

app = FastAPI()

top_query = merge_types("AllQuery", (QueryB, QueryA))
main_schema = strawberry.Schema(top_query)
graphql_router = GraphQLRouter(schema=main_schema)

app.include_router(graphql_router, prefix="/graphql")

src.a.init.py

from typing import Any

import strawberry
from strawberry.fastapi import BaseContext
from strawberry.types import Info


class ServiceA:
    def hello_from_a(self):
        print("Hello from A")


class ContextA(BaseContext):
    def __init__(self, service: ServiceA) -> None:
        self.service = service


@strawberry.type
class QueryA:
    @strawberry.field
    def a(self, info: Info[ContextA, Any]) -> str:
        info.context.service.hello_from_a()
        return "a"

src.b.init.py

from typing import Any

import strawberry
from strawberry.fastapi import BaseContext
from strawberry.types import Info


class ServiceB:
    def hello_from_b(self):
        print("Hello from B")


class ContextB(BaseContext):
    def __init__(self, service: ServiceB) -> None:
        self.service = service


@strawberry.type
class QueryB:
    @strawberry.field
    def b(self, info: Info[ContextB, Any]) -> str:
        info.context.service.hello_from_b()
        return "b"