vps交流
当前位置：嘟嘟社区 > bbs > 正文

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

2024-07-04 分类：bbs

脚本是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
  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"
  11. # 记录日志的函数
  12. log() {
  13.     echo "$(date ‘+%Y-%m-%d %H:%M:%S’) – $1" >> $LOG_FILE
  14. }
  16. # 记录带时间戳的日志到 BAIDUPCS_GO_LOG
  17. log_to_baidupcs_log() {
  18.     echo "$(date +"%Y-%m-%d %H:%M:%S") – $1" >> "$BAIDUPCS_GO_LOG"
  19. }
  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. }
  30. # 添加任务到队列的函数
  31. add_to_queue() {
  32.     local task="$1"
  33.     local path="$2"
  34.     echo ""${task}" "${path}"" >> $QUEUE_FILE
  35. }
  37. # 读取队列并缓存的函数
  38. load_cache() {
  39.     if [ -f $QUEUE_FILE ]; then
  40.         cat $QUEUE_FILE > $CACHE_FILE
  41.         > $QUEUE_FILE
  42.     fi
  43. }
  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. }
  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. }
  92. # 主函数
  93. main() {
  94.     if [ $# -ne 2 ]; then
  95.         echo "Usage: $0 <upload_task> <dest_path>"
  96.         exit 1
  97.     fi
  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
  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
  118.     process_queue
  120.     release_lock
  121.     if [ -f $QUEUE_FILE ]; then
  122.         rm -f $QUEUE_FILE
  123.     fi
  124. }
  126. release_lock() {
  127.     flock -u 200
  128.     rm -f $LOCK_FILE
  129. }
  131. main "$@"

复制代码
头像也是ai制作吗
谢谢分享，没准哪天就用上了呢 让gpt写了个qbittorrent下载完成后自动上传百度网盘的脚本
Claude写可以用，GPT4o写就需要掂量掂量了