scheduling_main.sh 1.9 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. #!/bin/bash
  2. crawler_dir=$1 # 爬虫执行路径,如: ./youtube/youtube_main/run_youtube_follow.py
  3. log_type=$2 # 日志命名格式,如: follow,则在 youtube/logs/目录下,生成 2023-02-08-follow.log
  4. crawler=$3 # 哪款爬虫,如: youtube / kanyikan / weixinzhishu
  5. env=$4 # 爬虫运行环境,正式环境: prod / 测试环境: dev
  6. #machine=$5 # 爬虫运行机器,阿里云服务器: aliyun_hk / aliyun / macpro / macair / local
  7. nohup_dir=$5 # nohup日志存储路径,如: ./youtube/nohup.log
  8. echo "开始"
  9. #echo ${env}
  10. if [ ${env} = "--env=hk" ];then
  11. piaoquan_crawler_dir=/root/piaoquan_crawler/
  12. profile_path=/etc/profile
  13. python=python3
  14. elif [ ${env} = "--env=prod" ];then
  15. piaoquan_crawler_dir=/root/piaoquan_crawler/
  16. profile_path=/etc/profile
  17. python=/usr/bin/python3
  18. elif [ ${env} = "--env=offline" ];then
  19. piaoquan_crawler_dir=/Users/piaoquan/Desktop/piaoquan_crawler/
  20. profile_path=./base_profile
  21. node_path=/usr/local/bin/node
  22. python=python3
  23. elif [ ${env} = "--env=dev" ];then
  24. piaoquan_crawler_dir=/Users/wangkun/Desktop/crawler/piaoquan_crawler/
  25. profile_path=/etc/profile
  26. node_path=/opt/homebrew/bin/node
  27. python=python3
  28. fi
  29. echo "$(date "+%Y-%m-%d %H:%M:%S") 更新环境变量..."
  30. cd ~ && source ${profile_path}
  31. echo "$(date "+%Y-%m-%d %H:%M:%S") 更新环境变量完成!"
  32. if [ ${env} = "--env=hk" ];then
  33. echo "升级yt-dlp"
  34. pip3 install yt-dlp -U
  35. else
  36. echo "$(date "+%Y-%m-%d %H:%M:%S") 正在更新代码..."
  37. cd ${piaoquan_crawler_dir} && git pull origin master --force && rm -f ${piaoquan_crawler_dir}main/nohup.log && rm -f ${piaoquan_crawler_dir}${nohup_dir}
  38. echo "$(date "+%Y-%m-%d %H:%M:%S") 代码更新完成!"
  39. fi
  40. echo "$(date "+%Y-%m-%d %H:%M:%S") 正在重启服务..."
  41. cd ${piaoquan_crawler_dir}
  42. nohup ${python} -u ${crawler_dir} ${log_type} ${crawler} ${env} >>${nohup_dir} 2>&1 &
  43. echo "$(date "+%Y-%m-%d %H:%M:%S") 服务重启完毕!"
  44. exit 0