From 45881396549e6e960afe48deaa674dc85dcc14a6 Mon Sep 17 00:00:00 2001 From: Cameron Hurst Date: Wed, 6 Jan 2021 13:57:55 -0500 Subject: [PATCH] feat: support version 3 for graphene --- graphene_file_upload/flask/__init__.py | 14 +++++--------- 1 file changed, 5 insertions(+), 9 deletions(-) diff --git a/graphene_file_upload/flask/__init__.py b/graphene_file_upload/flask/__init__.py index 4cf769f..9e9a2c2 100644 --- a/graphene_file_upload/flask/__init__.py +++ b/graphene_file_upload/flask/__init__.py @@ -1,6 +1,6 @@ """Apply multipart request spec to flask""" from flask import request -from flask_graphql import GraphQLView +from graphql_server.flask import GraphQLView from graphql_server import load_json_body from ..utils import place_files_in_operations @@ -12,12 +12,8 @@ class FileUploadGraphQLView(GraphQLView): def parse_body(self): """Handle multipart request spec for multipart/form-data""" content_type = request.mimetype - if content_type == 'multipart/form-data': - operations = load_json_body(request.form.get('operations', '{}')) - files_map = load_json_body(request.form.get('map', '{}')) - return place_files_in_operations( - operations, - files_map, - request.files - ) + if content_type == "multipart/form-data": + operations = load_json_body(request.form.get("operations", "{}")) + files_map = load_json_body(request.form.get("map", "{}")) + return place_files_in_operations(operations, files_map, request.files) return super(FileUploadGraphQLView, self).parse_body()