Add function which takes arg 'script_info' and calls create_app

metabrainz · Feb 28, 2018 · bc297b0 · bc297b0
1 parent 201002e
commit bc297b0
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 3 deletions.
diff --git a/db/dump_manage.py b/db/dump_manage.py
@@ -8,7 +8,7 @@
 import webserver
 
 
-cli = FlaskGroup(add_default_commands=False, create_app=webserver.create_app)
+cli = FlaskGroup(add_default_commands=False, create_app=webserver.create_app_flaskgroup)
 
 
 @cli.command()

diff --git a/manage.py b/manage.py
@@ -19,7 +19,7 @@
 
 ADMIN_SQL_DIR = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'admin', 'sql')
 
-cli = FlaskGroup(add_default_commands=False, create_app=webserver.create_app)
+cli = FlaskGroup(add_default_commands=False, create_app=webserver.create_app_flaskgroup)
 cli.add_command(shell_command)
 
 

diff --git a/webserver/__init__.py b/webserver/__init__.py
@@ -3,6 +3,11 @@
 
 API_PREFIX = '/api/'
 
+def create_app_flaskgroup(script_info):
+    """Factory function that accepts script_info and creates a Flask application"""
+    return create_app()
+
+
 def create_app_with_configuration(config_path=None):
     """Create a Flask application and load all configuration files"""
 

diff --git a/worker_manage.py b/worker_manage.py
@@ -7,7 +7,7 @@
 import hl_extractor.hl_calc
 import webserver
 
-cli = FlaskGroup(add_default_commands=False, create_app=webserver.create_app)
+cli = FlaskGroup(add_default_commands=False, create_app=webserver.create_app_flaskgroup)
 
 
 @cli.command('hl_extractor')