mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-14 17:22:17 -05:00
27 lines
798 B
Python
27 lines
798 B
Python
from flask_script import Manager, Server
|
|
from flask_migrate import Migrate, MigrateCommand
|
|
from app import (create_app, db, commands)
|
|
import os
|
|
|
|
default_env_file = '/home/ubuntu/environment'
|
|
environment = 'live'
|
|
|
|
if os.path.isfile(default_env_file):
|
|
with open(default_env_file, 'r') as environment_file:
|
|
environment = environment_file.readline().strip()
|
|
|
|
from app.config import configs
|
|
|
|
os.environ['NOTIFY_API_ENVIRONMENT'] = configs[environment]
|
|
|
|
application = create_app()
|
|
|
|
manager = Manager(application)
|
|
migrate = Migrate(application, db)
|
|
manager.add_command('db', MigrateCommand)
|
|
manager.add_command('purge_functional_test_data', commands.PurgeFunctionalTestDataCommand)
|
|
manager.add_command('custom_db_script', commands.CustomDbScript)
|
|
|
|
if __name__ == '__main__':
|
|
manager.run()
|