Example #1
0
def create_user():
    from walkoff.serverdb import add_user, User, Role, initialize_default_resources_admin, \
        initialize_default_resources_guest
    db.create_all()

    # Setup admin and guest roles
    initialize_default_resources_admin()
    initialize_default_resources_guest()

    # Setup admin user
    admin_role = Role.query.filter_by(id=1).first()
    admin_user = User.query.filter_by(username="******").first()
    if not admin_user:
        add_user(username='******', password='******', roles=[1])
    elif admin_role not in admin_user.roles:
        admin_user.roles.append(admin_role)

    db.session.commit()

    apps = set(helpers.list_apps(walkoff.config.Config.APPS_PATH)) - set([
        _app.name for _app in
        current_app.running_context.execution_db.session.query(App).all()
    ])
    current_app.logger.debug('Found new apps: {0}'.format(apps))
    for app_name in apps:
        current_app.running_context.execution_db.session.add(
            App(name=app_name, devices=[]))
    db.session.commit()
    current_app.running_context.execution_db.session.commit()
    send_all_cases_to_workers()
    current_app.logger.handlers = logging.getLogger('server').handlers
    def setUp(self):
        import walkoff.server.flaskserver
        self.app = walkoff.server.flaskserver.app.test_client(self)
        self.app.testing = True
        self.context = walkoff.server.flaskserver.app.test_request_context()
        self.context.push()

        from walkoff.serverdb import initialize_default_resources_admin, initialize_default_resources_guest, Role
        db.create_all()

        # Setup admin and guest roles
        initialize_default_resources_admin()
        initialize_default_resources_guest()

        # Setup admin user
        admin_role = Role.query.filter_by(id=1).first()
        admin_user = User.query.filter_by(username="******").first()
        if not admin_user:
            add_user(username='******', password='******', roles=[1])
        elif admin_role not in admin_user.roles:
            admin_user.roles.append(admin_role)

        db.session.commit()

        self.admin_id = db.session.query(User).filter_by(
            username='******').first().id
Example #3
0
def create_user():
    from walkoff import executiondb
    from walkoff.serverdb import add_user, User, Role, initialize_default_resources_admin, \
        initialize_default_resources_guest
    db.create_all()

    # Setup admin and guest roles
    initialize_default_resources_admin()
    initialize_default_resources_guest()

    # Setup admin user
    admin_role = Role.query.filter_by(id=1).first()
    admin_user = User.query.filter_by(username="******").first()
    if not admin_user:
        add_user(username='******', password='******', roles=[1])
    elif admin_role not in admin_user.roles:
        admin_user.roles.append(admin_role)

    db.session.commit()

    apps = set(helpers.list_apps()) - set([
        _app.name
        for _app in executiondb.execution_db.session.query(App).all()
    ])
    app.logger.debug('Found apps: {0}'.format(apps))
    for app_name in apps:
        executiondb.execution_db.session.add(App(name=app_name, devices=[]))
    db.session.commit()
    executiondb.execution_db.session.commit()
    CaseSubscription.sync_to_subscriptions()

    app.logger.handlers = logging.getLogger('server').handlers
Example #4
0
    def setUpClass(cls):
        import walkoff.server.flaskserver
        cls.context = walkoff.server.flaskserver.app.test_request_context()
        cls.context.push()

        execution_db_help.setup_dbs()

        initialize_default_resources_admin()
        db.create_all()
