Flask
Strawberry comes with a basic Flask integration. It provides a view that you can use to serve your GraphQL schema:
from flask import Flaskfrom strawberry.flask.views import GraphQLView
from api.schema import schema app = Flask(__name__)
app.add_url_rule( "/graphql", view_func=GraphQLView.as_view("graphql_view", schema=schema),)
if __name__ == "__main__": app.run()If you'd prefer to use an asynchronous view you can instead use the following
import which has the same interface as GraphQLView. This is helpful if using a
dataloader.
from strawberry.flask.views import AsyncGraphQLViewOptions
The GraphQLView accepts two options at the moment:
schema: mandatory, the schema created bystrawberry.Schema.graphiql:optional, defaults toTrue, whether to enable the GraphiQL interface.allow_queries_via_get: optional, defaults toTrue, whether to enable queries viaGETrequests
Extending the view
We allow to extend the base GraphQLView, by overriding the following methods:
get_context(self, response: Response) -> Anyget_root_value(self) -> Anyprocess_result(self, result: ExecutionResult) -> GraphQLHTTPResponseencode_json(self, response_data: GraphQLHTTPResponse) -> str
Note that the AsyncGraphQLView can also be extended by overriding the same
methods above, but get_context, get_root_value and process_result are
async functions.
get_context
get_context allows to provide a custom context object that can be used in your
resolver. You can return anything here, by default we return a dictionary with
the request. By default; the Response object from flask is injected via the
parameters.
class MyGraphQLView(GraphQLView): def get_context(self, response: Response) -> Any: return {"example": 1}
@strawberry.typeclass Query: @strawberry.field def example(self, info: Info) -> str: return str(info.context["example"])Here we are returning a custom context dictionary that contains only one item called "example".
Then we use the context in a resolver, the resolver will return "1" in this case.
get_root_value
get_root_value allows to provide a custom root value for your schema, this is
probably not used a lot but it might be useful in certain situations.
Here's an example:
class MyGraphQLView(GraphQLView): def get_root_value(self) -> Any: return Query(name="Patrick")
@strawberry.typeclass Query: name: strHere we are returning a Query where the name is "Patrick", so we when requesting the field name we'll return "Patrick" in this case.
process_result
process_result allows to customize and/or process results before they are sent
to the clients. This can be useful logging errors or hiding them (for example to
hide internal exceptions).
It needs to return an object of GraphQLHTTPResponse and accepts the execution
result.
from strawberry.http import GraphQLHTTPResponsefrom strawberry.types import ExecutionResult
from graphql.error.graphql_error import format_error as format_graphql_error
class MyGraphQLView(GraphQLView): def process_result(self, result: ExecutionResult) -> GraphQLHTTPResponse: data: GraphQLHTTPResponse = {"data": result.data}
if result.errors: data["errors"] = [format_graphql_error(err) for err in result.errors]
return dataIn this case we are doing the default processing of the result, but it can be tweaked based on your needs.
encode_json
encode_json allows to customize the encoding of the JSON response. By default
we use json.dumps but you can override this method to use a different encoder.
class MyGraphQLView(GraphQLView): def encode_json(self, data: GraphQLHTTPResponse) -> str: return json.dumps(data, indent=2)