def add_default_pool_if_not_exists(session=None): from airflow.models.pool import Pool if not Pool.get_pool(Pool.DEFAULT_POOL_NAME, session=session): default_pool = Pool( pool=Pool.DEFAULT_POOL_NAME, slots=conf.getint(section='core', key='non_pooled_task_slot_count', fallback=128), description="Default pool", ) session.add(default_pool) session.commit()
def add_default_pool_if_not_exists(session=None): """ Add default pool if it does not exist. """ if not Pool.get_pool(Pool.DEFAULT_POOL_NAME, session=session): default_pool = Pool( pool=Pool.DEFAULT_POOL_NAME, slots=conf.getint(section='core', key='non_pooled_task_slot_count', fallback=128), description="Default pool", ) session.add(default_pool) session.commit()
def get_pool(self, name): pool = Pool.get_pool(pool_name=name) if not pool: raise PoolNotFound(f"Pool {name} not found") return pool.pool, pool.slots, pool.description