Example #5
0
def create_user():
    from walkoff.serverdb import add_user, User, Role, initialize_default_resources_admin, \
        initialize_default_resources_guest
    from sqlalchemy_utils import database_exists, create_database

    if not database_exists(db.engine.url):
        create_database(db.engine.url)
    db.create_all()

    alembic_cfg = Config(walkoff.config.Config.ALEMBIC_CONFIG,
                         ini_section="walkoff",
                         attributes={'configure_logger': False})

    # This is necessary for a flask database
    connection = db.engine.connect()
    context = MigrationContext.configure(connection)
    script = ScriptDirectory.from_config(alembic_cfg)
    context.stamp(script, "head")

    # Setup admin and guest roles
    initialize_default_resources_admin()
    initialize_default_resources_guest()

    # Setup admin user
    admin_role = Role.query.filter_by(id=1).first()
    admin_user = User.query.filter_by(username="******").first()
    if not admin_user:
        add_user(username='******', password='******', roles=[1])
    elif admin_role not in admin_user.roles:
        admin_user.roles.append(admin_role)

    db.session.commit()

    apps = set(helpers.list_apps(walkoff.config.Config.APPS_PATH)) - set([
        _app.name for _app in
        current_app.running_context.execution_db.session.query(App).all()
    ])
    current_app.logger.debug('Found new apps: {0}'.format(apps))
    for app_name in apps:
        current_app.running_context.execution_db.session.add(
            App(name=app_name, devices=[]))
    db.session.commit()
    current_app.running_context.execution_db.session.commit()
    reschedule_all_workflows()
    current_app.logger.handlers = logging.getLogger('server').handlers
Example #6
0
def create_user():
    from walkoff.serverdb import add_user, User, Role, initialize_default_resources_admin, \
        initialize_default_resources_guest
    from sqlalchemy_utils import database_exists, create_database

    if not database_exists(db.engine.url):
        create_database(db.engine.url)
    db.create_all()

    alembic_cfg = Config(walkoff.config.Config.ALEMBIC_CONFIG, ini_section="walkoff",
                         attributes={'configure_logger': False})

    # This is necessary for a flask database
    connection = db.engine.connect()
    context = MigrationContext.configure(connection)
    script = ScriptDirectory.from_config(alembic_cfg)
    context.stamp(script, "head")

    # Setup admin and guest roles
    initialize_default_resources_admin()
    initialize_default_resources_guest()

    # Setup admin user
    admin_role = Role.query.filter_by(id=1).first()
    admin_user = User.query.filter_by(username="******").first()
    if not admin_user:
        add_user(username='******', password='******', roles=[1])
    elif admin_role not in admin_user.roles:
        admin_user.roles.append(admin_role)

    db.session.commit()

    apps = set(helpers.list_apps(walkoff.config.Config.APPS_PATH)) - set([_app.name
                                                                          for _app in
                                                                          current_app.running_context.execution_db.session.query(
                                                                              App).all()])
    current_app.logger.debug('Found new apps: {0}'.format(apps))
    for app_name in apps:
        current_app.running_context.execution_db.session.add(App(name=app_name, devices=[]))
    db.session.commit()
    current_app.running_context.execution_db.session.commit()
    reschedule_all_workflows()
    current_app.logger.handlers = logging.getLogger('server').handlers
Example #7
0
    def setUp(self):
        from walkoff.serverdb import initialize_default_resources_admin, initialize_default_resources_guest, Role
        db.create_all()

        # Setup admin and guest roles
        initialize_default_resources_admin()
        initialize_default_resources_guest()

        # Setup admin user
        admin_role = Role.query.filter_by(id=1).first()
        admin_user = User.query.filter_by(username="******").first()
        if not admin_user:
            add_user(username='******', password='******', roles=[1])
        elif admin_role not in admin_user.roles:
            admin_user.roles.append(admin_role)

        db.session.commit()

        self.admin_role_id = db.session.query(Role).filter_by(name='admin').first().id
        self.admin_user_id = db.session.query(User).filter_by(username='******').first().id
Example #8
0
    def setUp(self):
        from walkoff.serverdb import initialize_default_resources_admin, initialize_default_resources_guest, Role
        db.create_all()

        # Setup admin and guest roles
        initialize_default_resources_admin()
        initialize_default_resources_guest()

        # Setup admin user
        admin_role = Role.query.filter_by(id=1).first()
        admin_user = User.query.filter_by(username="******").first()
        if not admin_user:
            add_user(username='******', password='******', roles=[1])
        elif admin_role not in admin_user.roles:
            admin_user.roles.append(admin_role)

        db.session.commit()

        self.admin_role_id = db.session.query(Role).filter_by(
            name='admin').first().id
        self.admin_user_id = db.session.query(User).filter_by(
            username='******').first().id