feat: Allow using custom Swagger template for SwaggerView. (#1639)

dpgaspar · Jun 9, 2021 · 8fe7872 · 8fe7872
1 parent 1ff2283
commit 8fe7872
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/docs/config.rst b/docs/config.rst
@@ -241,6 +241,8 @@ Use config.py to configure the following parameters. By default it will use SQLL
 +----------------------------------------+--------------------------------------------+-----------+
 | FAB_API_SWAGGER_UI                     | Enables a Swagger UI view (Boolean)        |   No      |
 +----------------------------------------+--------------------------------------------+-----------+
+| FAB_API_SWAGGER_TEMPLATE               | Path of your custom Swagger Template       |   No      |
++----------------------------------------+--------------------------------------------+-----------+
 | FAB_API_ALLOW_JSON_QS                  | Allow query string parameters to be JSON   |           |
 |                                        | Default is True (Boolean)                  |   No      |
 +----------------------------------------+--------------------------------------------+-----------+

diff --git a/flask_appbuilder/api/manager.py b/flask_appbuilder/api/manager.py
@@ -86,7 +86,9 @@ class SwaggerView(BaseView):
     @has_access
     def show(self, version):
         return self.render_template(
-            "appbuilder/swagger/swagger.html",
+            self.appbuilder.app.config.get(
+                "FAB_API_SWAGGER_TEMPLATE", "appbuilder/swagger/swagger.html"
+            ),
             openapi_uri=self.openapi_uri.format(version),
         )