Add register_spec_plugin

marshmallow-code · Mar 24, 2017 · 0b0a49c · 0b0a49c
1 parent 43c8760
commit 0b0a49c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/flask_rest_api/__init__.py b/flask_rest_api/__init__.py
@@ -78,3 +78,7 @@ def register_converter(self, converter, conv_type, conv_format):
     def register_field(self, field, field_type, field_format):
         """Register Marshmallow field in docs"""
         self._apispec.register_field(field, field_type, field_format)
+
+    def register_spec_plugin(self, plugin):
+        """Register apispec plugin"""
+        self._apispec.register_spec_plugin(plugin)
diff --git a/flask_rest_api/spec/__init__.py b/flask_rest_api/spec/__init__.py
@@ -79,3 +79,6 @@ def register_converter(self, converter, conv_type, conv_format):
 
     def register_field(self, field, field_type, field_format):
         FIELD_MAPPING[field] = (field_type, field_format)
+
+    def register_spec_plugin(self, plugin):
+        self.spec.setup_plugin(plugin)