각 환경의 매개 변수를 Digdag로 설정합니다.

prd:
  aws:
    rds:
      hostname: rds.jp
      ro_hostname: rds-ro.jp
      user: root
      password: ...

stg:
  aws:
    rds:
      hostname: rds.stg.jp
      ro_hostname: rds-ro.stg.jp
      user: root
      password: ...

import os
import yaml
import digdag


def setup():
    f = open("config.yml")
    data = yaml.load(f)[os.environ["DIGDAG_ENV"]]
    digdag.env.store(data)

+setup:
  py>: tasks.setup

+sample_task:
  _export:
    rds_host: ${aws.rds.hostname}
    rds_user: ${aws.rds.user}
    rds_pass: ${aws.rds.password}
  py>: tasks.rds_connect