#!/usr/bin/env bash

#安装python调用相关的库文件
function update(){
  sudo apt-get update
  sudo apt-get upgrade -y
  #pip升级更新
  pip install --upgrade pip
  #安装脚本调用库
  pip install configparser
  pip install selenium
  pip install datetime
  pip install chinese_calendar
  return 0
}

#将谷歌浏览器驱动复制到/usr/bin目录下
function mv_chromedriver(){
  cd "$dirPath"/config || exit
  sudo cp chromedriver  /usr/bin
  return 0
}

#安装google浏览器
function chrome_install(){
  #卸载原有的谷歌浏览器
  dpkg --list | grep 'google-chrome-stable' \
  && sudo dpkg --purge google-chrome-stable
  cd "$dirPath"/config || exit
  sudo dpkg -i  google-chrome-stable_current_amd64.deb
  return 0
}

#清除多余的配置文件
function purge_file() {
  cd "$dirPath"/config || exit
  if cat ERPcookies.pickle;then
    rm -rf ERPcookies.pickle
  fi
  if cat userInfo.pickle;then
    rm -rf userInfo.pickle
  fi
}

function main(){
  dirPath=$( pwd )
  update
  mv_chromedriver
  chrome_install
  purge_file
  /usr/bin/python3  "$dirPath"/main.py
  exit 0
}

main