diff --git a/www/__init__.py b/www/__init__.py index ad09396..546d354 100644 --- a/www/__init__.py +++ b/www/__init__.py @@ -8,6 +8,7 @@ def create_app(): app.config.from_mapping( SQLALCHEMY_DATABASE_URI=os.environ.get("DATABASE_URI"), SQLALCHEMY_TRACK_MODIFICATIONS=os.environ.get("TRACK_MODIFICATIONS"), + SQLALCHEMY_POOL_RECYCLE=os.environ.get("POOL_RECYCLE"), USERNAME=os.environ.get("USERNAME"), PASSWORD=os.environ.get("PASSWORD"), SECRET_KEY=os.environ.get("SECRET_KEY") diff --git a/www/commands.py b/www/commands.py index 11aa410..2a472e5 100644 --- a/www/commands.py +++ b/www/commands.py @@ -41,6 +41,14 @@ def init_db_command(): generate_admin() generate_base_now() +@click.command('generate-admin') +@with_appcontext +def generate_admin_command(): + """ + Generate admin + """ + generate_admin() def init_app(app): app.cli.add_command(init_db_command) + app.cli.add_command(generate_admin_command)