Airflow KubernetesPodOperator

kubernetes_pod_operator.py

import time
from datetime import datetime

from airflow.decorators import dag

from airflow.providers.cncf.kubernetes.operators.kubernetes_pod import (
    KubernetesPodOperator,
)

doc = """
# KubernetesPodOperator

"""


@dag(
    start_date=datetime(2021, 12, 22, tzinfo=pendulum.timezone("Asia/Seoul")),
    schedule_interval=None,
    catchup=False,
    doc_md=doc,
    tags=["k8s"],
)
def kubernetes_pod_operator():
    
    image_uri = "docker_image_uri"

    KubernetesPodOperator(
        name="k8s_pod_operator",
        namespace="namespace",
        image=image_uri,
        cmds=["sh"],
        arguments=["apply.sh"],
        do_xcom_push=True,
        is_delete_operator_pod=True,
        in_cluster=True,
        task_id="k8s_pod_operator",
        get_logs=True,
    )


dag = kubernetes_pod_operator()