def check_migrations(args): """ Function to wait for all airflow migrations to complete. Used for launching airflow in k8s @param timeout: @return: """ db.check_migrations(timeout=args.migration_wait_timeout)
def needs_db_migrations() -> bool: """ Return a boolean representing if the database has unapplied migrations. """ log_alembic = logging.getLogger("alembic.runtime.migration") log_alembic_level = log_alembic.level try: log_alembic.setLevel("WARN") check_migrations(0) log_alembic.setLevel(log_alembic_level) return False except TimeoutError: return True