def build_coastline(**kwargs): env = Env() kwargs["ti"].xcom_push(key="build_name", value=env.build_name) run_generation( env, ( sd.StageDownloadAndConvertPlanet(), sd.StageCoastline(use_old_if_fail=False), sd.StageCleanup(), ), ) env.finish()
def build_prolog(**kwargs): params = MapsGenerationDAG.get_params(**kwargs) env = Env(**params) kwargs["ti"].xcom_push(key="build_name", value=env.build_name) run_generation( env, ( sd.StageDownloadAndConvertPlanet(), sd.StageCoastline(), sd.StagePreprocess(), sd.StageFeatures(), sd.StageDownloadDescriptions(), ), )
def update_planet(**kwargs): env = Env() kwargs["ti"].xcom_push(key="build_name", value=env.build_name) if settings.DEBUG: env.add_skipped_stage(sd.StageUpdatePlanet) run_generation( env, ( sd.StageDownloadAndConvertPlanet(), sd.StageUpdatePlanet(), sd.StageCleanup(), ), ) env.finish()
def run_generation_from_first_stage(env: Env, stages: Iterable[Stage], build_lock: bool = True): from_stage = get_stage_name(next(iter(stages))) run_generation(env, stages, from_stage, build_lock)