Running query

#!/bin/sh
#
# td command option
# -d : database name
# -w : waiting
# --output : output file name to local
# -f : file format
# -c : add column name (top line)

mode=$1

while read table
do
  echo "$table"
  
  if [ $mode  = "add_header" ]; then
    td query -d sample_datasets 'select count(1) from $table' -w -T presto --output test.csv -f csv -c
    mv test.csv test_tmp.csv
    cat '$table'.csv test_tmp.csv >> '$table'.csv
    rm test_tmp.csv

  else
    td query -d sample_datasets 'select count(1) from $table' -w -T presto --output test.csv -f csv 
    mv test.csv test_tmp.csv
    cat test_tmp.csv > '$table'.csv
    rm test_tmp.csv
  fi
  
done < table_list.txt