107 lines
3.3 KiB
Python
107 lines
3.3 KiB
Python
import hashlib
|
|
import json
|
|
import os
|
|
|
|
INPUT_JSON = "tvbox_config.json"
|
|
OUTPUT_JSON = "tvbox_modified.json"
|
|
LOCAL_SPIDER_FILE = "aowu.png" # 直接引用 download_spider.py 下载好的本地文件
|
|
|
|
JAR_KEY = "jar"
|
|
BASE_JAR_URL = (
|
|
"https://down.nigx.cn/raw.githubusercontent.com/woshishiq1/jiemi/main/aowu.png"
|
|
)
|
|
|
|
|
|
def get_local_file_md5(file_path: str) -> str | None:
|
|
"""直接计算本地下载好的 aowu.png 的 MD5，避免网络拉取失败"""
|
|
if not os.path.exists(file_path):
|
|
print(f"[!] 本地未找到 {file_path}，无法计算 MD5")
|
|
return None
|
|
|
|
try:
|
|
md5_hash = hashlib.md5()
|
|
with open(file_path, "rb") as f:
|
|
for chunk in iter(lambda: f.read(8192), b""):
|
|
md5_hash.update(chunk)
|
|
|
|
calculated_md5 = md5_hash.hexdigest()
|
|
print(f"[+] 本地 {file_path} MD5 计算成功: {calculated_md5}")
|
|
return calculated_md5
|
|
except Exception as e:
|
|
print(f"[!] 计算本地 MD5 失败: {e}")
|
|
return None
|
|
|
|
|
|
def add_jar_to_sites():
|
|
if not os.path.exists(INPUT_JSON):
|
|
print(f"[-] 未找到源文件 {INPUT_JSON}")
|
|
return
|
|
|
|
try:
|
|
with open(INPUT_JSON, "r", encoding="utf-8") as f:
|
|
config_data = json.load(f)
|
|
except Exception as e:
|
|
print(f"[-] 解析 JSON 失败: {e}")
|
|
return
|
|
|
|
sites_list = config_data.get("sites", [])
|
|
if not sites_list or not isinstance(sites_list, list):
|
|
print("[-] 未找到有效的 'sites' 列表。")
|
|
return
|
|
|
|
# 1. 优先读取本地 aowu.png 的 MD5 并拼接 URL
|
|
file_md5 = get_local_file_md5(LOCAL_SPIDER_FILE)
|
|
if file_md5:
|
|
final_jar_value = f"{BASE_JAR_URL};md5;{file_md5}"
|
|
else:
|
|
final_jar_value = BASE_JAR_URL
|
|
|
|
print(f"\n[+] 准备注入的完整 Jar 地址 → {final_jar_value}")
|
|
|
|
# 2. 同步更新顶级 spider 链接
|
|
config_data["spider"] = final_jar_value
|
|
print("[+] 已同步更新顶级 spider 链接")
|
|
|
|
updated_count = 0
|
|
skipped_count = 0
|
|
|
|
# 3. 遍历 sites 进行注入
|
|
for site in sites_list:
|
|
if isinstance(site, dict):
|
|
site_name = site.get("name", "未知站点")
|
|
current_jar = site.get(JAR_KEY)
|
|
|
|
# 已有有效 jar 地址则跳过
|
|
if (
|
|
current_jar
|
|
and isinstance(current_jar, str)
|
|
and current_jar.strip()
|
|
):
|
|
print(f"[~] 跳过已有 Jar 的站点: {site_name}")
|
|
skipped_count += 1
|
|
continue
|
|
|
|
# 没有 jar 或为空 -> 注入带有 MD5 的新地址
|
|
site[JAR_KEY] = final_jar_value
|
|
print(f"[+] 已为站点注入 Jar: {site_name}")
|
|
updated_count += 1
|
|
|
|
if updated_count == 0 and skipped_count == 0:
|
|
print("[-] sites 列表为空或格式异常")
|
|
return
|
|
|
|
# 4. 保存写出新 JSON
|
|
try:
|
|
with open(OUTPUT_JSON, "w", encoding="utf-8") as f:
|
|
json.dump(config_data, f, indent=2, ensure_ascii=False)
|
|
print(f"\n[+] 操作完成！")
|
|
print(f" • 新增/更新 Jar 的站点: {updated_count} 个")
|
|
print(f" • 跳过（已有 Jar）的站点: {skipped_count} 个")
|
|
print(f" • 输出文件: {OUTPUT_JSON}")
|
|
except Exception as e:
|
|
print(f"[-] 保存新 JSON 失败: {e}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
add_jar_to_sites()
|