diff --git a/sh/start.sh b/sh/start.sh index 6f12440..c01c29b 100644 --- a/sh/start.sh +++ b/sh/start.sh @@ -1,6 +1,7 @@ #!/bin/bash INDEX=$1 -PUBLIC_DIR="${2:-/shared}" +spider=$2 +PUBLIC_DIR="${3:-/shared}" username="user$INDEX" export DISPLAY=:$INDEX echo "当前显示桌面:$DISPLAY" @@ -14,7 +15,7 @@ sleep 3 start_spider(){ cd $PUBLIC_DIR/learn-spider - sudo -E ./start_spider.sh # -E 保留当前环境变量 + sudo -E ./start_spider.sh $spider # -E 保留当前环境变量 } echo "启动爬虫" diff --git a/start_spider.sh b/start_spider.sh index 3b66a6d..049bf4c 100644 --- a/start_spider.sh +++ b/start_spider.sh @@ -1,7 +1,13 @@ #!/bin/bash spider=$1 -if [ -z "$spider" ]; then - echo "Usage: $0 " - exit 1 + +if [ "$spider" = "01" ]; then + shell_name="china_net.py" +elif [ "$spider" = "02" ]; then + shell_name="mail_qq.py" +else + echo "爬虫名称错误" + exit -1 fi -#.venv/bin/python3.12 spider/china_net.py \ No newline at end of file + +.venv/bin/python3.12 "spider/$shell_name" \ No newline at end of file