diff --git a/backend/app.py b/backend/app.py index 2abd341..50e795c 100644 --- a/backend/app.py +++ b/backend/app.py @@ -1,9 +1,8 @@ from application import create_app import os -app = create_app() +app = create_app(os.environ.get('FLASK_ENV', None)) if __name__ == "__main__": PORT = os.environ.get('PORT', 33507) - print('app.run') app.run(host='0.0.0.0', port=PORT, DEBUG=True) diff --git a/backend/application/__init__.py b/backend/application/__init__.py index c42fed8..e869b30 100644 --- a/backend/application/__init__.py +++ b/backend/application/__init__.py @@ -6,14 +6,13 @@ import os db = SQLAlchemy() -def create_app(): +def create_app(flask_env='development'): app = Flask(__name__, instance_relative_config=False) - FLASK_ENV = os.environ.get('FLASK_ENV', None) - if FLASK_ENV == 'production': + if flask_env == 'production': app.config.from_object("config.ProductionConfig") - elif FLASK_ENV == 'staging': - app.config.from_object("config.StagingConfig") - elif FLASK_ENV == 'development': + elif flask_env == 'testing': + app.config.from_object("config.TestingConfig") + elif flask_env == 'development': app.config.from_object("config.DevelopmentConfig") else: app.config.from_object("config.Config") diff --git a/backend/config.py b/backend/config.py index 02b37aa..9de0dd3 100644 --- a/backend/config.py +++ b/backend/config.py @@ -30,8 +30,8 @@ class ProductionConfig(Config): SQLALCHEMY_TRACK_MODIFICATIONS = False -class StagingConfig(Config): - STAGING = True +class TestingConfig(Config): + TESTING = True SQLALCHEMY_ECHO = False SQLALCHEMY_TRACK_MODIFICATIONS = False