base.py 34 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959
  1. # -*- coding: utf-8 -*-
  2. """
  3. 平台发布基类
  4. 提供通用的发布接口和工具方法
  5. """
  6. import asyncio
  7. import json
  8. import os
  9. import uuid
  10. from abc import ABC, abstractmethod
  11. from dataclasses import dataclass, field
  12. from datetime import datetime
  13. from typing import List, Optional, Callable, Dict, Any
  14. from playwright.async_api import async_playwright, Browser, BrowserContext, Page
  15. @dataclass
  16. class PublishParams:
  17. """发布参数"""
  18. title: str
  19. video_path: str
  20. description: str = ""
  21. cover_path: Optional[str] = None
  22. tags: List[str] = field(default_factory=list)
  23. publish_date: Optional[datetime] = None
  24. location: str = "重庆市"
  25. def __post_init__(self):
  26. if not self.description:
  27. self.description = self.title
  28. @dataclass
  29. class PublishResult:
  30. """发布结果"""
  31. success: bool
  32. platform: str
  33. video_id: str = ""
  34. video_url: str = ""
  35. message: str = ""
  36. error: str = ""
  37. need_captcha: bool = False # 是否需要验证码
  38. captcha_type: str = "" # 验证码类型: phone, slider, image
  39. screenshot_base64: str = "" # 页面截图(Base64)
  40. page_url: str = "" # 当前页面 URL
  41. status: str = "" # 状态: uploading, processing, success, failed, need_captcha, need_action
  42. @dataclass
  43. class WorkItem:
  44. """作品数据"""
  45. work_id: str
  46. title: str
  47. cover_url: str = ""
  48. video_url: str = ""
  49. duration: int = 0 # 秒
  50. status: str = "published" # published, reviewing, rejected, draft
  51. publish_time: str = ""
  52. play_count: int = 0
  53. like_count: int = 0
  54. comment_count: int = 0
  55. share_count: int = 0
  56. collect_count: int = 0
  57. def to_dict(self) -> Dict[str, Any]:
  58. return {
  59. "work_id": self.work_id,
  60. "title": self.title,
  61. "cover_url": self.cover_url,
  62. "video_url": self.video_url,
  63. "duration": self.duration,
  64. "status": self.status,
  65. "publish_time": self.publish_time,
  66. "play_count": self.play_count,
  67. "like_count": self.like_count,
  68. "comment_count": self.comment_count,
  69. "share_count": self.share_count,
  70. "collect_count": self.collect_count,
  71. }
  72. @dataclass
  73. class CommentItem:
  74. """评论数据"""
  75. comment_id: str
  76. work_id: str
  77. content: str
  78. author_id: str = ""
  79. author_name: str = ""
  80. author_avatar: str = ""
  81. like_count: int = 0
  82. reply_count: int = 0
  83. create_time: str = ""
  84. is_author: bool = False # 是否是作者的评论
  85. replies: List['CommentItem'] = field(default_factory=list)
  86. def to_dict(self) -> Dict[str, Any]:
  87. return {
  88. "comment_id": self.comment_id,
  89. "work_id": self.work_id,
  90. "content": self.content,
  91. "author_id": self.author_id,
  92. "author_name": self.author_name,
  93. "author_avatar": self.author_avatar,
  94. "like_count": self.like_count,
  95. "reply_count": self.reply_count,
  96. "create_time": self.create_time,
  97. "is_author": self.is_author,
  98. "replies": [r.to_dict() for r in self.replies],
  99. }
  100. @dataclass
  101. class WorksResult:
  102. """作品列表结果"""
  103. success: bool
  104. platform: str
  105. works: List[WorkItem] = field(default_factory=list)
  106. total: int = 0
  107. has_more: bool = False
  108. next_page: Any = ""
  109. error: str = ""
  110. def to_dict(self) -> Dict[str, Any]:
  111. return {
  112. "success": self.success,
  113. "platform": self.platform,
  114. "works": [w.to_dict() for w in self.works],
  115. "total": self.total,
  116. "has_more": self.has_more,
  117. "next_page": self.next_page,
  118. "error": self.error,
  119. }
  120. @dataclass
  121. class CommentsResult:
  122. """评论列表结果"""
  123. success: bool
  124. platform: str
  125. work_id: str
  126. comments: List[CommentItem] = field(default_factory=list)
  127. total: int = 0
  128. has_more: bool = False
  129. error: str = ""
  130. def to_dict(self) -> Dict[str, Any]:
  131. return {
  132. "success": self.success,
  133. "platform": self.platform,
  134. "work_id": self.work_id,
  135. "comments": [c.to_dict() for c in self.comments],
  136. "total": self.total,
  137. "has_more": self.has_more,
  138. "error": self.error,
  139. }
  140. class BasePublisher(ABC):
  141. """
  142. 平台发布基类
  143. 所有平台发布器都需要继承此类
  144. """
  145. platform_name: str = "base"
  146. login_url: str = ""
  147. publish_url: str = ""
  148. cookie_domain: str = ""
  149. def __init__(self, headless: bool = True):
  150. self.headless = headless
  151. self.browser: Optional[Browser] = None
  152. self.context: Optional[BrowserContext] = None
  153. self.page: Optional[Page] = None
  154. self.on_progress: Optional[Callable[[int, str], None]] = None
  155. self.user_id: Optional[int] = None
  156. self.publish_task_id: Optional[int] = None
  157. self.publish_account_id: Optional[int] = None
  158. def set_progress_callback(self, callback: Callable[[int, str], None]):
  159. """设置进度回调"""
  160. self.on_progress = callback
  161. def report_progress(self, progress: int, message: str):
  162. """报告进度"""
  163. print(f"[{self.platform_name}] [{progress}%] {message}")
  164. if self.on_progress:
  165. self.on_progress(progress, message)
  166. @staticmethod
  167. def parse_cookies(cookies_str: str) -> list:
  168. """解析 cookie 字符串为列表"""
  169. try:
  170. cookies = json.loads(cookies_str)
  171. if isinstance(cookies, list):
  172. return cookies
  173. except json.JSONDecodeError:
  174. pass
  175. # 字符串格式: name=value; name2=value2
  176. cookies = []
  177. for item in cookies_str.split(';'):
  178. item = item.strip()
  179. if '=' in item:
  180. name, value = item.split('=', 1)
  181. cookies.append({
  182. 'name': name.strip(),
  183. 'value': value.strip(),
  184. 'domain': '',
  185. 'path': '/'
  186. })
  187. return cookies
  188. @staticmethod
  189. def cookies_to_string(cookies: list) -> str:
  190. """将 cookie 列表转换为字符串"""
  191. return '; '.join([f"{c['name']}={c['value']}" for c in cookies])
  192. async def init_browser(self, storage_state: str = None):
  193. """初始化浏览器"""
  194. print(f"[{self.platform_name}] init_browser: headless={self.headless}", flush=True)
  195. playwright = await async_playwright().start()
  196. self.browser = await playwright.chromium.launch(headless=self.headless)
  197. if storage_state and os.path.exists(storage_state):
  198. self.context = await self.browser.new_context(storage_state=storage_state)
  199. else:
  200. self.context = await self.browser.new_context()
  201. self.page = await self.context.new_page()
  202. return self.page
  203. async def set_cookies(self, cookies: list):
  204. """设置 cookies"""
  205. if not self.context:
  206. raise Exception("Browser context not initialized")
  207. # 设置默认域名
  208. for cookie in cookies:
  209. if 'domain' not in cookie or not cookie['domain']:
  210. cookie['domain'] = self.cookie_domain
  211. await self.context.add_cookies(cookies)
  212. async def close_browser(self):
  213. """关闭浏览器"""
  214. if self.context:
  215. await self.context.close()
  216. if self.browser:
  217. await self.browser.close()
  218. async def save_cookies(self, file_path: str):
  219. """保存 cookies 到文件"""
  220. if self.context:
  221. await self.context.storage_state(path=file_path)
  222. async def capture_screenshot(self) -> str:
  223. """截取当前页面截图,返回 Base64 编码"""
  224. import base64
  225. if not self.page:
  226. return ""
  227. try:
  228. screenshot_bytes = await self.page.screenshot(type="jpeg", quality=80)
  229. return base64.b64encode(screenshot_bytes).decode('utf-8')
  230. except Exception as e:
  231. print(f"[{self.platform_name}] 截图失败: {e}")
  232. return ""
  233. async def request_sms_code_from_frontend(self, phone: str = "", timeout_seconds: int = 120, message: str = "") -> str:
  234. node_api_url = os.environ.get('NODEJS_API_URL', 'http://localhost:3000').rstrip('/')
  235. internal_api_key = os.environ.get('INTERNAL_API_KEY', 'internal-api-key-default')
  236. if not self.user_id:
  237. raise Exception("缺少 user_id,无法请求前端输入验证码")
  238. captcha_task_id = f"py_{self.platform_name}_{uuid.uuid4().hex}"
  239. payload = {
  240. "user_id": self.user_id,
  241. "captcha_task_id": captcha_task_id,
  242. "type": "sms",
  243. "phone": phone or "",
  244. "message": message or "请输入短信验证码",
  245. "timeout_seconds": timeout_seconds,
  246. "publish_task_id": self.publish_task_id,
  247. "publish_account_id": self.publish_account_id,
  248. }
  249. import requests
  250. try:
  251. resp = requests.post(
  252. f"{node_api_url}/api/internal/captcha/request",
  253. headers={
  254. "Content-Type": "application/json",
  255. "X-Internal-API-Key": internal_api_key,
  256. },
  257. json=payload,
  258. timeout=timeout_seconds + 30,
  259. )
  260. except Exception as e:
  261. raise Exception(f"请求前端验证码失败: {e}")
  262. try:
  263. data = resp.json()
  264. except Exception:
  265. raise Exception(f"请求前端验证码失败: HTTP {resp.status_code}")
  266. if resp.status_code >= 400 or not data.get("success"):
  267. raise Exception(data.get("error") or data.get("message") or f"请求前端验证码失败: HTTP {resp.status_code}")
  268. code = data.get("code") or ""
  269. if not code:
  270. raise Exception("未收到验证码")
  271. return str(code)
  272. async def ai_analyze_sms_send_state(self, screenshot_base64: str = None) -> dict:
  273. import os
  274. import requests
  275. import json
  276. import re
  277. try:
  278. if not screenshot_base64:
  279. screenshot_base64 = await self.capture_screenshot()
  280. if not screenshot_base64:
  281. return {
  282. "has_sms_modal": False,
  283. "send_button_state": "unknown",
  284. "sent_likely": False,
  285. "block_reason": "unknown",
  286. "suggested_action": "manual_send",
  287. "confidence": 0,
  288. "notes": "无法获取截图",
  289. }
  290. ai_api_key = os.environ.get('DASHSCOPE_API_KEY', '')
  291. ai_base_url = os.environ.get('DASHSCOPE_BASE_URL', 'https://dashscope.aliyuncs.com/compatible-mode/v1')
  292. ai_vision_model = os.environ.get('AI_VISION_MODEL', 'qwen-vl-plus')
  293. if not ai_api_key:
  294. return {
  295. "has_sms_modal": True,
  296. "send_button_state": "unknown",
  297. "sent_likely": False,
  298. "block_reason": "no_ai_key",
  299. "suggested_action": "manual_send",
  300. "confidence": 0,
  301. "notes": "未配置 AI API Key",
  302. }
  303. prompt = """请分析这张网页截图,判断是否处于“短信验证码”验证弹窗/页面,并判断“发送验证码/获取验证码”是否已经触发成功。
  304. 你需要重点识别:
  305. 1) 是否存在短信验证码弹窗(包含“请输入验证码/短信验证码/手机号验证/获取验证码/发送验证码”等)
  306. 2) 发送按钮状态:enabled / disabled / countdown(出现xx秒) / hidden / unknown
  307. 3) 是否已发送成功:例如出现倒计时、按钮禁用、出现“已发送/重新发送/xx秒后重试”等
  308. 4) 是否被阻塞:例如出现滑块/人机验证、频繁发送、风控提示、网络异常等
  309. 请以 JSON 返回:
  310. ```json
  311. {
  312. "has_sms_modal": true,
  313. "send_button_state": "enabled|disabled|countdown|hidden|unknown",
  314. "sent_likely": true,
  315. "block_reason": "none|need_click_send|slider|risk|rate_limit|network|unknown",
  316. "suggested_action": "wait|click_send|solve_slider|manual_send",
  317. "confidence": 0-100,
  318. "notes": "一句话说明你看到的证据"
  319. }
  320. ```"""
  321. headers = {
  322. 'Authorization': f'Bearer {ai_api_key}',
  323. 'Content-Type': 'application/json'
  324. }
  325. payload = {
  326. "model": ai_vision_model,
  327. "messages": [
  328. {
  329. "role": "user",
  330. "content": [
  331. {
  332. "type": "image_url",
  333. "image_url": {
  334. "url": f"data:image/jpeg;base64,{screenshot_base64}"
  335. }
  336. },
  337. {
  338. "type": "text",
  339. "text": prompt
  340. }
  341. ]
  342. }
  343. ],
  344. "max_tokens": 500
  345. }
  346. response = requests.post(
  347. f"{ai_base_url}/chat/completions",
  348. headers=headers,
  349. json=payload,
  350. timeout=30
  351. )
  352. if response.status_code != 200:
  353. return {
  354. "has_sms_modal": True,
  355. "send_button_state": "unknown",
  356. "sent_likely": False,
  357. "block_reason": "network",
  358. "suggested_action": "manual_send",
  359. "confidence": 0,
  360. "notes": f"AI API 返回错误 {response.status_code}",
  361. }
  362. result = response.json()
  363. ai_response = result.get('choices', [{}])[0].get('message', {}).get('content', '')
  364. json_match = re.search(r'```json\\s*([\\s\\S]*?)\\s*```', ai_response)
  365. if json_match:
  366. json_str = json_match.group(1)
  367. else:
  368. json_match = re.search(r'\\{[\\s\\S]*\\}', ai_response)
  369. json_str = json_match.group(0) if json_match else '{}'
  370. try:
  371. data = json.loads(json_str)
  372. except Exception:
  373. data = {}
  374. return {
  375. "has_sms_modal": bool(data.get("has_sms_modal", True)),
  376. "send_button_state": data.get("send_button_state", "unknown"),
  377. "sent_likely": bool(data.get("sent_likely", False)),
  378. "block_reason": data.get("block_reason", "unknown"),
  379. "suggested_action": data.get("suggested_action", "manual_send"),
  380. "confidence": int(data.get("confidence", 0) or 0),
  381. "notes": data.get("notes", ""),
  382. }
  383. except Exception as e:
  384. return {
  385. "has_sms_modal": True,
  386. "send_button_state": "unknown",
  387. "sent_likely": False,
  388. "block_reason": "unknown",
  389. "suggested_action": "manual_send",
  390. "confidence": 0,
  391. "notes": f"AI 分析异常: {e}",
  392. }
  393. async def ai_check_captcha(self, screenshot_base64: str = None) -> dict:
  394. """
  395. 使用 AI 分析截图检测验证码
  396. Args:
  397. screenshot_base64: 截图的 Base64 编码,如果为空则自动获取当前页面截图
  398. Returns:
  399. dict: {
  400. "has_captcha": bool, # 是否有验证码
  401. "captcha_type": str, # 验证码类型: slider, image, phone, rotate, puzzle
  402. "captcha_description": str, # 验证码描述
  403. "confidence": float, # 置信度 0-100
  404. "need_headful": bool # 是否需要切换到有头浏览器
  405. }
  406. """
  407. import os
  408. import requests
  409. try:
  410. # 获取截图
  411. if not screenshot_base64:
  412. screenshot_base64 = await self.capture_screenshot()
  413. if not screenshot_base64:
  414. print(f"[{self.platform_name}] AI验证码检测: 无法获取截图")
  415. return {
  416. "has_captcha": False,
  417. "captcha_type": "",
  418. "captcha_description": "",
  419. "confidence": 0,
  420. "need_headful": False
  421. }
  422. # 获取 AI 配置
  423. ai_api_key = os.environ.get('DASHSCOPE_API_KEY', '')
  424. ai_base_url = os.environ.get('DASHSCOPE_BASE_URL', 'https://dashscope.aliyuncs.com/compatible-mode/v1')
  425. ai_vision_model = os.environ.get('AI_VISION_MODEL', 'qwen-vl-plus')
  426. if not ai_api_key:
  427. print(f"[{self.platform_name}] AI验证码检测: 未配置 AI API Key,使用传统方式检测")
  428. return await self._traditional_captcha_check()
  429. # 构建 AI 请求
  430. prompt = """请分析这张网页截图,判断页面上是否存在验证码。
  431. 请检查以下类型的验证码:
  432. 1. 滑块验证码(需要滑动滑块到指定位置)
  433. 2. 图片验证码(需要选择正确的图片、点击图片上的文字等)
  434. 3. 旋转验证码(需要旋转图片到正确角度)
  435. 4. 拼图验证码(需要拖动拼图块到正确位置)
  436. 5. 手机验证码(需要输入手机收到的验证码)
  437. 6. 计算验证码(需要输入计算结果)
  438. 请以 JSON 格式返回结果:
  439. ```json
  440. {
  441. "has_captcha": true/false,
  442. "captcha_type": "slider/image/phone/rotate/puzzle/calculate/none",
  443. "captcha_description": "验证码的具体描述",
  444. "confidence": 0-100
  445. }
  446. ```
  447. 注意:
  448. - 如果页面有明显的验证码弹窗或验证区域,has_captcha 为 true
  449. - 如果只是普通的登录页面或表单,没有特殊的验证步骤,has_captcha 为 false
  450. - confidence 表示你对判断结果的信心,100 表示非常确定"""
  451. headers = {
  452. 'Authorization': f'Bearer {ai_api_key}',
  453. 'Content-Type': 'application/json'
  454. }
  455. payload = {
  456. "model": ai_vision_model,
  457. "messages": [
  458. {
  459. "role": "user",
  460. "content": [
  461. {
  462. "type": "image_url",
  463. "image_url": {
  464. "url": f"data:image/jpeg;base64,{screenshot_base64}"
  465. }
  466. },
  467. {
  468. "type": "text",
  469. "text": prompt
  470. }
  471. ]
  472. }
  473. ],
  474. "max_tokens": 500
  475. }
  476. print(f"[{self.platform_name}] AI验证码检测: 正在分析截图...")
  477. response = requests.post(
  478. f"{ai_base_url}/chat/completions",
  479. headers=headers,
  480. json=payload,
  481. timeout=30
  482. )
  483. if response.status_code != 200:
  484. print(f"[{self.platform_name}] AI验证码检测: API 返回错误 {response.status_code}")
  485. return await self._traditional_captcha_check()
  486. result = response.json()
  487. ai_response = result.get('choices', [{}])[0].get('message', {}).get('content', '')
  488. print(f"[{self.platform_name}] AI验证码检测响应: {ai_response[:200]}...")
  489. # 解析 AI 响应
  490. import re
  491. json_match = re.search(r'```json\s*([\s\S]*?)\s*```', ai_response)
  492. if json_match:
  493. json_str = json_match.group(1)
  494. else:
  495. # 尝试直接解析
  496. json_match = re.search(r'\{[\s\S]*\}', ai_response)
  497. if json_match:
  498. json_str = json_match.group(0)
  499. else:
  500. json_str = '{}'
  501. try:
  502. ai_result = json.loads(json_str)
  503. except:
  504. ai_result = {}
  505. has_captcha = ai_result.get('has_captcha', False)
  506. captcha_type = ai_result.get('captcha_type', '')
  507. captcha_description = ai_result.get('captcha_description', '')
  508. confidence = ai_result.get('confidence', 0)
  509. # 如果检测到验证码,需要切换到有头浏览器
  510. need_headful = has_captcha and captcha_type not in ['none', '']
  511. print(f"[{self.platform_name}] AI验证码检测结果: has_captcha={has_captcha}, type={captcha_type}, confidence={confidence}")
  512. return {
  513. "has_captcha": has_captcha,
  514. "captcha_type": captcha_type if captcha_type != 'none' else '',
  515. "captcha_description": captcha_description,
  516. "confidence": confidence,
  517. "need_headful": need_headful
  518. }
  519. except Exception as e:
  520. print(f"[{self.platform_name}] AI验证码检测异常: {e}")
  521. import traceback
  522. traceback.print_exc()
  523. return await self._traditional_captcha_check()
  524. async def _traditional_captcha_check(self) -> dict:
  525. """传统方式检测验证码(基于 DOM 元素)"""
  526. if not self.page:
  527. return {
  528. "has_captcha": False,
  529. "captcha_type": "",
  530. "captcha_description": "",
  531. "confidence": 0,
  532. "need_headful": False
  533. }
  534. try:
  535. # 检查常见的验证码选择器
  536. captcha_selectors = [
  537. # 滑块验证码
  538. ('[class*="slider"]', 'slider', '滑块验证码'),
  539. ('[class*="slide-verify"]', 'slider', '滑块验证码'),
  540. ('text="滑动"', 'slider', '滑块验证码'),
  541. ('text="拖动"', 'slider', '滑块验证码'),
  542. # 图片验证码
  543. ('[class*="captcha"]', 'image', '图片验证码'),
  544. ('[class*="verify-img"]', 'image', '图片验证码'),
  545. ('text="点击"', 'image', '图片验证码'),
  546. ('text="选择"', 'image', '图片验证码'),
  547. # 手机验证码
  548. ('text="验证码"', 'phone', '手机验证码'),
  549. ('text="获取验证码"', 'phone', '手机验证码'),
  550. ('[class*="sms-code"]', 'phone', '手机验证码'),
  551. # 旋转验证码
  552. ('text="旋转"', 'rotate', '旋转验证码'),
  553. ('[class*="rotate"]', 'rotate', '旋转验证码'),
  554. ]
  555. for selector, captcha_type, description in captcha_selectors:
  556. try:
  557. count = await self.page.locator(selector).count()
  558. if count > 0:
  559. # 检查是否可见
  560. element = self.page.locator(selector).first
  561. if await element.is_visible():
  562. print(f"[{self.platform_name}] 传统检测: 发现验证码 - {selector}")
  563. return {
  564. "has_captcha": True,
  565. "captcha_type": captcha_type,
  566. "captcha_description": description,
  567. "confidence": 80,
  568. "need_headful": True
  569. }
  570. except:
  571. pass
  572. return {
  573. "has_captcha": False,
  574. "captcha_type": "",
  575. "captcha_description": "",
  576. "confidence": 80,
  577. "need_headful": False
  578. }
  579. except Exception as e:
  580. print(f"[{self.platform_name}] 传统验证码检测异常: {e}")
  581. return {
  582. "has_captcha": False,
  583. "captcha_type": "",
  584. "captcha_description": "",
  585. "confidence": 0,
  586. "need_headful": False
  587. }
  588. async def get_page_url(self) -> str:
  589. """获取当前页面 URL"""
  590. if not self.page:
  591. return ""
  592. try:
  593. return self.page.url
  594. except:
  595. return ""
  596. async def check_publish_status(self) -> dict:
  597. """
  598. 检查发布状态
  599. 返回: {status, screenshot_base64, page_url, message}
  600. """
  601. if not self.page:
  602. return {"status": "error", "message": "页面未初始化"}
  603. try:
  604. screenshot = await self.capture_screenshot()
  605. page_url = await self.get_page_url()
  606. # 检查常见的成功/失败标志
  607. page_content = await self.page.content()
  608. # 检查成功标志
  609. success_keywords = ['发布成功', '上传成功', '发表成功', '提交成功']
  610. for keyword in success_keywords:
  611. if keyword in page_content:
  612. return {
  613. "status": "success",
  614. "screenshot_base64": screenshot,
  615. "page_url": page_url,
  616. "message": "发布成功"
  617. }
  618. # 检查验证码标志
  619. captcha_keywords = ['验证码', '身份验证', '请完成验证', '滑动验证', '图形验证']
  620. for keyword in captcha_keywords:
  621. if keyword in page_content:
  622. return {
  623. "status": "need_captcha",
  624. "screenshot_base64": screenshot,
  625. "page_url": page_url,
  626. "message": f"检测到{keyword}"
  627. }
  628. # 检查失败标志
  629. fail_keywords = ['发布失败', '上传失败', '提交失败', '操作失败']
  630. for keyword in fail_keywords:
  631. if keyword in page_content:
  632. return {
  633. "status": "failed",
  634. "screenshot_base64": screenshot,
  635. "page_url": page_url,
  636. "message": keyword
  637. }
  638. # 默认返回处理中
  639. return {
  640. "status": "processing",
  641. "screenshot_base64": screenshot,
  642. "page_url": page_url,
  643. "message": "处理中"
  644. }
  645. except Exception as e:
  646. return {
  647. "status": "error",
  648. "screenshot_base64": "",
  649. "page_url": "",
  650. "message": str(e)
  651. }
  652. async def wait_for_upload_complete(self, success_selector: str, timeout: int = 300):
  653. """等待上传完成"""
  654. if not self.page:
  655. raise Exception("Page not initialized")
  656. for _ in range(timeout // 3):
  657. try:
  658. count = await self.page.locator(success_selector).count()
  659. if count > 0:
  660. return True
  661. except:
  662. pass
  663. await asyncio.sleep(3)
  664. self.report_progress(30, "正在上传视频...")
  665. return False
  666. @abstractmethod
  667. async def publish(self, cookies: str, params: PublishParams) -> PublishResult:
  668. """
  669. 发布视频 - 子类必须实现
  670. Args:
  671. cookies: cookie 字符串或 JSON
  672. params: 发布参数
  673. Returns:
  674. PublishResult: 发布结果
  675. """
  676. pass
  677. async def get_works(self, cookies: str, page: int = 0, page_size: int = 20) -> WorksResult:
  678. """
  679. 获取作品列表 - 子类可覆盖实现
  680. Args:
  681. cookies: cookie 字符串或 JSON
  682. page: 页码(从0开始)
  683. page_size: 每页数量
  684. Returns:
  685. WorksResult: 作品列表结果
  686. """
  687. return WorksResult(
  688. success=False,
  689. platform=self.platform_name,
  690. error="该平台暂不支持获取作品列表"
  691. )
  692. async def get_comments(self, cookies: str, work_id: str, cursor: str = "") -> CommentsResult:
  693. """
  694. 获取作品评论 - 子类可覆盖实现
  695. Args:
  696. cookies: cookie 字符串或 JSON
  697. work_id: 作品ID
  698. cursor: 分页游标
  699. Returns:
  700. CommentsResult: 评论列表结果
  701. """
  702. return CommentsResult(
  703. success=False,
  704. platform=self.platform_name,
  705. work_id=work_id,
  706. error="该平台暂不支持获取评论"
  707. )
  708. async def run(self, cookies: str, params: PublishParams) -> PublishResult:
  709. """
  710. 运行发布任务
  711. 包装了 publish 方法,添加了异常处理和资源清理
  712. """
  713. try:
  714. return await self.publish(cookies, params)
  715. except Exception as e:
  716. import traceback
  717. traceback.print_exc()
  718. return PublishResult(
  719. success=False,
  720. platform=self.platform_name,
  721. error=str(e)
  722. )
  723. finally:
  724. await self.close_browser()
  725. async def run_get_works(self, cookies: str, page: int = 0, page_size: int = 20) -> WorksResult:
  726. """
  727. 运行获取作品任务
  728. """
  729. try:
  730. return await self.get_works(cookies, page, page_size)
  731. except Exception as e:
  732. import traceback
  733. traceback.print_exc()
  734. return WorksResult(
  735. success=False,
  736. platform=self.platform_name,
  737. error=str(e)
  738. )
  739. finally:
  740. await self.close_browser()
  741. async def run_get_comments(self, cookies: str, work_id: str, cursor: str = "") -> CommentsResult:
  742. """
  743. 运行获取评论任务
  744. """
  745. try:
  746. return await self.get_comments(cookies, work_id, cursor)
  747. except Exception as e:
  748. import traceback
  749. traceback.print_exc()
  750. return CommentsResult(
  751. success=False,
  752. platform=self.platform_name,
  753. work_id=work_id,
  754. error=str(e)
  755. )
  756. finally:
  757. await self.close_browser()
  758. async def check_login_status(self, cookies: str) -> dict:
  759. """
  760. 检查 Cookie 登录状态(通过浏览器访问后台页面检测)
  761. Args:
  762. cookies: cookie 字符串或 JSON
  763. Returns:
  764. dict: {
  765. "success": True,
  766. "valid": True/False,
  767. "need_login": True/False,
  768. "message": "状态描述"
  769. }
  770. """
  771. try:
  772. await self.init_browser()
  773. cookie_list = self.parse_cookies(cookies)
  774. await self.set_cookies(cookie_list)
  775. if not self.page:
  776. raise Exception("Page not initialized")
  777. # 访问平台后台首页
  778. home_url = self.login_url
  779. print(f"[{self.platform_name}] 访问后台页面: {home_url}")
  780. await self.page.goto(home_url, wait_until='domcontentloaded', timeout=30000)
  781. await asyncio.sleep(3)
  782. # 检查当前 URL 是否被重定向到登录页
  783. current_url = self.page.url
  784. print(f"[{self.platform_name}] 当前 URL: {current_url}")
  785. # 登录页特征
  786. login_indicators = ['login', 'passport', 'signin', 'auth']
  787. is_login_page = any(indicator in current_url.lower() for indicator in login_indicators)
  788. # 检查页面是否有登录弹窗
  789. need_login = is_login_page
  790. if not need_login:
  791. # 检查页面内容是否有登录提示
  792. login_selectors = [
  793. 'text="请先登录"',
  794. 'text="登录后继续"',
  795. 'text="请登录"',
  796. '[class*="login-modal"]',
  797. '[class*="login-dialog"]',
  798. '[class*="login-popup"]',
  799. ]
  800. for selector in login_selectors:
  801. try:
  802. if await self.page.locator(selector).count() > 0:
  803. need_login = True
  804. print(f"[{self.platform_name}] 检测到登录弹窗: {selector}")
  805. break
  806. except:
  807. pass
  808. if need_login:
  809. return {
  810. "success": True,
  811. "valid": False,
  812. "need_login": True,
  813. "message": "Cookie 已过期,需要重新登录"
  814. }
  815. else:
  816. return {
  817. "success": True,
  818. "valid": True,
  819. "need_login": False,
  820. "message": "登录状态有效"
  821. }
  822. except Exception as e:
  823. import traceback
  824. traceback.print_exc()
  825. return {
  826. "success": False,
  827. "valid": False,
  828. "need_login": True,
  829. "error": str(e)
  830. }
  831. finally:
  832. await self.close_browser()