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()