让gpt写了个qbittorrent下载完成后自动上传百度网盘的脚本

aoki

脚本是gpt-4o写的，虽然还有优化空间，但整体可用，我也就没再折腾。
有闲情的大佬可以再优化下。

具体功能是：
qbittorrent下载完成后调用BaiduPCS-Go上传到百度网盘，之前论坛里有rclone上传的脚本，不过微软网盘给的小气。还是百度就很大，就是会吃某些视频，懂得都懂，至于存什么。哎嘿~。

目前脚本已知bug：
1.会出现某些任务路径非法的提示，目测是BaiduPCS-Go的问题，暂时不知道怎么解决。
2.日志文件有点乱，反正能看。
3.没有上传完成后自动删除功能。（因为可能会因为上传出现上传失败的问题，所以种子留下了，等待人工效验。反正我是会大概看一下有没有上传失败的。）


运行命令是bash /downloads/up.sh "%F"  "<目标文件夹>"
使用方法：
上传脚本然后
qbittorrent设置-》启用“torrent 完成时运行外部程序”->输入“bash /downloads/up.sh "%F"  "<目标文件夹>"”->保存


<目标文件夹>替换为实际路径，注意用双引号保持数据完整性。

其他的，代码里面注释很多，自己看着改应该没问题吧？

1. #!/bin/bash
   
2. 
   
3. # 定义一些全局变量
   
4. LOCK_FILE="/downloads/upload_script.lock"
   
5. QUEUE_FILE="/downloads/upload_queue.txt"
   
6. LOG_FILE="/downloads/upload_log.txt"
   
7. ERROR_LOG_FILE="/downloads/upload_error_log.txt"
   
8. CACHE_FILE="/downloads/upload_cache.txt"
   
9. BAIDUPCS_GO_LOG="/downloads/BAIDUPCS-GO_LOG.txt"
   
10. 
    
11. # 记录日志的函数
    
12. log() {
    
13.     echo "$(date '+%Y-%m-%d %H:%M:%S') - $1" >> $LOG_FILE
    
14. }
    
15. 
    
16. # 记录带时间戳的日志到 BAIDUPCS_GO_LOG
    
17. log_to_baidupcs_log() {
    
18.     echo "$(date +"%Y-%m-%d %H:%M:%S") - $1" >> "$BAIDUPCS_GO_LOG"
    
19. }
    
20. 
    
21. # 记录错误日志的函数
    
22. error_log() {
    
23.     local error_message="$1"
    
24.     local error_task="$2"
    
25.     local error_step="$3"
    
26.     local error_action="$4"
    
27.     echo "$(date '+%Y-%m-%d %H:%M:%S') - Task: ${error_task}, Step: ${error_step}, Error: ${error_message}, Action: ${error_action}" >> $ERROR_LOG_FILE
    
28. }
    
29. 
    
30. # 添加任务到队列的函数
    
31. add_to_queue() {
    
32.     local task="$1"
    
33.     local path="$2"
    
34.     echo ""${task}" "${path}"" >> $QUEUE_FILE
    
35. }
    
36. 
    
37. # 读取队列并缓存的函数
    
38. load_cache() {
    
39.     if [ -f $QUEUE_FILE ]; then
    
40.         cat $QUEUE_FILE > $CACHE_FILE
    
41.         > $QUEUE_FILE
    
42.     fi
    
43. }
    
44. 
    
45. # 执行上传任务的函数
    
