From 15431263075ce9a782487555a8599ffea9c8f5e4 Mon Sep 17 00:00:00 2001 From: Daniel afx Date: Fri, 4 Feb 2022 15:27:05 +0200 Subject: [PATCH] fix pep requierments --- flask/manage.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/flask/manage.py b/flask/manage.py index 1b4f7fd..c8ae3e0 100644 --- a/flask/manage.py +++ b/flask/manage.py @@ -5,7 +5,8 @@ import subprocess, shlex from forest import app, db, User from flask.cli import FlaskGroup -from flask_migrate import Migrate, MigrateCommand +from flask_migrate import Migrate +from flask_migrate import :wMigrateCommand cli = FlaskGroup(app) migrate = Migrate(app, db) @@ -18,11 +19,13 @@ def create_db(): db.create_all() db.session.commit() + @cli.command("seed_db") def seed_db(): db.session.add(User(email="daniel@deflax.net")) db.session.commit() + @cli.command("upgrade_db") def upgrade_db(): """Run deployment tasks.""" @@ -35,6 +38,7 @@ def upgrade_db(): # create user roles Role.insert_roles() + @cli.command("restore_db") def restore_db(restore_file): """ recreate db from grid export with python3 manage.py restore /path/grid.tar.bz2 """ @@ -44,6 +48,7 @@ def restore_db(restore_file): db.session.add(User(email=str(user), password=str(password), confirmed=True, confirmed_on=datetime.datetime.now())) db.session.commit() + if __name__ == "__main__": cli()