def _prepare_db_queue_for_dql()

in evalbench/evaluator/db_manager.py [0:0]

• 15 lines of code
• 3 McCabe index (conditional complexity)

def _prepare_db_queue_for_dql(core_db: DB, db_name, db_config, setup_config, num_dbs):
    """For DQL, use the same single DB with a user that has only DQL access."""
    db_queue = Queue[DB]()
    dql_db_config = deepcopy(db_config)
    if setup_config:
        setup_scripts, data = _get_setup_values(
            setup_config, db_name, db_config.get("db_type")
        )
        core_db.set_setup_instructions(setup_scripts, data)
        core_db.resetup_database(False, True)
        dql_db_config["user_name"] = core_db.get_dql_user()
        dql_db_config["password"] = core_db.get_tmp_user_password()
    singular_db = get_database(dql_db_config, db_name)
    for _ in range(num_dbs):
        db_queue.put(singular_db)
    return db_queue