def add_default_users(db): try: UserRole.add_default_roles() User.add_default_users() except sqlalchemy.exc.IntegrityError as e: db.session.rollback() print(e)
def admin_login(app, client): Role.add_default_roles() User.add_default_users() return client.post(url_for('account.login'), data=dict(email=app.config['ADMIN_EMAIL'], password=app.config['ADMIN_PASSWORD']), follow_redirects=True)
def db_create(): """ Creates a local database """ with app.app_context(): db.create_all() Role.add_default_roles() User.add_default_users() ControlLists.add_default_lists() db.session.commit() click.echo("Created the database")
def setUp(self): """Setup the test driver and create test users""" db.session.commit() db.drop_all() db.create_all() db.session.commit() # add default roles & admin user Role.add_default_roles() User.add_default_users() self.create_driver() self.driver.get(self.get_server_url())
def db_recreate(): """ Recreates a local database. You probably should not use this on production. """ with app.app_context(): db.drop_all() db.create_all() Role.add_default_roles() User.add_default_users() ControlLists.add_default_lists() db.session.commit() click.echo("Dropped then recreated the database")
def add_default_users(db): UserRole.add_default_roles() db.session.flush() User.add_default_users()