Update to use entrypoint

Signed-off-by: Jacob Woffenden <[email protected]>

Author: jacobwoffenden

Dockerfile

@@ -1,9 +1,9 @@
 FROM ghcr.io/ministryofjustice/analytical-platform-airflow-python-base:1.6.0
 
 COPY requirements.txt requirements.txt
-COPY src/main.py main.py
+COPY src/ .
 RUN <<EOF
 pip install --no-cache-dir --requirement requirements.txt
 EOF
 
-ENTRYPOINT ["python3", "main.py"]
+ENTRYPOINT ["bash", "entrypoint.sh"]

src/entrypoint.sh

@@ -0,0 +1,21 @@
+#!/usr/bin/env bash
+
+MODE="${MODE:-basic}"
+
+case "${MODE}" in
+  basic)
+    echo "Running in basic mode"
+    env
+    aws sts get-caller-identity
+    ;;
+  benchmark)
+    echo "Running in benchmark mode"
+    env
+    aws sts get-caller-identity
+    python3 main.py
+    ;;
+  *)
+    echo "Unknown mode: ${MODE}"
+    exit 1
+    ;;
+esac