def generate_random_run_id(prefix, random_str_len=4, **context):
run_id = '_'.join([
prefix,
airflow_docker_utils.get_start_date_from_context(context).strftime(
'%Y-%m-%d_%H:%M'),
airflow_docker_utils.id_generator(size=random_str_len)
])
logging.info('Generated run_id: ' + run_id)
return run_id
def check_data_push_callable(task_ids, **context):
exec_date = airflow_docker_utils.get_start_date_from_context(context)
if exec_date.hour == 18 and exec_date.minute == 0:
return task_ids[0]
else:
return task_ids[1]
