From 77601d0c8fae6b3fb8f0f380ab5a3cd012bf2e63 Mon Sep 17 00:00:00 2001 From: Gadi Date: Fri, 8 Nov 2019 13:19:54 +0200 Subject: [PATCH] add static url variables to init --- flask_apispec/extension.py | 17 +++++++++++++---- 1 file changed, 13 insertions(+), 4 deletions(-) diff --git a/flask_apispec/extension.py b/flask_apispec/extension.py index 68636b1..bc13eca 100644 --- a/flask_apispec/extension.py +++ b/flask_apispec/extension.py @@ -35,15 +35,24 @@ def get_pet(pet_id): docs.register(get_pet) :param Flask app: App associated with API documentation + :param String static_folder: Static files folder location + :param String template_folder: Template files folder location + :param String static_url_path: URL to serve static files :param APISpec spec: apispec specification associated with API documentation """ - def __init__(self, app=None): + def __init__(self, app=None, + static_folder='./static', + template_folder='./templates', + static_url_path='/flask-apispec/static'): self._deferred = [] self.app = app self.view_converter = None self.resource_converter = None self.spec = None + self.static_folder = static_folder + self.template_folder = template_folder + self.static_url_path = static_url_path if app: self.init_app(app) @@ -71,9 +80,9 @@ def add_swagger_routes(self): blueprint = flask.Blueprint( 'flask-apispec', __name__, - static_folder='./static', - template_folder='./templates', - static_url_path='/flask-apispec/static', + static_folder=self.static_folder, + template_folder=self.template_folder, + static_url_path=self.static_url_path, ) json_url = self.app.config.get('APISPEC_SWAGGER_URL', '/swagger/')