ctp_gcs_task_id = 'covid_tracking_project_to_gcs'
ctp_gcs_payload = util.generate_gcs_payload(_CTP_WORKFLOW_ID,
filename=_CTP_GCS_FILENAME,
url=_CTP_DOWNLOAD_URL)
ctp_gcs_operator = util.create_gcs_ingest_operator(ctp_gcs_task_id,
ctp_gcs_payload,
data_ingestion_dag)
# Standardize and write to BQ
ctp_bq_payload = util.generate_bq_payload(_CTP_WORKFLOW_ID,
_CTP_DATASET,
filename=_CTP_GCS_FILENAME)
ctp_bq_op = util.create_bq_ingest_operator('ctp_standardize', ctp_bq_payload,
data_ingestion_dag)
# Join with FIPS
ctp_aggregator_payload = {'dataset_name': _CTP_DATASET}
ctp_aggregator_operator = util.create_aggregator_operator(
'covid_tracking_project_aggregator', ctp_aggregator_payload,
data_ingestion_dag)
# Export to json
ctp_exporter_payload = {'dataset_name': _CTP_DATASET}
ctp_exporter_operator = util.create_exporter_operator(
'covid_tracking_project_exporter', ctp_exporter_payload,
data_ingestion_dag)
# Covid Tracking Project Ingestion DAG
(ctp_gcs_operator >> ctp_bq_op >> ctp_aggregator_operator >>
ctp_exporter_operator)
'acs_hhi_ingestion_dag',
default_args=default_args,
schedule_interval='@yearly',
description='Ingestion configuration for ACS Household Income')
acs_hhi_gcs_task_id = 'acs_hhi_to_gcs'
acs_hhi_gcs_payload = util.generate_gcs_payload(_ACS_WORKFLOW_ID,
url=_ACS_BASE_URL)
acs_hhi_gcs_operator = util.create_gcs_ingest_operator(acs_hhi_gcs_task_id,
acs_hhi_gcs_payload,
data_ingestion_dag)
acs_hhi_bq_payload = util.generate_bq_payload(_ACS_WORKFLOW_ID,
_ACS_DATASET_NAME,
url=_ACS_BASE_URL)
acs_hhi_bq_operator = util.create_bq_ingest_operator('acs_hhi_to_bq',
acs_hhi_bq_payload,
data_ingestion_dag)
acs_hhi_aggregator_payload = {'dataset_name': _ACS_DATASET_NAME}
acs_hhi_aggregator_operator = util.create_aggregator_operator(
'acs_hhi_aggregator', acs_hhi_aggregator_payload, data_ingestion_dag)
acs_hhi_exporter_payload = {'dataset_name': _ACS_DATASET_NAME}
acs_hhi_exporter_operator = util.create_exporter_operator(
'acs_hhi_exporter', acs_hhi_exporter_payload, data_ingestion_dag)
# Ingestion DAG
(acs_hhi_gcs_operator >> acs_hhi_bq_operator >> acs_hhi_aggregator_operator >>
acs_hhi_exporter_operator)
_CDC_RESTRICTED_DATASET = 'cdc_restricted_data'
default_args = {'start_date': days_ago(0)}
data_ingestion_dag = DAG(
'cdc_restricted_data_dag',
default_args=default_args,
description='Ingestion configuration for CDC Restricted Data')
# Standardize the CDC restricted data
cdc_bq_payload = util.generate_bq_payload(
_CDC_RESTRICTED_WORKFLOW_ID,
_CDC_RESTRICTED_DATASET,
gcs_bucket=Variable.get('GCS_MANUAL_UPLOADS_BUCKET'),
filename=_CDC_RESTRICTED_GCS_FILENAMES)
cdc_restricted_bq_op = util.create_bq_ingest_operator(
'cdc_restricted_gcs_to_bq', cdc_bq_payload, data_ingestion_dag)
cdc_restricted_aggregator_payload = {'dataset_name': _CDC_RESTRICTED_DATASET}
cdc_restricted_aggregator_operator = util.create_aggregator_operator(
'cdc_restricted_aggregator', cdc_restricted_aggregator_payload,
data_ingestion_dag)
cdc_restricted_exporter_payload = {'dataset_name': _CDC_RESTRICTED_DATASET}
cdc_restricted_exporter_operator = util.create_exporter_operator(
'cdc_restricted_exporter', cdc_restricted_exporter_payload,
data_ingestion_dag)
# CDC Restricted Data Ingestion DAG
cdc_restricted_bq_op >> cdc_restricted_aggregator_operator >> cdc_restricted_exporter_operator
schedule_interval="@yearly",
description="Ingestion configuration for ACS Health Insurance",
)
acs_hi_gcs_task_id = "acs_health_insurance_to_gcs"
acs_hi_gcs_payload = util.generate_gcs_payload(_ACS_WORKFLOW_ID,
url=_ACS_BASE_URL)
acs_hi_gcs_operator = util.create_gcs_ingest_operator(acs_hi_gcs_task_id,
acs_hi_gcs_payload,
data_ingestion_dag)
acs_hi_bq_payload = util.generate_bq_payload(_ACS_WORKFLOW_ID,
_ACS_DATASET_NAME,
url=_ACS_BASE_URL)
acs_hi_bq_operator = util.create_bq_ingest_operator(
"acs_health_insurance_to_bq", acs_hi_bq_payload, data_ingestion_dag)
acs_hi_aggregator_payload = {"dataset_name": _ACS_DATASET_NAME}
acs_hi_aggregator_operator = util.create_aggregator_operator(
"acs_health_insurance_aggregator", acs_hi_aggregator_payload,
data_ingestion_dag)
acs_hi_exporter_payload = {"dataset_name": _ACS_DATASET_NAME}
acs_hi_exporter_operator = util.create_exporter_operator(
"acs_health_insurance_exporter", acs_hi_exporter_payload,
data_ingestion_dag)
# Ingestion DAG
(acs_hi_gcs_operator >> acs_hi_bq_operator >> acs_hi_aggregator_operator >>
acs_hi_exporter_operator)
import util
_UHC_WORKFLOW_ID = 'UHC_DATA'
_UHC_DATASET_NAME = 'uhc_data'
default_args = {
'start_date': days_ago(0),
}
data_ingestion_dag = DAG('uhc_ingestion_dag',
default_args=default_args,
schedule_interval=None,
description='Ingestion configuration for UHC')
uhc_bq_payload = util.generate_bq_payload(_UHC_WORKFLOW_ID, _UHC_DATASET_NAME)
uhc_pop_bq_operator = util.create_bq_ingest_operator('uhc_to_bq',
uhc_bq_payload,
data_ingestion_dag)
uhc_aggregator_payload = {'dataset_name': _UHC_DATASET_NAME}
uhc_aggregator_operator = util.create_aggregator_operator(
'uhc_aggregator', uhc_aggregator_payload, data_ingestion_dag)
uhc_exporter_payload = {'dataset_name': _UHC_DATASET_NAME}
uhc_exporter_operator = util.create_exporter_operator('uhc_exporter',
uhc_exporter_payload,
data_ingestion_dag)
# Ingestion DAG
uhc_pop_bq_operator >> uhc_aggregator_operator >> uhc_exporter_operator
