Airflow Variable 설정

목차

1. Airflow Variable
   • 1) Variable 설정(UI)
   • 2) Variable 설정(Code)
   • 3) Variable 사용

Airflow Variable

Airflow에서 Variable을 설정하는 방법에 대해 알아보도록 하겠습니다.
Variable은 전역적으로 사용할 수 있는 값을 미리 정의해 두고 사용할 수 있습니다.

1) Variable 설정(UI)

1. 상단 메뉴에서 Admin > Variables를 클릭
   
2. 파란색 + 버튼 클릭 (Add a new record) 클릭
   
3. Variable의 키, 값 입력 후 저장
   값에는 text, list, json 등 다양한 자료형의 값을 입력할 수 있습니다.
   
4. 설정된 값 확인

2) Variable 설정(Code)

from airflow.models import Variable


a = Variable.set("key2", "value2")

이러한 형태로 Variable을 설정할 수 있습니다.

3) Variable 사용

# 기본 호출 방법
value = Variable.get("key")

# 호출 시 default 값 설정
value = Variable.get("key", default_var="value")

# json 형태의 값 호출
value = Variable.get("json_key", deserialize_json=True)

json 형태의 값이 아닌 경우에는 deserialize_json 옵션을 넣으면 오류가 발생한다.

---

다음 글로는 airflow Variables을 이용한 동적 DAG 생성 및 dependency 설정에 대해 알아보겠습니다.