siruku6/fx_alarm_py

#!/bin/bash
menu=$(cat << EOS
/ / / / / / / / / / / / / / / /
/     Dev Assistance Menu     /
/ / / / / / / / / / / / / / / /
Select an operation from below.
{1} :pipenv run python src/handlers/trade_hist.py
{2} :execute beta_cord
{5} :find (find ./[dir] -type f -print | xargs grep [str])
{6} :pytest -vv

{10}:deploy the function 'tradehist' only
{11}:make the zip for being uploaded to Lambda
{12}:upload the zip of pip packages for Lambda Layer
{80}:adjust date and hwclock
{99}:LINUX shutdown
{*} :exit
Select a number：
EOS
)

# - - - Functions - - -
wait_display () {
  echo press Enter ...
  read Wait
}

search_menu () {
  echo -n '検索対象ディレクトリを入力してね：'
  read dir
  echo -n '検索したい文字列は？：'
  read str
  find ./${dir} -type f -print | xargs grep -n ${str}
}

# TODO: アップロードしたいものだけアップロードできるように修正する
# 多分、現状はいらないものがアップロードされてる（気がする）
make_zip_for_lambda () {
  echo -e 'Input 1(copy only source) or 10(prepare source & modules):'
  read select
  DirName='tmp/zips/app'
  ModuleDirName='tmp/zips/layer_module'
  # Clean up directory
  result=`find ${DirName} -maxdepth 1 -name "*.py" 2>/dev/null`
  if [ -n "$result" ]; then
    yes | rm -r ${DirName}/src
    # yes | rm ${DirName}/main.py
  fi

  # Install modules
  if test $select = 10; then
    result=`find ${ModuleDirName}/python/ -maxdepth 1 -name "*.py" 2>/dev/null`
    if [ -n "$result" ]; then
      yes | rm -r ${ModuleDirName}/python/*
    fi
    pip install -t ${ModuleDirName}/python -r serverless_resources/requirements.txt
    # INFO: .dist-info, __pycache__ are unnecessary on Lambda
    # https://medium.com/@korniichuk/lambda-with-pandas-fd81aa2ff25e
    rm -r ${ModuleDirName}/python/*.dist-info
    rm -r ${ModuleDirName}/python/*/__pycache__
    rm -r ${ModuleDirName}/python/*/tests
    rm -r ${ModuleDirName}/python/__pycache__
    rm ${ModuleDirName}/python/THIRD-PARTY-LICENSES
  fi

  cp -r src ${DirName}/

  # Create Archive
  echo -e 'Make zip now? y(yes) n(no):'
  read select2
  if test $select2 = 'y'; then
    cd ${DirName}
    # INFO: remove *.zip (which is created previously)
    result=`find . -maxdepth 1 -name "*.zip" 2>/dev/null`
    if [ -n "$result" ]; then
      rm ./*.zip
    fi
    zip fx_archive -r ./*
    cd -

    if test $select = 10; then
      cd ${ModuleDirName}
      # INFO: remove *.zip (which is created previously)
      result=`find . -maxdepth 1 -name "*.zip" 2>/dev/null`
      if [ -n "$result" ]; then
        rm ./*.zip
      fi
      zip fx_module_archive -r ./python
      cd -
    fi
  fi
}

upload_zip_to_s3 () {
  # Upload Archive Zip
  echo -e 'Upload zip now? y(yes) n(no):'
  read select
  if test $select = 'y'; then
    DirName='tmp/zips/app'
    cd ${DirName}
    pwd

    aws s3 cp ./fx_archive.zip s3://fx-trade-with-lambda --storage-class ONEZONE_IA
  fi
}

upload_package_zip_to_s3 () {
  # Upload Archive Zip
  echo -e 'Upload zip now? y(yes) n(no):'
  read select
  if test $select = 'y'; then
    ModuleDirName='tmp/zips/layer_module'
    cd ${ModuleDirName}
    pwd

    aws s3 cp ./fx_module_archive.zip s3://fx-trade-with-lambda --storage-class ONEZONE_IA
  fi
}

adjust_clock () {
  sudo ntpdate -v ntp.nict.jp
  sudo hwclock --systoh
  sudo timedatectl

}

shutdown_menu () {
  echo -e 'Input 1(DO SHUTDOWN) or 9(cancel):'
  read select
  case $select in
    1)
      sudo shutdown -h now;;
    9)
      ;;
    *)
      echo 'シャットダウンをキャンセルしました'
      ;;
  esac
}

# - - - Main - - -
clear
while true; do
  echo -e -n "$menu"
  read select

  case $select in
    1)
      echo 'running src/handlers/trade_hist.py'
      pipenv run python src/handlers/trade_hist.py
      wait_display
      ;;
    2)
      echo 'running beta_cord.py'
      pipenv run python beta_cord.py
      wait_display
      ;;
    5)
      search_menu
      wait_display
      ;;
    6)
      echo 'running pytest ...'
      pipenv run pytest -vv
      wait_display
      ;;
    10)
      sls deploy function -f tradehist
      wait_display
      ;;
    11)
      make_zip_for_lambda
      upload_zip_to_s3
      wait_display
      ;;
    12)
      upload_package_zip_to_s3
      wait_display
      ;;
    80)
      adjust_clock
      wait_display
      ;;
    99)
      shutdown_menu
      ;;
    *)
      exit ;;
  esac
done