# Copyright 2021 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

---
resources:

  - type: bigquery_table
    table_id: national_trends
    description: "COVID-19 Italy National Trends"

  - type: bigquery_table
    dataset_id: covid19_italy_eu
    table_id: national_trends
    description: "COVID-19 Italy National Trends"

dag:
  airflow_version: 2
  initialize:
    dag_id: national_trends
    default_args:
      owner: "Google"
      depends_on_past: False
      start_date: '2022-10-03'
    max_active_runs: 1
    schedule_interval: "@daily"
    catchup: False
    default_view: graph

  tasks:
    - operator: "KubernetesPodOperator"
      description: "Run CSV transform within kubernetes pod"
      args:
        task_id: "national_trends_transform_csv"
        startup_timeout_seconds: 600
        name: "covid19_italy_national_trends"
        namespace: "composer-user-workloads"
        service_account_name: "default"
        config_file: "/home/airflow/composer_kube_config"
        image_pull_policy: "Always"
        image: "{{ var.json.covid19_italy.container_registry.run_csv_transform_kub }}"
        env_vars:
          SOURCE_URL: "https://raw.githubusercontent.com/pcm-dpc/COVID-19/master/dati-andamento-nazionale/dpc-covid19-ita-andamento-nazionale.csv"
          SOURCE_FILE: "files/data.csv"
          TARGET_FILE: "files/data_output.csv"
          TARGET_GCS_BUCKET: "{{ var.value.composer_bucket }}"
          TARGET_GCS_PATH: "data/covid19_italy/national_trends/data_output.csv"
          CSV_HEADERS: >-
           ["date","country","hospitalized_patients_symptoms","hospitalized_patients_intensive_care","total_hospitalized_patients","home_confinement_cases","total_current_confirmed_cases","new_current_confirmed_cases","new_total_confirmed_cases","recovered","deaths","total_confirmed_cases","tests_performed","note"]
          RENAME_MAPPINGS: >-
           {"data": "date","stato": "country","ricoverati_con_sintomi": "hospitalized_patients_symptoms","terapia_intensiva": "hospitalized_patients_intensive_care","totale_ospedalizzati": "total_hospitalized_patients","isolamento_domiciliare": "home_confinement_cases","totale_positivi": "total_current_confirmed_cases","variazione_totale_positivi": "new_current_confirmed_cases","nuovi_positivi": "new_total_confirmed_cases","dimessi_guariti": "recovered","deceduti": "deaths","totale_casi": "total_confirmed_cases","tamponi": "tests_performed","note": "note"}
          PIPELINE_NAME: "national_trends"
        container_resources:
          memory:
            request: "32Gi"
          cpu:
            request: "2"
          ephemeral-storage:
            request: "10Gi"

    - operator: "GoogleCloudStorageToBigQueryOperator"
      description: "Task to load CSV data to a BigQuery table"
      args:
        task_id: "load_national_trends_to_bq"
        bucket: "{{ var.value.composer_bucket }}"
        source_objects: ["data/covid19_italy/national_trends/data_output.csv"]
        source_format: "CSV"
        destination_project_dataset_table: "covid19_italy.national_trends"
        skip_leading_rows: 1
        write_disposition: "WRITE_TRUNCATE"
        schema_fields:
          - name: "date"
            type: "TIMESTAMP"
            mode: "NULLABLE"
          - name: "country"
            type: "STRING"
            mode: "NULLABLE"
          - name: "hospitalized_patients_symptoms"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "hospitalized_patients_intensive_care"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "total_hospitalized_patients"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "home_confinement_cases"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "total_current_confirmed_cases"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "new_current_confirmed_cases"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "new_total_confirmed_cases"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "recovered"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "deaths"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "total_confirmed_cases"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "tests_performed"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "note"
            type: "STRING"
            mode: "NULLABLE"

    - operator: "BashOperator"
      description: "Task to copy bq uploadable data file to bucket in EU"
      args:
        task_id: "copy_data_file_EU"
        bash_command: "gsutil cp gs://{{ var.value.composer_bucket }}/data/covid19_italy/national_trends/data_output.csv {{ var.json.covid19_italy.destination_gcs_national_trends }}"

    - operator: "GoogleCloudStorageToBigQueryOperator"
      description: "Task to load CSV data to a BigQuery table"
      args:
        task_id: "load_national_trends_to_bq_eu"
        bucket: "{{ var.json.covid19_italy.destination_gcs_bucket }}"
        source_objects: national/data_output.csv
        source_format: "CSV"
        destination_project_dataset_table: "covid19_italy_eu.national_trends"
        skip_leading_rows: 1
        write_disposition: "WRITE_TRUNCATE"
        schema_fields:
          - name: "date"
            type: "TIMESTAMP"
            mode: "NULLABLE"
          - name: "country"
            type: "STRING"
            mode: "NULLABLE"
          - name: "hospitalized_patients_symptoms"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "hospitalized_patients_intensive_care"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "total_hospitalized_patients"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "home_confinement_cases"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "total_current_confirmed_cases"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "new_current_confirmed_cases"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "new_total_confirmed_cases"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "recovered"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "deaths"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "total_confirmed_cases"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "tests_performed"
            type: "INTEGER"
            mode: "NULLABLE"
          - name: "note"
            type: "STRING"
            mode: "NULLABLE"

  graph_paths:
    - "national_trends_transform_csv >> copy_data_file_EU >> [load_national_trends_to_bq, load_national_trends_to_bq_eu]"
