Airflow 공부 - 7 Python Operator로 Python 함수 실행하기

• from airflow.operators.python import PythonOperator

• PythonOperator
  • 파이썬 함수를 실행시키는 오퍼레이터
• BranchPythonOperator
  • 파이썬 함수 실행 결과에 따라 Task를 선택적으로 실행시킴
• ShortCircuitOperator
  • 파이썬 함수 실행 결과에 따라 후행 Task를 실행하지 않고 종료시킴
• PythonVirtualenvOperator
  • 파이썬 가상환경을 생성하고 Job을 수행함, 마무리 되면 가상환경 삭제
• ExternalPythonOperator
  • 기존에 존재하는 파이썬 가상환경에서 Job을 수행하게 함

with DAG(
    dag_id="dags_python_operator",
    schedule="0 0 * * *", 
    start_date=pendulum.datetime(2023, 3, 1, tz="Asia/Seoul"),
    catchup=False
) as dag:
    
    def select_fruit():
        fruits = ["apple", "banana", "cherry"]
        print(random.choice(fruits))
        return random.choice(fruits)
        
    py_t1 = PythonOperator(
        task_id="py_t1",
        python_callable=select_fruit
    )
        
    py_t1