def run(self):
command.upgrade(alembic_cfg, "head")
roles = ['admin', 'user', 'superuser']
user_datastore = SQLAlchemyUserDatastore(db, User, Role)
for r in roles:
user_datastore.find_or_create_role(r)
try:
db.session.commit()
except IntegrityError:
# Happens if records already exist.
db.session.rollback()
def run(self):
command.upgrade(alembic_cfg, "head")
roles = ['admin', 'user', 'superuser']
user_datastore = SQLAlchemyUserDatastore(db, User, Role)
for r in roles:
user_datastore.find_or_create_role(r)
try:
db.session.commit()
except IntegrityError:
# Happens if records already exist.
db.session.rollback()
def run(self, user):
from app import db
user_datastore = SQLAlchemyUserDatastore(db, User, Role)
u = User.query.filter_by(email=user).first()
role = user_datastore.find_or_create_role('admin')
user_datastore.add_role_to_user(u, role)
db.session.commit()
def run(self, user):
from app import db
user_datastore = SQLAlchemyUserDatastore(db, User, Role)
u = User.query.filter_by(email=user).first()
role = user_datastore.find_or_create_role('admin')
user_datastore.add_role_to_user(u, role)
db.session.commit()
