Python 오퍼레이터 기본

우상욱·2024년 3월 23일

Airflow Master Class

목록 보기
8/24

파이썬 오퍼레이터


  • 라이브러리 가져오기
    : from airflow.operators.python import PythonOperator
    : from 패키지명 impor 오퍼레이터명(클래스)
  • 정의된 파이썬 함수를 실행시키는 오퍼레이터
  • 가장 많이 쓰이는 오퍼레이터

파이썬 모듈에는 어떤 오퍼레이터가 있나?


Dag 예제


from airflow import DAG
import datetime
import pendulum
from airflow.operators.python import PythonOperator
import random

with DAG(
    dag_id="dags_python_operator",
    schedule="30 6 * * *",
    start_date=pendulum.datetime(2023, 3, 1, tz="Asia/Seoul"),
    catchup=False,
) as dag:

    def select_fruit():
        fruit = ["APPLE", "BANANA", "ORANGE", "AVOCADO"]
        rand_int = random.randint(0, 3)
        print(fruit[rand_int])

    py_t1 = PythonOperator(
        task_id="pyt1",
        python_callable=select_fruit,  # 실행할 함수 입력
    )
    py_t1
profile
데이터엔지니어

0개의 댓글