Added progress bar to src/batch_download.sh

biobricks-ai · Feb 25, 2025 · 39e1679 · 39e1679
1 parent b113c1d
commit 39e1679
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/batch_download.sh b/src/batch_download.sh
@@ -75,6 +75,9 @@ contents=$(cat $listfile)
 # see https://stackoverflow.com/questions/918886/how-do-i-split-a-string-on-a-delimiter-in-bash#tab-top
 IFS=',' read -ra tokens <<< "$contents"
 
+total=${#tokens[@]}
+count=0
+
 for token in "${tokens[@]}"
 do
   if [ "$cif" == true ]
@@ -110,6 +113,12 @@ do
     download ${token}_mr.str.gz $outdir
   fi
 
+  ((count++))
+  progress=$((count * 100 / total))
+  bar=$(printf "#%.0s" $(seq 1 $((progress / 2))))
+  spaces=$(printf " %.0s" $(seq 1 $((50 - ${#bar}))))
+  echo -ne "\r[$bar$spaces] $progress%"
+
 done
 
 

diff --git a/stages/01_download.sh b/stages/01_download.sh
@@ -11,7 +11,7 @@ FILE=$srcpath/pdb_ids.txt
 BATCH_SCRIPT=$srcpath/batch_download.sh
 
 # Ensure batch_download.sh is present
-if [ ! -f BATCH_SCRIPT ]; then
+if [ ! -f "$BATCH_SCRIPT" ]; then
     echo "Error: batch_download.sh not found. Please place it in the src directory." >&2
     exit 1
 fi