#!/bin/bash

echo "* 安装相关程序"
sudo apt install python3-pip
pip3 install requests mysql-connector selenium pandas numpy Pillow xlrd xlwt

echo "* 生成配置文件"
if [ ! -f config.cfg ]; then
   cp config.cfg.module config.cfg
fi
echo "* 为相关程序添加执行权限"
chmod u+x driver/chromedriver get_ipnet.py get_wangyi.py one_key.py
echo "* 注册定时任务，网驿每15分钟执行一次，IPNET每小时执行一次"
crontab -l > /dev/null 2>&1
if [ $? = '1' ];then
	echo "10 * * * * export DISPLAY=:0.0 && python3 `pwd`/get_ipnet.py >> /tmp/sjsdata.log 2>&1" > conf 
	echo "*/15 * * * * export DISPLAY=:0.0 && python3 `pwd`/get_wangyi.py >> /tmp/sjsdata.log 2>&1" >> conf
	echo "*/15 * * * * export DISPLAY=:0.0 && python3 `pwd`/get_wangguan.py >> /tmp/sjsdata.log 2>&1" >> conf && crontab conf && rm -f conf
else
	crontab -l > conf && echo "10 * * * * export DISPLAY=:0.0 && python3 `pwd`/get_ipnet.py >> /tmp/sjsdata.log 2>&1" >> conf
	echo "*/15 * * * * export DISPLAY=:0.0 && python3 `pwd`/get_wangyi.py >> /tmp/sjsdata.log 2>&1" >> conf 
	echo "*/15 * * * * export DISPLAY=:0.0 && python3 `pwd`/get_wangguan.py >> /tmp/sjsdata.log 2>&1" >> conf && crontab conf && rm -f conf
fi