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
