butubb
2112c1a870
移植自 mac-agent-os 的 mediacrawler_adapter：不起子进程、不开页面，用浏览器里那份
登录态直接调抖音 Web 接口。产物键名照抄 store/douyin，所以 ingest 那条链路一个字不用改。
**目前能用的（真环境实测，非推断）**：
profile/other : 200, 7075 字节 —— 博主主页指标（粉丝/获赞/作品数/昵称）
aweme/detail : 200, 45425 字节 —— 单条作品详情（含点赞/评论/收藏/分享）
**目前不能用的：作品列表 `aweme/post`。** 两个互相独立的原因：
1. 这个接口被抖音单独升级成了真校验：不带 x-tt-argus 回 403「Uifid Not Found」，
带上 dummy 值回 200 + **空 body**。也就是说「头在不在」骗得过，「真校验」过不了。
同一套头打 profile/other 和 aweme/detail 都是通的 —— 抖音是挑着接口加保护的，
挑中的恰好是「批量拉作品列表」这个最敏感的动作。
2. 改走页面截获也不行：CDP 浏览器打开博主主页会落到「验证码中间页」（当天大量探测的
代价，过几小时要重测）。
所以现在的边界是：**已知作品的指标刷新能做，自动发现新作品做不了**。
**排查中控住变量后得到的两条事实**（都写进注释了）：
· `Accept` / `Accept-Language` / `Referer` 才是主页接口能返回真数据的原因 —— 只有
UA+client hints+Cookie 时是 200 但仅 121 字节的空壳，补上这三个头变 7074 字节。
（我先前猜的 sec-ch-ua 不是关键。）
· 因此 UA 与 client hints 必须**成套地取自同一个浏览器**，所以 BrowserIdentity 一次
从 CDP 取齐 cookie + UA + hints，而不是各自写死。
「200 + 空 body 必须当场报错」也是刻意写死的：放过去它会在下游变成「这个博主没作品」，
把一次失败伪装成一条正常结果 —— 爬虫那条路正是这么栽的，还被翻译成「账号被封」。
测试 +11：cookie 解析、请求头成套性（含 uifid 缺失/回退）、产物键名与 store 对齐、
以及 _get 的三条失败路径（空 body / 403 带网关原话 / 正常返回）。
266 lines
8.6 KiB
Python
266 lines
8.6 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
skill_scanner.py — 技能状态扫描器
|
||
|
||
扫描所有 SKILL_CARD.yaml，报告每个技能的状态：
|
||
- 是否有身份证（SKILL_CARD.yaml）
|
||
- 入口脚本是否存在
|
||
- 核心依赖是否已安装
|
||
- 主机角色是否匹配（只在 master_editor 机器上写入）
|
||
|
||
使用：
|
||
python3 skill_scanner.py
|
||
python3 skill_scanner.py --root ~/workbuddy-agent-os/agent-sync
|
||
python3 skill_scanner.py --check-deps # 同时检查 pip 依赖
|
||
python3 skill_scanner.py --json # JSON格式输出
|
||
"""
|
||
|
||
import argparse
|
||
import json
|
||
import os
|
||
import subprocess
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
|
||
def parse_args():
|
||
parser = argparse.ArgumentParser(description="AgentOS 技能状态扫描器")
|
||
parser.add_argument("--root", default="~/workbuddy-agent-os/agent-sync", help="agent-os 根目录")
|
||
parser.add_argument("--local", default="~/workbuddy-agent-os/agent-local", help="~/workbuddy-agent-os/agent-local")
|
||
parser.add_argument("--check-deps", action="store_true", help="检查 pip 依赖是否已安装")
|
||
parser.add_argument("--json", action="store_true", dest="json_output", help="JSON 格式输出")
|
||
return parser.parse_args()
|
||
|
||
|
||
def load_yaml_simple(file_path: Path) -> dict:
|
||
"""简单 YAML 解析（不依赖 pyyaml）"""
|
||
data = {}
|
||
current_key = None
|
||
current_list = None
|
||
current_section = data
|
||
|
||
try:
|
||
lines = file_path.read_text(encoding="utf-8").splitlines()
|
||
except Exception:
|
||
return {}
|
||
|
||
for line in lines:
|
||
stripped = line.strip()
|
||
if not stripped or stripped.startswith("#"):
|
||
continue
|
||
|
||
indent = len(line) - len(line.lstrip())
|
||
|
||
# 列表项
|
||
if stripped.startswith("- "):
|
||
if current_list is not None:
|
||
current_list.append(stripped[2:].strip().strip('"'))
|
||
continue
|
||
|
||
if ":" in stripped:
|
||
key, _, val = stripped.partition(":")
|
||
key = key.strip()
|
||
val = val.strip().strip('"').strip("'")
|
||
|
||
if val:
|
||
if indent == 0:
|
||
data[key] = val
|
||
current_section = data
|
||
current_list = None
|
||
else:
|
||
current_section[key] = val
|
||
current_list = None
|
||
else:
|
||
# 嵌套块或列表开始
|
||
if indent == 0:
|
||
data[key] = {}
|
||
current_section = data[key]
|
||
current_list = None
|
||
elif key:
|
||
parent = current_section
|
||
parent[key] = []
|
||
current_list = parent[key]
|
||
|
||
return data
|
||
|
||
|
||
def get_host_id(local_root: Path) -> str:
|
||
"""读取本机 HOST_ID"""
|
||
# 从 machine_info.json 读取
|
||
machine_info = local_root / "machine_info.json"
|
||
if machine_info.exists():
|
||
try:
|
||
info = json.loads(machine_info.read_text(encoding="utf-8"))
|
||
return info.get("machine", {}).get("id", "unknown")
|
||
except Exception:
|
||
pass
|
||
# 从 HOST_ID.md 读取
|
||
host_id_md = local_root.parent / "agent-os" / "01_core" / "HOST_ID.md"
|
||
# 已知路径
|
||
for path in [Path("~/workbuddy-agent-os/agent-sync/01_core/HOST_ID.md").expanduser()]:
|
||
if path.exists():
|
||
content = path.read_text(encoding="utf-8")
|
||
import re
|
||
m = re.search(r'`([^`]+_(?:master|maintainer|node))`', content)
|
||
if m:
|
||
return m.group(1)
|
||
return "unknown"
|
||
|
||
|
||
def check_pip_installed(packages: list) -> dict:
|
||
"""检查 pip 包是否已安装"""
|
||
results = {}
|
||
for pkg in packages:
|
||
pkg_name = pkg.split(">=")[0].split("==")[0].split("[")[0].strip()
|
||
try:
|
||
result = subprocess.run(
|
||
[sys.executable, "-c", f"import {pkg_name.replace('-', '_')}"],
|
||
capture_output=True, timeout=5
|
||
)
|
||
results[pkg] = result.returncode == 0
|
||
except Exception:
|
||
results[pkg] = False
|
||
return results
|
||
|
||
|
||
def scan_skill(skill_dir: Path, check_deps: bool, host_id: str) -> dict:
|
||
"""扫描单个技能目录"""
|
||
name = skill_dir.name
|
||
result = {
|
||
"name": name,
|
||
"path": str(skill_dir),
|
||
"has_skill_md": (skill_dir / "SKILL.md").exists(),
|
||
"has_skill_card": (skill_dir / "SKILL_CARD.yaml").exists(),
|
||
"has_entry": False,
|
||
"entry_point": None,
|
||
"status": "unknown",
|
||
"version": "?",
|
||
"category": "?",
|
||
"master_editor": "?",
|
||
"is_master": False,
|
||
"deps_ok": None,
|
||
"missing_deps": [],
|
||
"issues": []
|
||
}
|
||
|
||
# 读取 SKILL_CARD.yaml
|
||
card_path = skill_dir / "SKILL_CARD.yaml"
|
||
if card_path.exists():
|
||
card = load_yaml_simple(card_path)
|
||
skill_info = card.get("skill", {})
|
||
result["version"] = skill_info.get("version", "?")
|
||
result["status"] = skill_info.get("status", "?")
|
||
result["category"] = skill_info.get("category", "?")
|
||
|
||
# 读取 ownership
|
||
ownership = card.get("ownership", {})
|
||
result["master_editor"] = ownership.get("master_editor", "?")
|
||
result["is_master"] = host_id != "unknown" and host_id == result["master_editor"]
|
||
|
||
# 读取入口脚本
|
||
usage = card.get("usage", {})
|
||
entry = usage.get("entry_point", "")
|
||
if entry:
|
||
result["entry_point"] = entry
|
||
result["has_entry"] = (skill_dir / entry).exists()
|
||
if not result["has_entry"]:
|
||
result["issues"].append(f"入口脚本不存在: {entry}")
|
||
|
||
# 检查依赖
|
||
if check_deps:
|
||
env_info = card.get("environment", {})
|
||
pip_deps = env_info.get("pip_dependencies", [])
|
||
if isinstance(pip_deps, list) and pip_deps:
|
||
dep_results = check_pip_installed(pip_deps)
|
||
missing = [pkg for pkg, ok in dep_results.items() if not ok]
|
||
result["deps_ok"] = len(missing) == 0
|
||
result["missing_deps"] = missing
|
||
if missing:
|
||
result["issues"].append(f"缺失依赖: {', '.join(missing)}")
|
||
else:
|
||
result["issues"].append("缺少 SKILL_CARD.yaml")
|
||
|
||
# 没有 SKILL.md 也标记
|
||
if not result["has_skill_md"]:
|
||
result["issues"].append("缺少 SKILL.md")
|
||
|
||
return result
|
||
|
||
|
||
STATUS_ICON = {
|
||
"active": "✅",
|
||
"inactive": "⏸ ",
|
||
"experimental": "🧪",
|
||
"unknown": "❓",
|
||
"?": "❓",
|
||
}
|
||
|
||
|
||
def print_report(skills: list, host_id: str):
|
||
"""打印扫描报告"""
|
||
print("=" * 65)
|
||
print(f"AgentOS 技能状态报告 (本机: {host_id})")
|
||
print("=" * 65)
|
||
|
||
has_card = [s for s in skills if s["has_skill_card"]]
|
||
no_card = [s for s in skills if not s["has_skill_card"]]
|
||
|
||
if has_card:
|
||
print("\n📋 已注册技能（有 SKILL_CARD.yaml）:\n")
|
||
for s in has_card:
|
||
icon = STATUS_ICON.get(s["status"], "❓")
|
||
master_tag = " 🔑[主机]" if s["is_master"] else ""
|
||
issues_tag = f" ⚠️ {'; '.join(s['issues'])}" if s["issues"] else ""
|
||
entry_tag = f" → {s['entry_point']}" if s["entry_point"] else ""
|
||
print(f" {icon} {s['name']:<22} v{s['version']:<8} [{s['category']}]{master_tag}{entry_tag}{issues_tag}")
|
||
|
||
if no_card:
|
||
print("\n📂 未注册技能（无 SKILL_CARD.yaml）:\n")
|
||
for s in no_card:
|
||
print(f" ❓ {s['name']}")
|
||
|
||
print(f"\n统计: 共 {len(skills)} 个技能，"
|
||
f"已注册 {len(has_card)}，"
|
||
f"未注册 {len(no_card)}，"
|
||
f"本机主机 {sum(1 for s in skills if s['is_master'])} 个")
|
||
|
||
issues_all = [s for s in skills if s["issues"]]
|
||
if issues_all:
|
||
print(f"\n⚠️ 需要处理的问题:\n")
|
||
for s in issues_all:
|
||
for issue in s["issues"]:
|
||
print(f" [{s['name']}] {issue}")
|
||
|
||
print("\n图例: ✅=活跃 ⏸=暂停 🧪=实验性 🔑=本机主编辑")
|
||
|
||
|
||
def main():
|
||
args = parse_args()
|
||
root = Path(args.root).expanduser()
|
||
local_root = Path(args.local).expanduser()
|
||
skills_dir = root / "02_skills"
|
||
|
||
if not skills_dir.exists():
|
||
print(f"[ERROR] 技能目录不存在: {skills_dir}")
|
||
sys.exit(1)
|
||
|
||
host_id = get_host_id(local_root)
|
||
skills = []
|
||
|
||
for skill_dir in sorted(skills_dir.iterdir()):
|
||
if not skill_dir.is_dir():
|
||
continue
|
||
if skill_dir.name.startswith(("_", ".")):
|
||
continue
|
||
skill = scan_skill(skill_dir, args.check_deps, host_id)
|
||
skills.append(skill)
|
||
|
||
if args.json_output:
|
||
print(json.dumps(skills, ensure_ascii=False, indent=2))
|
||
else:
|
||
print_report(skills, host_id)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|