[airflow] 04/07: Parameterized bash/python in the prod image (#9157)

[po...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

potiuk pushed a commit to branch v1-10-test
in repository https://gitbox.apache.org/repos/asf/airflow.git

commit 57225c6fdd49407799663364daa6dad0e6eeb12a
Author: Jarek Potiuk <ja...@polidea.com>
AuthorDate: Sat Jun 6 09:48:02 2020 +0200

    Parameterized bash/python in the prod image (#9157)
    
    (cherry picked from commit b809afa3e33ec5cc76f846f525ab8d6bdcf9d1f0)
---
 entrypoint.sh | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/entrypoint.sh b/entrypoint.sh
index 3d436e2..220e9b7 100755
--- a/entrypoint.sh
+++ b/entrypoint.sh
@@ -102,8 +102,12 @@ if [[ -n ${AIRFLOW__CELERY__BROKER_URL} ]] && \
     verify_db_connection "${AIRFLOW__CELERY__BROKER_URL}"
 fi
 
-if [[ ${AIRFLOW_COMMAND} == "" ]]; then
-   exec "/bin/bash"
+if [[ ${AIRFLOW_COMMAND} == "bash" ]]; then
+   shift
+   exec "/bin/bash" "${@}"
+elif [[ ${AIRFLOW_COMMAND} == "python" ]]; then
+   shift
+   exec "python" "${@}"
 fi
 
 # Run the command