Handle forms exception (consumes) on class too

Author: noirbizarre

flask_restplus/swagger.py

@@ -363,8 +363,10 @@ def serialize_operation(self, doc, method):
         if doc.get('deprecated') or doc[method].get('deprecated'):
             operation['deprecated'] = True
         # Handle form exceptions:
-        if operation['parameters'] and any(p['in'] == 'formData' for p in operation['parameters']):
-            if any(p['type'] == 'file' for p in operation['parameters']):
+        doc_params = list(doc.get('params', {}).values())
+        all_params = doc_params + (operation['parameters'] or [])
+        if all_params and any(p['in'] == 'formData' for p in all_params):
+            if any(p['type'] == 'file' for p in all_params):
                 operation['consumes'] = ['multipart/form-data']
             else:
                 operation['consumes'] = ['application/x-www-form-urlencoded', 'multipart/form-data']

tests/test_swagger.py

@@ -710,6 +710,33 @@ def get(self):
 
         assert op['consumes'] == ['multipart/form-data']
 
+    def test_parser_parameter_in_files_on_class(self, api, client):
+        parser = api.parser()
+        parser.add_argument('in_files', type=FileStorage, location='files')
+
+        @api.route('/with-parser/', endpoint='with-parser')
+        @api.expect(parser)
+        class WithParserResource(restplus.Resource):
+            def get(self):
+                return {}
+
+        data = client.get_specs()
+        assert '/with-parser/' in data['paths']
+
+        path = data['paths']['/with-parser/']
+        assert len(path['parameters']) == 1
+
+        parameter = path['parameters'][0]
+        assert parameter['name'] == 'in_files'
+        assert parameter['type'] == 'file'
+        assert parameter['in'] == 'formData'
+
+        assert 'consumes' not in path
+
+        op = path['get']
+        assert 'consumes' in op
+        assert op['consumes'] == ['multipart/form-data']
+
     def test_explicit_parameters(self, api, client):
         @api.route('/name/<int:age>/', endpoint='by-name')
         class ByNameResource(restplus.Resource):