biliup → AstrBot(QQ) 录播完成通知 修复

1
#!/usr/bin/env python3
2
"""
3
watch_backup.py — 录播备份目录监听通知
4

5
当 biliup 等录制工具把完成的媒体文件 mv 到监控目录后，
6
本脚本会检测新文件（稳定 N 秒后）并推送到 AstrBot 的 push_lite 插件，
7
进而发到指定 QQ 群 / 私聊。
8

9
配置全部通过环境变量注入（不要把 token 写死进仓库）：
10
  PUSH_LITE_URL    推送地址，默认 http://127.0.0.1:9966/send
11
  PUSH_LITE_TOKEN  push_lite 的 Bearer token（必填）
12
  PUSH_LITE_UMO    目标会话，如 ai:GroupMessage:123456
13
  WATCH_DIR        监控目录，如 /opt/biliup/backup
14
  STATE_FILE       去重状态文件，默认 <脚本目录>/watched_state.json
15
  LOCK_FILE        单实例锁文件，默认 <脚本目录>/watch.lock
16
  POLL             轮询间隔秒，默认 30
17
  STABLE_SECS      文件需稳定多少秒才认为写完，默认 5
18

19
依赖：仅 Python 3 标准库，无需 pip install。
20
"""
21
import os
22
import sys
23
import json
24
import time
25
import glob
26
import fcntl
27
import urllib.request
28

29
URL = os.environ.get("PUSH_LITE_URL", "http://127.0.0.1:9966/send")
30
TOKEN = os.environ.get("PUSH_LITE_TOKEN", "")
31
UMO = os.environ.get("PUSH_LITE_UMO", "ai:GroupMessage:000000")
32
WATCH_DIR = os.environ.get("WATCH_DIR", "/opt/biliup/backup")
33
_HERE = os.path.dirname(os.path.abspath(__file__))
34
STATE = os.environ.get("STATE_FILE", os.path.join(_HERE, "watched_state.json"))
35
LOCK = os.environ.get("LOCK_FILE", os.path.join(_HERE, "watch.lock"))
36
POLL = int(os.environ.get("POLL", "30"))
37
STABLE_SECS = int(os.environ.get("STABLE_SECS", "5"))
38

39
# 只推送这些媒体扩展名，避免把 download.log 之类当媒体误推
40
MEDIA = {".flv", ".mp4", ".mkv", ".ts", ".mov", ".avi", ".m4v",
41
         ".mp3", ".m4a", ".m4s"}
42

43
_lf = open(LOCK, "w")
44
try:
45
    fcntl.flock(_lf, fcntl.LOCK_EX | fcntl.LOCK_NB)
46
except OSError:
47
    print("another instance running, exit", flush=True)
48
    sys.exit(0)
49

50

51
def load_state():
52
    try:
53
        return json.load(open(STATE))
54
    except Exception:
55
        return {}
56

57

58
def save_state(s):
59
    tmp = STATE + ".tmp"
60
    json.dump(s, open(tmp, "w"), ensure_ascii=False)
61
    os.replace(tmp, STATE)
62

63

64
def notify(fname):
65
    content = "录播已上传完成\n文件：" + fname
66
    payload = json.dumps({"content": content, "umo": UMO}).encode("utf-8")
67
    req = urllib.request.Request(URL, data=payload, method="POST")
68
    req.add_header("Content-Type", "application/json")
69
    req.add_header("Authorization", "Bearer " + TOKEN)
70
    try:
71
        with urllib.request.urlopen(req, timeout=15) as r:
72
            return r.status
73
    except Exception as e:
74
        print("notify failed:", e, flush=True)
75
        return None
76

77

78
def main():
79
    state = load_state()
80
    print("watcher started, watch_dir=%s" % WATCH_DIR, flush=True)
81
    while True:
82
        try:
83
            files = [f for f in glob.glob(os.path.join(WATCH_DIR, "*"))
84
                     if os.path.isfile(f)]
85
            now = time.time()
86
            for f in files:
87
                if os.path.splitext(f)[1].lower() not in MEDIA:
88
                    continue
89
                if now - os.path.getmtime(f) < STABLE_SECS:
90
                    continue
91
                key = os.path.basename(f)
92
                if key in state:
93
                    continue
94
                st = notify(key)
95
                if st == 200:
96
                    state[key] = int(now)
97
                    save_state(state)
98
                    print("notified:", key, flush=True)
99
        except Exception as e:
100
            print("loop err:", e, flush=True)
101
        time.sleep(POLL)
102

103

104
if __name__ == "__main__":
105
    main()