slurm_list_jobdir for running jobs with -R, clean jobdir option

Author: Han Lin Mai

utils/actual_usage/clean_jobdir

@@ -0,0 +1 @@
+find . -type f \( -name CHG -o -name CHGCAR* -o -name AECCAR* -o -name WAVECAR* -o -name *.swp* -o -name OUTCAR-* -o -name *core* \) -delete

utils/actual_usage/slurm_list_jobdir

@@ -3,29 +3,43 @@
 # Specify the username
 username="hmai"
 
-# Get the list of job IDs and statuses for the specified user
-job_info=$(squeue -u $username -h -o "%i %t" | awk '{print $1, $2}')
-
-# Initialize a counter for running jobs
-running_job_count=0
-
-# Print the table header
-echo "Job ID   Job Status   Working Directory"
-echo "------   ----------   -----------------"
-
-# Loop through each line of job information
-while read -r job_line; do
-    job_id=$(echo "$job_line" | awk '{print $1}')
-    job_status=$(echo "$job_line" | awk '{print $2}')
-    working_directory=$(scontrol show job $job_id | grep "WorkDir" | awk -F'=' '{print $2}' | tr -d ' ')
-
-    echo "$job_id       $job_status          $working_directory"
-
-    if [ "$job_status" == "R" ]; then
-        ((running_job_count++))
-    fi
-done <<< "$job_info"
-
-echo "--------------------------"
-echo "Number of running jobs for $username: $running_job_count"
-
+if [ "$1" == "-R" ]; then
+    echo "Listing running job directories for user $username:"
+    running_dirs_count=0
+
+    for job in $(squeue -u $username --state=R -h -o %i); do 
+        # Get the working directory for the running job
+        working_directory=$(scontrol show job $job | grep "WorkDir" | awk -F'=' '{print $2}' | tr -d ' ')
+        echo "Job $job: $working_directory"
+        ((running_dirs_count++))
+    done
+
+    echo "--------------------------"
+    echo "Number of running directories for $username: $running_dirs_count"
+else
+    # Get the list of job IDs and statuses for the specified user
+    job_info=$(squeue -u $username -h -o "%i %t" | awk '{print $1, $2}')
+
+    # Initialize a counter for running jobs
+    running_job_count=0
+
+    # Print the table header
+    echo "Job ID   Job Status   Working Directory"
+    echo "------   ----------   -----------------"
+
+    # Loop through each line of job information
+    while read -r job_line; do
+        job_id=$(echo "$job_line" | awk '{print $1}')
+        job_status=$(echo "$job_line" | awk '{print $2}')
+        working_directory=$(scontrol show job $job_id | grep "WorkDir" | awk -F'=' '{print $2}' | tr -d ' ')
+
+        echo "$job_id       $job_status          $working_directory"
+
+        if [ "$job_status" == "R" ]; then
+            ((running_job_count++))
+        fi
+    done <<< "$job_info"
+
+    echo "--------------------------"
+    echo "Number of running jobs for $username: $running_job_count"
+fi