在 Amazon MWAA 上變更 DAG 的時區 - Amazon Managed Workflows for Apache Airflow

本文為英文版的機器翻譯版本,如內容有任何歧義或不一致之處,概以英文版為準。

在 Amazon MWAA 上變更 DAG 的時區

根據預設,Apache Airflow 會在 UTC+0 中排程您的導向無環圖 (DAG)。下列步驟說明如何變更 Amazon MWAA 使用 Pendulum 執行 DAGs的時區。或者,本主題示範如何建立自訂外掛程式,以變更環境 Apache Airflow 日誌的時區。

版本

  • 您可以在 Python 3.10 中使用此頁面上的程式碼範例搭配 Apache Airflow v2

先決條件

若要使用此頁面上的範例程式碼,您需要下列項目:

許可

  • 使用此頁面上的程式碼範例不需要額外的許可。

建立外掛程式以變更 Airflow 日誌中的時區

Apache Airflow 會在啟動時在 plugins目錄中執行 Python 檔案。使用下列外掛程式,您可以覆寫執行器的時區,以修改 Apache Airflow 寫入日誌的時區。

  1. 為您的自訂外掛程式建立名為 plugins 的目錄,然後導覽至 目錄。例如:

    $ mkdir plugins $ cd plugins
  2. 複製下列程式碼範例的內容,並在本機儲存為 dag-timezone-plugin.py 資料夾中的 plugins

    import time import os os.environ['TZ'] = 'America/Los_Angeles' time.tzset()
  3. plugins目錄中,建立名為 的空 Python 檔案__init__.py。您的plugins目錄應該類似於以下內容:

    plugins/ |-- __init__.py |-- dag-timezone-plugin.py

建立 plugins.zip

下列步驟說明如何建立 plugins.zip。此範例的內容可以與其他外掛程式和二進位檔合併為單一plugins.zip檔案。

  1. 在命令提示中,導覽至上一個步驟的plugins目錄。例如:

    cd plugins
  2. 壓縮plugins目錄中的內容。

    zip -r ../plugins.zip ./
  3. plugins.zip 上傳至 S3 儲存貯體

    $ aws s3 cp plugins.zip s3://your-mwaa-bucket/

範例程式碼

若要變更 DAG 執行的預設時區 (UTC+0),我們將使用名為 Pendulum 的程式庫,Python 程式庫可用來處理時區感知日期時間。

  1. 在命令提示中,導覽至存放 DAGs目錄。例如:

    $ cd dags
  2. 複製下列範例的內容並儲存為 tz-aware-dag.py

    from airflow import DAG from airflow.operators.bash_operator import BashOperator from datetime import datetime, timedelta # Import the Pendulum library. import pendulum # Instantiate Pendulum and set your timezone. local_tz = pendulum.timezone("America/Los_Angeles") with DAG( dag_id = "tz_test", schedule_interval="0 12 * * *", catchup=False, start_date=datetime(2022, 1, 1, tzinfo=local_tz) ) as dag: bash_operator_task = BashOperator( task_id="tz_aware_task", dag=dag, bash_command="date" )
  3. 執行下列 AWS CLI 命令,將 DAG 複製到您環境的儲存貯體,然後使用 Apache Airflow UI 觸發 DAG。

    $ aws s3 cp your-dag.py s3://your-environment-bucket/dags/
  4. 如果成功,您將輸出類似 tz_test DAG 中 的任務日誌tz_aware_task中的下列項目:

    [2022-08-01, 12:00:00 PDT] {{subprocess.py:74}} INFO - Running command: ['bash', '-c', 'date']
    [2022-08-01, 12:00:00 PDT] {{subprocess.py:85}} INFO - Output:
    [2022-08-01, 12:00:00 PDT] {{subprocess.py:89}} INFO - Mon Aug  1 12:00:00 PDT 2022
    [2022-08-01, 12:00:00 PDT] {{subprocess.py:93}} INFO - Command exited with return code 0
    [2022-08-01, 12:00:00 PDT] {{taskinstance.py:1280}} INFO - Marking task as SUCCESS. dag_id=tz_test, task_id=tz_aware_task, execution_date=20220801T190033, start_date=20220801T190035, end_date=20220801T190035
    [2022-08-01, 12:00:00 PDT] {{local_task_job.py:154}} INFO - Task exited with return code 0
    [2022-08-01, 12:00:00 PDT] {{local_task_job.py:264}} INFO - 0 downstream tasks scheduled from follow-on schedule check
                    

後續步驟?

  • 進一步了解如何在此範例中將plugins.zip檔案上傳至 中的 Amazon S3 儲存貯體安裝自訂外掛程式