Update gen_pop_data.sh

gen_pop_data.sh

@@ -2,14 +2,14 @@
 
 # Function to display help
 function show_help() {
-    echo "Usage: $0 <host> <user> <password>"
+    echo "Usage: $0 [<host> <user> <password>]"
     echo ""
     echo "This script sets up the database and loads the data."
     echo ""
-    echo "Arguments:"
-    echo "  host       Database host"
-    echo "  user       Database user"
-    echo "  password   Database password"
+    echo "Arguments (optional if environment variables are set):"
+    echo "  host       Database host (or set DB_HOST)"
+    echo "  user       Database user (or set DB_USER)"
+    echo "  password   Database password (or set DB_PASSWORD)"
 }
 
 # Check if help is requested
@@ -18,18 +18,25 @@ if [[ "$1" == "-h" || "$1" == "--help" ]]; then
     exit 0
 fi
 
-# Check if all three parameters are provided
-if [[ $# -ne 3 ]]; then
-    echo "Error: Missing arguments."
-    show_help
-    exit 1
+# Check if all three arguments are provided or fallback to environment variables
+if [[ $# -eq 3 ]]; then
+    HOST=$1
+    USER=$2
+    PASSWORD=$3
+else
+    # Check for environment variables
+    HOST=${DB_HOST}
+    USER=${DB_USER}
+    PASSWORD=${DB_PASSWORD}
+    
+    # If either arguments or environment variables are missing, display help
+    if [[ -z "$HOST" || -z "$USER" || -z "$PASSWORD" ]]; then
+        echo "Error: Missing arguments or environment variables."
+        show_help
+        exit 1
+    fi
 fi
 
-# Assign parameters to variables
-HOST=$1
-USER=$2
-PASSWORD=$3
-
 # create database
 ./inputs/setup_db.sh "$HOST" "$USER" "$PASSWORD"