From c35f622d249e816a990e63d8aaa549f08956b4ea Mon Sep 17 00:00:00 2001 From: deflax Date: Mon, 1 Apr 2024 20:00:56 +0300 Subject: [PATCH] seed roles in seed_db --- src/forest/manage.py | 22 ++++++---------------- 1 file changed, 6 insertions(+), 16 deletions(-) diff --git a/src/forest/manage.py b/src/forest/manage.py index b6d2e94..9a74fb6 100644 --- a/src/forest/manage.py +++ b/src/forest/manage.py @@ -4,9 +4,9 @@ import os from flask.cli import FlaskGroup from app import app, db -from models import User +from models import Role, User -from flask_migrate import Migrate +from flask_migrate import Migrate, upgrade cli = FlaskGroup(app) @@ -19,25 +19,15 @@ 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.""" - from flask_migrate import upgrade - from app.models import Role, User - - # migrate database to latest revision - upgrade() - # create user roles Role.insert_roles() +@cli.command("upgrade_db") +def upgrade_db(): + # migrate database to latest revision + upgrade() @cli.command("restore_db") def restore_db(restore_file):