在 Amazon MWAA 上變更 DAG 的時區

為您的自訂外掛程式建立名為 plugins 的目錄，然後導覽至 目錄。例如：

$ mkdir plugins
$ cd plugins

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

import time
import os

os.environ['TZ'] = 'America/Los_Angeles'
time.tzset()

在 plugins目錄中，建立名為 的空 Python 檔案__init__.py。您的plugins目錄應該類似於以下內容：

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

在命令提示中，導覽至上一個步驟的plugins目錄。例如：

cd plugins

壓縮plugins目錄中的內容。

zip -r ../plugins.zip ./

plugins.zip 上傳至 S3 儲存貯體

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

在命令提示中，導覽至存放 DAGs目錄。例如：

$ cd dags

複製下列範例的內容並儲存為 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"
    )

執行下列 AWS CLI 命令，將 DAG 複製到您環境的儲存貯體，然後使用 Apache Airflow UI 觸發 DAG。

$ aws s3 cp your-dag.py s3://your-environment-bucket/dags/

如果成功，您將輸出類似 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