import datetime

from airflow.providers.standard.operators.bash import BashOperator
from airflow.providers.standard.operators.python import PythonOperator
from airflow.sdk import DAG


def fetch_xcom(**kwargs):
    return kwargs["ti"].xcom_pull(key="choice")

def push_xcom(**kwargs):
    kwargs["ti"].xcom_push(key="choice", value="foo")

dag = DAG(
    'test_api_dag',
    start_date=datetime.datetime(2025, 8,25),
    schedule='@daily',
    is_paused_upon_creation=False,
    catchup=True,
)

hello_task = BashOperator(
    task_id="hello",
    bash_command="echo hello",
    dag=dag
)

bye_task = BashOperator(
    task_id="bye",
    bash_command="echo bye",
    dag=dag
)

hello_task >> bye_task