如何在 Airflow 中运行 bash 脚本文件

Question

我有一个 bash 脚本，它创建了一个我想在 Airflow 中运行的文件（如果它不存在），但是当我尝试时它失败了。 我该怎么做？

#!/bin/bash
#create_file.sh

file=filename.txt

if [ ! -e "$file" ] ; then
    touch "$file"
fi

if [ ! -w "$file" ] ; then
    echo cannot write to $file
    exit 1
fi

这是我在 Airflow 中的调用方式：

create_command = """
 ./scripts/create_file.sh
"""
t1 = BashOperator(
        task_id= 'create_file',
        bash_command=create_command,
        dag=dag
)

lib/python2.7/site-packages/airflow/operators/bash_operator.py", line 83, in execute
    raise AirflowException("Bash command failed")
airflow.exceptions.AirflowException: Bash command failed

Answer 1

从教程这是可以的：

t2 = BashOperator(
    task_id='sleep',
    bash_command='sleep 5',
    retries=3,
    dag=dag)

但是你向它传递了一个多行命令

create_command = """
 ./scripts/create_file.sh
"""

应该是

create_command = "./scripts/create_file.sh "

此外，您还必须确保您位于正确的目录中，以避免出现神秘错误。 例如这样做：

create_command = "./scripts/create_file.sh "
if os.path.exists(create_command):
   t1 = BashOperator(
        task_id= 'create_file',
        bash_command=create_command,
        dag=dag
   )
else:
    raise Exception("Cannot locate {}".format(create_command))

Answer 2

从文档：

t2 = BashOperator(
    task_id='bash_example',
    # "scripts" folder is under "/usr/local/airflow/dags"
    bash_command="scripts/test.sh",
    dag=dag)