add args

.github/workflows/baidupush.yml

@@ -29,4 +29,4 @@ jobs:
         run: pip install requests
 
       - name: parse sitemap
-        run: python baidupush.py
+        run: npm run baidupush

baidupush.py

@@ -1,11 +1,18 @@
 import re
 import ssl
 import requests
-ssl._create_default_https_context = ssl._create_unverified_context
-url = 'https://www.ghlerrix.cn/sitemap.xml'
-result = requests.get(url)
-big = re.findall('<loc>(.*?)</loc>', result.content.decode('utf-8'), re.S)
-for i in big:
-    print(i)
-    # op_xml_txt = open('xml.txt', 'a')
-    # op_xml_txt.write('%s\n' % i)
+import argparse
+
+
+if __name__ == '__main__':
+    ssl._create_default_https_context = ssl._create_unverified_context
+    parser = argparse.ArgumentParser(description='parse sitemap')
+    parser.add_argument('url', help='The url of your website')
+    args = parser.parse_args()
+    url = f'{args.url}/sitemap.xml'
+    result = requests.get(url)
+    big = re.findall('<loc>(.*?)</loc>', result.content.decode('utf-8'), re.S)
+    for i in big:
+        print(i)
+        # op_xml_txt = open('xml.txt', 'a')
+        # op_xml_txt.write('%s\n' % i)

baidupush.sh

@@ -0,0 +1,16 @@
+#!/usr/bin/env sh
+
+# 确保脚本抛出遇到的错误
+set -e
+
+# 检查是否传入了正确的参数
+if [ $# -lt 1 ]; then
+  echo "Please provide the your website as an argument."
+  exit 1
+fi
+
+# 百度链接推送
+python baidupush.py $1
+# curl -H 'Content-Type:text/plain' --data-binary @urls.txt "http://data.zz.baidu.com/urls?site=https://vuepress.ghlerrix.cn&token=oUldnU4HZvSTlh0e"
+
+rm -rf urls.txt # 删除文件

package.json

@@ -18,7 +18,8 @@
     "start": "next start",
     "post-build": "next-sitemap --config next-sitemap.config.js",
     "export": "next build && next-sitemap --config next-sitemap.config.js && next export",
-    "bundle-report": "ANALYZE=true yarn build"
+    "bundle-report": "ANALYZE=true yarn build",
+    "baidupush": "bash baiduPush.sh 'www.ghlerrix.cn'"
   },
   "dependencies": {
     "@giscus/react": "^2.2.6",