1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
from typing import Any
from fastapi import BackgroundTasks, Depends, FastAPI, Request, WebSocket
from strawberry.fastapi import GraphQLRouter
from tests.views.schema import schema
def custom_context_dependency() -> str:
return "Hi!"
async def get_context(
background_tasks: BackgroundTasks,
request: Request = None,
ws: WebSocket = None,
custom_value=Depends(custom_context_dependency),
) -> dict[str, Any]:
return {
"custom_value": custom_value,
"request": request or ws,
"background_tasks": background_tasks,
}
async def get_root_value(
request: Request = None, ws: WebSocket = None
) -> Request | WebSocket:
return request or ws
def create_app(schema=schema, **kwargs: Any) -> FastAPI:
app = FastAPI()
graphql_app = GraphQLRouter(
schema, context_getter=get_context, root_value_getter=get_root_value, **kwargs
)
app.include_router(graphql_app, prefix="/graphql")
return app
|