46. execute_tasks() {
    
47.     while [ -s $CACHE_FILE ]; do
    
48.         local line=$(head -n 1 $CACHE_FILE)
    
49.         local task=$(echo $line | awk -F" '{print $2}')
    
50.         local dest_path=$(echo $line | awk -F" '{print $4}')
    
51.         
    
52.         log "开始上传 ${task} 到网盘 ${dest_path} 目录"
    
53.         
    
54.         # 执行上传任务 并将运行结果记录到BAIDUPCS-GO_LOG.txt当中
    
55.         # /downloads/BaiduPCS-Go 为BaiduPCS-Go的绝对路径
    
56.         /downloads/BaiduPCS-Go upload "${task}" "${dest_path}" 2>&1 | while IFS= read -r line; do log_to_baidupcs_log "$line"; done
    
57.         
    
58.         
    
59.         #这里只记录BaiduPCS-Go最后一次的运行结果。
    
60.         #所以对于文件夹，也是记录最后一个上传信息，详细的上传日志还得去BAIDUPCS_GO_LOG看
    
61.         # 检查日志文件中是否包含“文件路径含有非法字符，已跳过!”
    
62.         if grep -q "文件路径含有非法字符，已跳过!" "$BAIDUPCS_GO_LOG"; then
    
63.             local skip_message=$(grep "文件路径含有非法字符，已跳过!" "$BAIDUPCS_GO_LOG" | tail -n 1)
    
64.             log "$skip_message"
    
65.         fi
    
66.         
    
67.         # 检查日志文件中是否包含上传成功的提示信息
    
68.         if grep -q "上传文件成功, 保存到网盘路径:" "$BAIDUPCS_GO_LOG"; then
    
69.             local success_message=$(grep "上传文件成功, 保存到网盘路径:" "$BAIDUPCS_GO_LOG" | tail -n 1)
    
70.             log "$success_message"
    
71.         fi
    
72.         
    
73.         log "${task} 上传完成"
    
74.         
    
75.         # 移除已完成任务
    
76.         sed -i '1d' $CACHE_FILE
    
77.     done
    
78. }
    
79. 
    
80. # 处理队列的函数
    
81. process_queue() {
    
82.     load_cache
    
83.    
    
84.     while [ -s $CACHE_FILE ]; do
    
85.         execute_tasks
    
86.         load_cache
    
87.     done
    
88.    
    
89.     rm -f $CACHE_FILE
    
90. }
    
91. 
    
92. # 主函数
    
93. main() {
    
94.     if [ $# -ne 2 ]; then
    
95.         echo "Usage: $0 <upload_task> <dest_path>"
    
96.         exit 1
    
97.     fi
    
98. 
    
99.     local task="$1"
    
100.     local dest_path="$2"
     
101.    
     
102.     # 尝试获取锁
     
103.     exec 200>$LOCK_FILE
     
104.     flock -n 200
     
105.     if [ $? -ne 0 ]; then
     
106.         error_log "Failed to acquire lock" "$task" "Acquiring lock" "Adding task to queue"
     
107.         add_to_queue "$task" "$dest_path"
     
108.         exit 0
     
109.     fi
     
110. 
     
111.     # 如果获取到了锁，处理任务
     
112.     if [ -f $QUEUE_FILE ]; then
     
113.         add_to_queue "$task" "$dest_path"
     
114.     else
     
115.         echo ""${task}" "${dest_path}"" > $QUEUE_FILE
     
116.     fi
     
117. 
     
118.     process_queue
     
119. 
     
120.     release_lock
     
121.     if [ -f $QUEUE_FILE ]; then
     
122.         rm -f $QUEUE_FILE
     
123.     fi
     
124. }
     
125. 
     
126. release_lock() {
     
127.     flock -u 200
     
128.     rm -f $LOCK_FILE
     
129. }
     
130. 
     
131. main "$@"
     

复制代码

一笑奈何

头像也是ai制作吗

arlodaddy

谢谢分享，没准哪天就用上了呢

Favorite3000

Claude写可以用，GPT4o写就需要掂量掂量了

aoki

一笑奈何 发表于 2024-7-3 23:57
头像也是ai制作吗

头像是暴机少女

van

有下载Baidu盘的api吗

aoki

Favorite3000 发表于 2024-7-4 00:43
Claude写可以用，GPT4o写就需要掂量掂量了

脚本测试可用的。 至少我几台vps上稳定运行