86 lines
2.6 KiB
Python
86 lines
2.6 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
获取群聊列表 - 通过 NapCat (OneBot) HTTP API
|
||
|
||
用法:
|
||
python3 qq_get_groups.py # 输出格式化群列表
|
||
python3 qq_get_groups.py --json # 输出 JSON（供 agent 使用）
|
||
python3 qq_get_groups.py --keyword 我的世界 # 搜索群名包含"我的世界"的群
|
||
"""
|
||
|
||
import json
|
||
import sys
|
||
import argparse
|
||
import requests
|
||
|
||
CQHTTP_URL = "http://YOUR_NAPCAT_HOST:25570"
|
||
|
||
|
||
def get_group_list():
|
||
"""获取群聊列表，返回群字典列表"""
|
||
url = f"{CQHTTP_URL}/get_group_list"
|
||
resp = requests.post(url, json={}, timeout=10)
|
||
data = resp.json()
|
||
if data.get("status") == "ok":
|
||
return data.get("data", [])
|
||
raise Exception(f"API 返回异常: {json.dumps(data, ensure_ascii=False)}")
|
||
|
||
|
||
def format_groups(groups, keyword=None):
|
||
"""格式化为可读文本"""
|
||
if keyword:
|
||
keyword = keyword.lower()
|
||
filtered = []
|
||
for g in groups:
|
||
gid = str(g.get("group_id", ""))
|
||
name = g.get("group_name", "")
|
||
if keyword in gid or keyword in name.lower():
|
||
filtered.append(g)
|
||
groups = filtered
|
||
|
||
if not groups:
|
||
return "暂无群聊数据" if not keyword else f"未找到包含「{keyword}」的群"
|
||
|
||
lines = [f"共 {len(groups)} 个群\n"]
|
||
for g in groups:
|
||
gid = g.get("group_id", "?")
|
||
name = g.get("group_name", "?")
|
||
member_count = g.get("member_count", "?")
|
||
max_member = g.get("max_member_count", "?")
|
||
lines.append(f"• {name} ({gid}) — {member_count}/{max_member} 人")
|
||
return "\n".join(lines)
|
||
|
||
|
||
def main():
|
||
parser = argparse.ArgumentParser(description="获取 QQ 群聊列表")
|
||
parser.add_argument("--json", action="store_true", help="输出 JSON 格式")
|
||
parser.add_argument("--keyword", type=str, help="按群名关键词筛选")
|
||
args = parser.parse_args()
|
||
|
||
try:
|
||
groups = get_group_list()
|
||
|
||
if args.keyword:
|
||
kw = args.keyword.lower()
|
||
groups = [
|
||
g for g in groups
|
||
if kw in str(g.get("group_id", ""))
|
||
or kw in g.get("group_name", "").lower()
|
||
]
|
||
|
||
if args.json:
|
||
print(json.dumps(groups, ensure_ascii=False, indent=2))
|
||
else:
|
||
print(format_groups(groups, args.keyword))
|
||
|
||
except requests.exceptions.ConnectionError:
|
||
print(f"❌ 连接失败: 无法连接到 NapCat ({CQHTTP_URL})")
|
||
sys.exit(1)
|
||
except Exception as e:
|
||
print(f"❌ 获取群聊列表失败: {e}")
|
||
sys.exit(1)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|