5c62a21f32
全局重命名： - 包名: agent_eyes → agent_reach - CLI: agent-eyes → agent-reach - 类名: AgentEyes → AgentReach - 显示名: Agent Eyes → Agent Reach - GitHub: Panniantong/agent-eyes → Panniantong/Agent-Reach 所有 36 个测试通过，CLI/doctor/read/search 全部正常。
128 lines
4.9 KiB
Python
128 lines
4.9 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""Reddit — via Reddit JSON API + optional proxy.
|
||
|
||
Backend: Reddit public JSON API (append .json to any URL)
|
||
Swap to: any Reddit access method
|
||
"""
|
||
|
||
import os
|
||
import requests
|
||
from urllib.parse import urlparse
|
||
from .base import Channel, ReadResult
|
||
|
||
|
||
class RedditChannel(Channel):
|
||
name = "reddit"
|
||
description = "Reddit 帖子和评论"
|
||
backends = ["Reddit JSON API"]
|
||
tier = 2
|
||
|
||
USER_AGENT = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36"
|
||
|
||
def can_handle(self, url: str) -> bool:
|
||
domain = urlparse(url).netloc.lower()
|
||
return "reddit.com" in domain or "redd.it" in domain
|
||
|
||
def check(self, config=None):
|
||
proxy = config.get("reddit_proxy") if config else None
|
||
has_bot = bool(os.environ.get("REDDIT_CLIENT_ID"))
|
||
if proxy and has_bot:
|
||
return "ok", "完整可用（代理 + OAuth Bot）"
|
||
elif proxy:
|
||
return "ok", "代理已配置，可读取帖子。配置 REDDIT_CLIENT_ID/SECRET 可解锁高级搜索和发帖"
|
||
elif has_bot:
|
||
return "warn", "OAuth Bot 已配置，但服务器直连可能被封。配个代理更稳定：agent-reach configure proxy URL"
|
||
else:
|
||
return "off", "搜索用 Exa 免费可用。读帖子需配个代理：agent-reach configure proxy URL"
|
||
|
||
async def read(self, url: str, config=None) -> ReadResult:
|
||
proxy = config.get("reddit_proxy") if config else None
|
||
proxies = {"http": proxy, "https": proxy} if proxy else None
|
||
|
||
# Clean URL: remove query params, trailing slash, then add .json
|
||
parsed = urlparse(url)
|
||
clean_path = parsed.path.rstrip("/")
|
||
# Remove trailing .json if already present (avoid double .json)
|
||
if clean_path.endswith(".json"):
|
||
clean_path = clean_path[:-5]
|
||
json_url = f"https://www.reddit.com{clean_path}.json"
|
||
|
||
try:
|
||
resp = requests.get(
|
||
json_url,
|
||
headers={"User-Agent": self.USER_AGENT},
|
||
proxies=proxies,
|
||
params={"limit": 50},
|
||
timeout=15,
|
||
)
|
||
resp.raise_for_status()
|
||
except requests.exceptions.HTTPError as e:
|
||
status = e.response.status_code if e.response is not None else 0
|
||
if status in (403, 429):
|
||
return ReadResult(
|
||
title="Reddit",
|
||
content="⚠️ Reddit blocked this request (403 Forbidden). "
|
||
"Reddit blocks most server IPs.\n"
|
||
"Fix: agent-reach configure proxy http://user:pass@ip:port\n"
|
||
"Cheap option: https://www.webshare.io ($1/month)\n\n"
|
||
"Alternatively, search Reddit via Exa (free, no proxy needed): "
|
||
"agent-reach search-reddit \"your query\"",
|
||
url=url,
|
||
platform="reddit",
|
||
)
|
||
raise
|
||
|
||
data = resp.json()
|
||
|
||
if isinstance(data, list) and len(data) >= 1:
|
||
# Post page: [post_listing, comments_listing]
|
||
post = data[0]["data"]["children"][0]["data"]
|
||
title = post.get("title", "")
|
||
author = post.get("author", "")
|
||
selftext = post.get("selftext", "")
|
||
score = post.get("score", 0)
|
||
subreddit = post.get("subreddit", "")
|
||
|
||
# Extract comments
|
||
comments_text = ""
|
||
if len(data) >= 2:
|
||
comments_text = self._extract_comments(data[1])
|
||
|
||
content = selftext
|
||
if comments_text:
|
||
content += f"\n\n---\n## Comments\n{comments_text}"
|
||
|
||
return ReadResult(
|
||
title=title,
|
||
content=content,
|
||
url=url,
|
||
author=f"u/{author}",
|
||
platform="reddit",
|
||
extra={"subreddit": subreddit, "score": score},
|
||
)
|
||
|
||
raise ValueError(f"Could not parse Reddit response for: {url}")
|
||
|
||
def _extract_comments(self, comments_data: dict, depth: int = 0, max_depth: int = 3) -> str:
|
||
"""Recursively extract comments."""
|
||
lines = []
|
||
children = comments_data.get("data", {}).get("children", [])
|
||
|
||
for child in children:
|
||
if child.get("kind") != "t1":
|
||
continue
|
||
data = child.get("data", {})
|
||
author = data.get("author", "[deleted]")
|
||
body = data.get("body", "")
|
||
score = data.get("score", 0)
|
||
indent = " " * depth
|
||
|
||
lines.append(f"{indent}**u/{author}** ({score} points):")
|
||
lines.append(f"{indent}{body}")
|
||
lines.append("")
|
||
|
||
# Recurse into replies
|
||
if depth < max_depth and data.get("replies") and isinstance(data["replies"], dict):
|
||
lines.append(self._extract_comments(data["replies"], depth + 1, max_depth))
|
||
|
||
return "\n".join(lines)
|