flask_graphql_file_upload_with_graphene_upload_scalar.py

1import graphene
2from flask import Flask
3from graphene_file_upload.flask import FileUploadGraphQLView
4from graphene_file_upload.scalars import Upload
5
6class UploadFile(graphene.Mutation):
7    class Arguments:
8        file = Upload(required=True)
9
10    success = graphene.Boolean()
11
12    def mutate(self, info, file, **mutation_input):
13        # file is a Werkzeug FileStorage object
14        print(f"Received file: {file.filename}")
15        return UploadFile(success=True)
16
17class Mutation(graphene.ObjectType):
18    upload_file = UploadFile.Field()
19
20schema = graphene.Schema(mutation=Mutation)
21
22app = Flask(__name__)
23
24app.add_url_rule(
25    '/graphql',
26    view_func=FileUploadGraphQLView.as_view(
27        'graphql',
28        schema=schema,
29        graphiql=True,
30    )
31)
32
33if __name__ == '__main__':
34    app.run(debug=True)