use setup.sh for env

kbase · Jun 10, 2024 · d6e9f1c · d6e9f1c
1 parent 03e2d69
commit d6e9f1c
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 11 deletions.
diff --git a/scripts/entrypoint.sh b/scripts/entrypoint.sh
@@ -1,6 +1,6 @@
 #!/bin/bash
 
-/opt/scripts/setup.sh
+. /opt/scripts/setup.sh
 
 if [ "$SPARK_MODE" = "notebook" ]; then
     exec /opt/scripts/notebook_entrypoint.sh "$@"

diff --git a/scripts/notebook_entrypoint.sh b/scripts/notebook_entrypoint.sh
@@ -2,16 +2,11 @@
 
 echo "starting jupyter notebook"
 
-lowercase_usage_mode=${USAGE_MODE,,}
-
-case "$lowercase_usage_mode" in
-    dev)
-        export NOTEBOOK_DIR="$CDM_SHARED_DIR"
-        ;;
-    *)
-        export NOTEBOOK_DIR="$CDM_SHARED_DIR/user_shared_workspace"
-        ;;
-esac
+# Ensure NOTEBOOK_DIR is set
+if [ -z "$NOTEBOOK_DIR" ]; then
+    echo "ERROR: NOTEBOOK_DIR is not set. Please run setup.sh first."
+    exit 1
+fi
 
 mkdir -p "$NOTEBOOK_DIR" && cd "$NOTEBOOK_DIR"
 

diff --git a/scripts/setup.sh b/scripts/setup.sh
@@ -45,9 +45,11 @@ set_environment() {
     case "$lowercase_usage_mode" in
         dev)
             export DATANUCLEUS_AUTO_CREATE_TABLES=true
+            export NOTEBOOK_DIR="$CDM_SHARED_DIR"
             ;;
         *)
             export DATANUCLEUS_AUTO_CREATE_TABLES=false
+            export NOTEBOOK_DIR="$CDM_SHARED_DIR/user_shared_workspace"
             ;;
     esac
     update_config