3
0

yaoex_snapshot_crawl.py 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769
  1. import base64
  2. import hashlib
  3. import json
  4. import random
  5. import signal
  6. import socket
  7. import sys
  8. import time
  9. from pathlib import Path
  10. from urllib.parse import quote
  11. import requests
  12. from Crypto.Cipher import AES
  13. from DrissionPage import ChromiumPage, ChromiumOptions
  14. from commons.Logger import get_spider_logger
  15. from commons.scheduler import CrawlerScheduler
  16. from commons.config import YYC_ACCOUNT, YYC_DEVICE_ID
  17. from oss_upload.oss_upload import AliyunOSSUploader
  18. from pipelines.drug_pipelines import DrugPipeline
  19. from area_info.city_name_to_id import get_city
  20. from Crypto.Util.Padding import unpad
  21. logger = get_spider_logger("yaoex_snapshot")
  22. # 平台常量(API 调度用)
  23. PLATFORM_ID = 6
  24. HEARTBEAT_INTERVAL_SECONDS = 60
  25. TASK_API_BASE = "http://pricesys2.kailin.com.cn:8083/api/collect_task"
  26. TASK_API_TOKEN = "zhijiayun_crawler_2026"
  27. CAPTCHA_TOKEN = "zPzmt1mG1ouCU6GTzsZN2Lmm8pdZypapPcLJTBRETco"
  28. CAPTCHA_API_URL = "http://api.jfbym.com/api/YmServer/customApi"
  29. chrome_path = r"C:\Program Files\Google\Chrome\Application\chrome.exe"
  30. # 项目根目录 → spiders/yaoex(与从哪执行脚本无关)
  31. PROJECT_ROOT = Path(__file__).resolve().parents[2]
  32. YAOEX_SPIDER_DIR = PROJECT_ROOT / "spiders" / "yaoex"
  33. SLIDER_OFFSET_FIX = 10
  34. DETAIL_GET_TIMEOUT = 15
  35. DETAIL_URL_WAIT = 10
  36. DETAIL_DOM_WAIT = 8
  37. DETAIL_NAV_RETRIES = 3
  38. DETAIL_CONTENT_XPATH = "xpath://div[contains(@class,'yaoex-product-detail__content')]"
  39. REQUEST_RETRY_COUNT = 3
  40. REQUEST_TIMEOUT_SEC = 20
  41. NOT_PRODUCT_BREAK = 15
  42. headers = {
  43. "Accept": "application/json, text/plain, */*",
  44. "Accept-Language": "zh-CN,zh;q=0.9",
  45. "Connection": "keep-alive",
  46. "Content-Type": "application/x-www-form-urlencoded",
  47. "Origin": "https://mall.yaoex.com",
  48. "Referer": "https://mall.yaoex.com/",
  49. "Sec-Fetch-Dest": "empty",
  50. "Sec-Fetch-Mode": "cors",
  51. "Sec-Fetch-Site": "cross-site",
  52. "User-Agent": (
  53. "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
  54. "AppleWebKit/537.36 (KHTML, like Gecko) "
  55. "Chrome/146.0.0.0 Safari/537.36"
  56. ),
  57. "X-Request-Agent": "Axios",
  58. "X-Requested-With": "XMLHttpRequest",
  59. "sec-ch-ua": '"Chromium";v="146", "Not-A.Brand";v="24", "Google Chrome";v="146"',
  60. "sec-ch-ua-mobile": "?0",
  61. "sec-ch-ua-platform": '"Windows"',
  62. }
  63. class YaoexSnapshotCrawl:
  64. def __init__(self, drug_dict=None):
  65. self.driver = None
  66. self.platform = 6
  67. self.pipeline = DrugPipeline("yaoex")
  68. self.task_dict = drug_dict or {}
  69. self.ossuploader = AliyunOSSUploader()
  70. self.start_page = 1
  71. self.end_page = 1
  72. self.account_name = YYC_ACCOUNT.get("username", "yyc_default")
  73. self._shop_cache = {}
  74. self._register_signal_handler()
  75. if self.task_dict:
  76. self.get_product_data()
  77. self.success = True
  78. self.is_not_product = 0
  79. self.user_id = YYC_ACCOUNT["user_id"]
  80. self.token = YYC_ACCOUNT["token"]
  81. self.scheduler = CrawlerScheduler(YYC_DEVICE_ID, str(PLATFORM_ID), heartbeat_interval=HEARTBEAT_INTERVAL_SECONDS)
  82. # 独立回告(不依赖 commons,commons 经常变)
  83. self.TASK_REPORT_URL = "http://pricesys2.kailin.com.cn:8083/api/collect_task/report"
  84. self.TASK_API_TOKEN = "zhijiayun_crawler_2026"
  85. def get_product_data(self):
  86. self.task_id = self.task_dict["id"]
  87. self.company_id = self.task_dict["company_id"]
  88. self.product = self.task_dict["product_name"]
  89. self.product_desc = self.task_dict.get("product_specs", "")
  90. self.brand = self.task_dict.get("product_brand", "")
  91. self.product_keyword = self.task_dict.get("product_keyword", "")
  92. self.collect_task_id = self.task_dict.get("collect_task_id", "")
  93. self.sampling_cycle = self.task_dict.get("sampling_cycle", "")
  94. self.sampling_start_time = self.task_dict.get("sampling_start_time", "")
  95. self.sampling_end_time = self.task_dict.get("sampling_end_time", "")
  96. self.collect_equipment_id = self.task_dict.get("collect_equipment_id", "")
  97. self.account_id = self.task_dict.get("collect_equipment_account_id", "")
  98. self.collect_region_id = self.task_dict.get("collect_region_id", "")
  99. self.collect_round = self.task_dict.get("collect_round", 1)
  100. self.start_page = self._parse_page(self.task_dict.get("start_page"), 1)
  101. self.end_page = self._parse_page(self.task_dict.get("end_page"), 100)
  102. # ---------- API report helpers ----------
  103. def _send_report(self, params: dict, log_msg: str):
  104. try:
  105. resp = requests.post(
  106. self.TASK_REPORT_URL, json=params, timeout=10,
  107. headers={'X-Crawler-Token': self.TASK_API_TOKEN},
  108. )
  109. if resp.status_code == 200:
  110. try:
  111. data = resp.json()
  112. if data.get("code") == "success":
  113. logger.info("回告成功:%s", log_msg)
  114. return "success"
  115. else:
  116. msg = data.get("msg", "")
  117. if "已达平台限额" in msg or "限额" in msg:
  118. logger.warning("回告接口返回限额错误:code=%s, msg=%s", data.get('code'), msg)
  119. return "limit_reached"
  120. else:
  121. logger.warning("回告接口返回错误:code=%s, msg=%s", data.get('code'), msg)
  122. return "error"
  123. except ValueError:
  124. logger.warning("回告响应非 JSON:%s", resp.text[:200])
  125. return "error"
  126. else:
  127. logger.warning("回告 HTTP %s", resp.status_code)
  128. return "error"
  129. except Exception as e:
  130. logger.error("回告失败:%s,错误:%s", log_msg, str(e))
  131. return "error"
  132. def _report_start(self):
  133. payload = {
  134. "task_id": self.task_id,
  135. "platform": str(self.platform),
  136. "username": YYC_DEVICE_ID,
  137. "current_page": 0,
  138. "total_pages": 0,
  139. "crawled_count": 0,
  140. "is_finished": 0,
  141. }
  142. logger.info("上报开始 task_id=%s", self.task_id)
  143. return self._send_report(payload, "开始上报(is_finished=0)")
  144. def _report_progress(self, current_page: int, total_pages: int = 0):
  145. payload = {
  146. "task_id": self.task_id,
  147. "platform": str(self.platform),
  148. "username": YYC_DEVICE_ID,
  149. "current_page": current_page,
  150. "total_pages": total_pages,
  151. "crawled_count": self.pipeline.crawl_count,
  152. "is_finished": 0,
  153. }
  154. return self._send_report(payload, f"进度上报 page={current_page}")
  155. def _report_end(self, success: bool, real_count: int, page_no: int = 0, total_pages: int = 0):
  156. payload = {
  157. "task_id": self.task_id,
  158. "platform": str(self.platform),
  159. "username": YYC_DEVICE_ID,
  160. "current_page": page_no,
  161. "total_pages": total_pages if total_pages > 0 else page_no,
  162. "crawled_count": real_count,
  163. "is_finished": 1, # 到了 _report_end 必然已经跑完了
  164. }
  165. if not success:
  166. payload["exception_type"] = 5
  167. payload["remark"] = "采集异常或API请求失败"
  168. logger.info("上报结束 task_id=%s payload=%s", self.task_id, payload)
  169. self._send_report(payload, "结束上报(is_finished=1)")
  170. @staticmethod
  171. def _parse_page(value, default=1):
  172. try:
  173. page = int(value)
  174. return page if page >= 1 else default
  175. except (TypeError, ValueError):
  176. return default
  177. def _register_signal_handler(self):
  178. def handler(signum, frame):
  179. logger.info("收到退出信号,正在关闭浏览器...")
  180. self._quit_browser()
  181. sys.exit(0)
  182. signal.signal(signal.SIGINT, handler)
  183. if hasattr(signal, "SIGTERM"):
  184. signal.signal(signal.SIGTERM, handler)
  185. @staticmethod
  186. def _timestamp_ms() -> str:
  187. return str(int(time.time() * 1000))
  188. def _quit_browser(self):
  189. if self.driver:
  190. try:
  191. self.driver.quit()
  192. except Exception:
  193. pass
  194. self.driver = None
  195. @staticmethod
  196. def _get_free_port():
  197. with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
  198. s.bind(("127.0.0.1", 0))
  199. return s.getsockname()[1]
  200. def _resolve_browser_profile_dir(self):
  201. """浏览器数据目录: <项目根>/spiders/yaoex/<账号>"""
  202. profile_dir = YAOEX_SPIDER_DIR / self.account_name
  203. # 仅兼容历史误路径,新建不再使用 chrome_profile
  204. legacy_nested = YAOEX_SPIDER_DIR / "spiders" / "yaoex" / self.account_name
  205. legacy_chrome_profile = YAOEX_SPIDER_DIR / "chrome_profile" / self.account_name
  206. for candidate in (profile_dir, legacy_nested, legacy_chrome_profile):
  207. if (candidate / "Default").is_dir() or (candidate / "Local State").is_file():
  208. logger.info("使用已有浏览器配置目录: %s", candidate)
  209. return candidate
  210. profile_dir.mkdir(parents=True, exist_ok=True)
  211. logger.info("新建浏览器配置目录: %s", profile_dir)
  212. return profile_dir
  213. def init_browser(self):
  214. co = ChromiumOptions().set_browser_path(chrome_path)
  215. debug_port = self._get_free_port()
  216. profile_dir = self._resolve_browser_profile_dir()
  217. profile_dir.mkdir(parents=True, exist_ok=True)
  218. co.set_user_data_path(str(profile_dir))
  219. logger.info("浏览器用户目录(绝对路径): %s", profile_dir.resolve())
  220. co.set_local_port(debug_port)
  221. co.set_argument(f"--remote-debugging-port={debug_port}")
  222. co.set_argument("--remote-debugging-address=127.0.0.1")
  223. co.set_argument("--disable-dev-shm-usage")
  224. co.set_argument("--start-maximized")
  225. co.set_argument("--no-first-run")
  226. co.set_argument("--no-default-browser-check")
  227. self.driver = ChromiumPage(co)
  228. def _is_logged_in(self):
  229. return bool(self.driver.ele("xpath=//span[@class='login-out']", timeout=5))
  230. def _call_captcha_api(self, image_bytes):
  231. try:
  232. b64 = base64.b64encode(image_bytes).decode()
  233. resp = requests.post(
  234. CAPTCHA_API_URL,
  235. json={"token": CAPTCHA_TOKEN, "type": "22222", "image": b64},
  236. headers={"Content-Type": "application/json"},
  237. timeout=15,
  238. ).json()
  239. logger.info("验证码 API 返回: %s", resp)
  240. return resp["data"]["data"]
  241. except Exception as e:
  242. logger.exception("验证码识别失败: %s", e)
  243. return None
  244. @staticmethod
  245. def _generate_human_track(distance):
  246. tracks = []
  247. current = 0
  248. mid = distance * 0.7
  249. t = 0.2
  250. v = 0
  251. move_points = []
  252. while current < mid:
  253. a = random.uniform(2, 4)
  254. v0 = v
  255. v = v0 + a * t
  256. move = v0 * t + 0.5 * a * t * t
  257. current += move
  258. move_points.append(move)
  259. while current < distance:
  260. a = -random.uniform(0.5, 1.5)
  261. v0 = v
  262. v = v0 + a * t
  263. if v < 0.5:
  264. v = 0.5
  265. move = v0 * t + 0.5 * a * t * t
  266. current += move
  267. move_points.append(move)
  268. total_points = len(move_points)
  269. for i, move in enumerate(move_points):
  270. y_offset = random.randint(-2, 2) if i % random.randint(2, 4) == 0 else 0
  271. if i < total_points * 0.3:
  272. duration = random.uniform(0.01, 0.03)
  273. elif i > total_points * 0.7:
  274. duration = random.uniform(0.03, 0.08)
  275. else:
  276. duration = random.uniform(0.02, 0.05)
  277. if random.random() < 0.05:
  278. duration += random.uniform(0.05, 0.1)
  279. tracks.append((move, y_offset, duration))
  280. if random.random() < 0.7:
  281. tracks.append((-random.randint(1, 3), 0, 0.05))
  282. return tracks
  283. def _simulate_slider_drag(self, slider_element, target_distance):
  284. self.driver.actions.move_to(slider_element).hold()
  285. for offset_x, offset_y, duration in self._generate_human_track(target_distance):
  286. self.driver.actions.move(offset_x, offset_y, duration=duration / 1000)
  287. self.driver.actions.release()
  288. def _solve_slider_if_present(self):
  289. modal = self.driver.ele("xpath://div[@class='yidun_modal']", timeout=3)
  290. if not modal:
  291. return True
  292. logger.info("检测到滑块验证码,开始处理")
  293. jpg_bytes = modal.get_screenshot(as_bytes="jpg")
  294. distance = self._call_captcha_api(jpg_bytes)
  295. if distance is None:
  296. return False
  297. slider = self.driver.ele(
  298. "xpath://div[contains(@class,'yidun_slider--hover')]", timeout=5
  299. )
  300. if not slider:
  301. logger.error("未找到滑块元素")
  302. return False
  303. self._simulate_slider_drag(slider, float(distance) + SLIDER_OFFSET_FIX)
  304. time.sleep(3)
  305. return True
  306. def login(self):
  307. self.driver.get("https://mall.yaoex.com/login", timeout=15)
  308. self.driver.wait.doc_loaded(timeout=10)
  309. input_name = self.driver.ele("xpath://input[@name='username']", timeout=5)
  310. if not input_name:
  311. logger.error("未找到用户名输入框")
  312. return False
  313. input_name.input(YYC_ACCOUNT["username"])
  314. time.sleep(random.uniform(1.2, 2.0))
  315. input_pass = self.driver.ele("xpath://input[@name='password']", timeout=5)
  316. if not input_pass:
  317. logger.error("未找到密码输入框")
  318. return False
  319. input_pass.input(YYC_ACCOUNT["password"])
  320. time.sleep(random.uniform(1.2, 2.0))
  321. geetest_click = self.driver.ele(
  322. "xpath=//div[contains(@class,'geetest_btn_click')]", timeout=3
  323. )
  324. if geetest_click:
  325. geetest_click.click()
  326. time.sleep(1.5)
  327. login_button = self.driver.ele("xpath://input[@id='login-btn']", timeout=5)
  328. if not login_button:
  329. logger.error("未找到登录按钮")
  330. return False
  331. login_button.click()
  332. self.driver.wait.doc_loaded(timeout=10)
  333. time.sleep(2)
  334. if not self._solve_slider_if_present():
  335. return False
  336. return self._is_logged_in()
  337. def decrypt_price(self, ciphertext_b64):
  338. if not ciphertext_b64 or not str(ciphertext_b64).strip():
  339. return ""
  340. _KEY_FIXED = "GDLSAUO1KUMIIBCE"
  341. if not self.user_id:
  342. key = _KEY_FIXED.encode("utf-8")
  343. else:
  344. uid = str(self.user_id)[:6].rjust(6, "0")
  345. key = (_KEY_FIXED[:10] + uid).encode("utf-8")
  346. raw = base64.b64decode(ciphertext_b64.strip())
  347. cipher = AES.new(key, AES.MODE_ECB)
  348. plain = unpad(cipher.decrypt(raw), AES.block_size)
  349. return plain.decode("utf-8")
  350. def _post_with_retry(self, url, payload, retries=REQUEST_RETRY_COUNT, timeout=REQUEST_TIMEOUT_SEC):
  351. last_err = None
  352. for attempt in range(1, retries + 1):
  353. try:
  354. resp = requests.post(
  355. url,
  356. headers=headers,
  357. data=payload,
  358. timeout=timeout,
  359. )
  360. resp.raise_for_status()
  361. return resp
  362. except Exception as e:
  363. last_err = e
  364. if attempt < retries:
  365. logger.warning("请求失败,第%s/%s次重试: %s", attempt, retries, e)
  366. time.sleep(min(2 * attempt, 5))
  367. else:
  368. logger.error("请求失败,已达最大重试次数(%s): %s", retries, e)
  369. raise last_err
  370. def _shop_payload(self, enterprise_id):
  371. return {
  372. "traderName": "yaoex_pc",
  373. "trader": "pc",
  374. "closesignature": "yes",
  375. "signature_method": "md5",
  376. "signature": "****",
  377. "timestamp": self._timestamp_ms(),
  378. "token": self.token,
  379. "userToken": self.token,
  380. "enterpriseId": enterprise_id,
  381. }
  382. def _list_payload(self, keyword, page):
  383. return {
  384. "traderName": "yaoex_pc",
  385. "trader": "pc",
  386. "closesignature": "yes",
  387. "signature_method": "md5",
  388. "signature": "****",
  389. "timestamp": self._timestamp_ms(),
  390. "token": self.token,
  391. "userToken": self.token,
  392. "userId": self.user_id,
  393. "roleId": "101",
  394. "userType": "下游客户",
  395. "buyerCode": self.user_id,
  396. "nowPage": str(page),
  397. "per": "20",
  398. "keyword": keyword,
  399. "catSearchId": "",
  400. "specs": "",
  401. "factoryIds": "",
  402. "sellerCodes": "",
  403. "sellerFilterMode": "0",
  404. "sortColumn": "default",
  405. "sortMode": "default",
  406. "ver": "1",
  407. "stock_mode": "1",
  408. "showExtendCard": "true",
  409. "needDinnerPrice": "true",
  410. "limitStart": "",
  411. "limitEnd": "",
  412. "deadLineStart": "",
  413. "deadLineEnd": "",
  414. "filterDtos": "",
  415. "showWholePurchase": "true",
  416. }
  417. def fetch_list_page(self, keyword, page):
  418. list_url = "https://gateway-b2b.fangkuaiyi.com/home/search/homeSearchList"
  419. resp = self._post_with_retry(list_url, self._list_payload(keyword, page))
  420. data = resp.json()
  421. recall_status = data.get("data", {}).get("recallStatus", 0)
  422. data_block = data.get("data", {}) or {}
  423. # 尝试提取总页数(常见字段名依次尝试)
  424. total_pages = 0
  425. for key in ("totalPage", "totalPages", "pageCount", "total", "totalCount", "totalNum", "totalSize"):
  426. val = data_block.get(key)
  427. if val is not None and int(val) > 0:
  428. total_pages = int(val)
  429. break
  430. if total_pages:
  431. logger.info("📊 列表 API 检测到总页数:%s (page=%s)", total_pages, page)
  432. else:
  433. logger.debug("列表 API 未返回总页数字段,data keys: %s", list(data_block.keys())[:20])
  434. if int(recall_status) == 1:
  435. return data_block.get("shopProducts", []) or [], total_pages
  436. else:
  437. return [], 0
  438. def fetch_shop(self, seller_code):
  439. detail_url = "https://gateway-b2b.fangkuaiyi.com/ycapp/shop/enterpriseQualification"
  440. resp = self._post_with_retry(detail_url, self._shop_payload(seller_code))
  441. shop_res = resp.json().get("data", {})
  442. base_info = shop_res.get("baseInfo", {})
  443. return base_info.get("address", ""), base_info.get("enterpriseName", "")
  444. def _get_shop_info(self, seller_code):
  445. if seller_code in self._shop_cache:
  446. return self._shop_cache[seller_code]
  447. try:
  448. shop_info = self.fetch_shop(seller_code)
  449. except Exception as e:
  450. logger.warning("fetch_shop 失败 seller_code=%s: %s", seller_code, e)
  451. shop_info = ("", "")
  452. self._shop_cache[seller_code] = shop_info
  453. return shop_info
  454. def _current_url(self):
  455. try:
  456. return self.driver.url or ""
  457. except Exception:
  458. return ""
  459. def _url_has_product(self, spu_code, seller_code):
  460. url = self._current_url()
  461. spu_code = str(spu_code or "")
  462. seller_code = str(seller_code or "")
  463. if spu_code and seller_code:
  464. return spu_code in url and seller_code in url
  465. return bool(spu_code and spu_code in url)
  466. def _wait_detail_ready(self, spu_code, seller_code, timeout=DETAIL_URL_WAIT):
  467. deadline = time.time() + timeout
  468. while time.time() < deadline:
  469. if self._url_has_product(spu_code, seller_code):
  470. if self.driver.ele(DETAIL_CONTENT_XPATH, timeout=1):
  471. time.sleep(0.3)
  472. return True
  473. time.sleep(0.4)
  474. return False
  475. def _build_detail_url(self, item):
  476. if not item.get("productId") and item.get("groupBuyProductDto"):
  477. item = item.get("groupBuyProductDto") or {}
  478. spu_code = item.get("spuCode", "")
  479. seller_code = item.get("sellerCode", "")
  480. group_buying_id = item.get("groupBuyingId", "")
  481. p_json = json.dumps(
  482. {"id": group_buying_id, "s": seller_code, "sp": spu_code},
  483. separators=(",", ":"),
  484. )
  485. detail_url = (
  486. f"https://mall.yaoex.com/groupBuying/#/productDetail?p={quote(p_json)}"
  487. )
  488. else:
  489. seller_code = item.get("sellerCode")
  490. spu_code = item.get("spuCode")
  491. detail_url = (
  492. f"https://mall.yaoex.com/v2/product/#/spuCode/{spu_code}/sellerCode/{seller_code}"
  493. )
  494. return item, detail_url, spu_code, seller_code
  495. def _goto_detail_page(self, detail_url, spu_code, seller_code):
  496. """get 后 refresh 一次,让 SPA 按当前 URL 重新渲染详情。"""
  497. for attempt in range(1, DETAIL_NAV_RETRIES + 1):
  498. try:
  499. self.driver.get(detail_url, timeout=DETAIL_GET_TIMEOUT)
  500. time.sleep(0.5)
  501. self.driver.refresh()
  502. time.sleep(2)
  503. ele = self.driver.ele("xpath=//div[@class='yaoex-product-detail__product-detail']")
  504. if ele:
  505. return True
  506. else:
  507. continue
  508. except Exception as e:
  509. logger.warning(
  510. "跳转详情异常 spu=%s seller=%s attempt=%s: %s",
  511. spu_code, seller_code, attempt, e,
  512. )
  513. time.sleep(random.uniform(0.8, 1.5))
  514. return False
  515. def _take_snapshot(self, upload_key):
  516. time.sleep(1)
  517. try:
  518. detail_ele = self.driver.ele(DETAIL_CONTENT_XPATH, timeout=2)
  519. if detail_ele:
  520. jpg_bytes = detail_ele.get_screenshot(as_bytes="jpg")
  521. else:
  522. jpg_bytes = self.driver.get_screenshot(as_bytes="jpg")
  523. if not jpg_bytes:
  524. logger.warning("截图为空 upload_key=%s", upload_key)
  525. return ""
  526. img_url = self.ossuploader.upload_from_bytes(jpg_bytes, str(upload_key))
  527. except Exception:
  528. logger.exception("截图或 OSS 上传失败 upload_key=%s", upload_key)
  529. return ""
  530. if not img_url:
  531. logger.warning("OSS 未返回有效地址 upload_key=%s", upload_key)
  532. return ""
  533. logger.info("截图上传完成 upload_key=%s url=%s", upload_key, img_url)
  534. time.sleep(random.uniform(1, 2))
  535. return img_url
  536. def parse_product(self, item, detail_url, snap_url):
  537. seller_code = item.get("sellerCode")
  538. spu_code = item.get("spuCode")
  539. name_part = (item.get("productName") or "").strip()
  540. short_part = (item.get("shortName") or "").strip()
  541. product_name = f"{name_part} {short_part}".strip()
  542. shop_url = f"https://mall.yaoex.com/v2/store/#/detail/{seller_code}/home"
  543. company_address, company_name = self._get_shop_info(seller_code)
  544. address = item.get("cityName", "")
  545. city_id = province_id = city = province = ""
  546. if address:
  547. city_id, province_id, city, province = get_city(address.split("市")[0])
  548. price = self.decrypt_price(item.get("price"))
  549. hash_text = f"{seller_code}_{spu_code}_{price}"
  550. item_id = hashlib.md5(hash_text.encode("utf-8")).hexdigest()
  551. is_sold_out = 1 if "商品已售罄" in (item.get("statusDescription") or "") else 0
  552. shop_name = item.get("storeName") or item.get("shopName")
  553. anonymous_store_name = ""
  554. if shop_name == "预约配送中心":
  555. anonymous_store_name = item.get("supplyName", "")
  556. inventory = item.get("currentInventory") or item.get("stockCount")
  557. now = time.strftime("%Y-%m-%d %H:%M:%S")
  558. return {
  559. "platform": self.platform,
  560. "item_id": item_id,
  561. "enterprise_id": self.company_id,
  562. "product_name": product_name,
  563. "spec": item.get("spec"),
  564. "one_price": "",
  565. "detail_url": detail_url,
  566. "shop_name": shop_name,
  567. "anonymous_store_name": anonymous_store_name,
  568. "shop_url": shop_url,
  569. "city_name": city,
  570. "city_id": city_id,
  571. "province_name": province,
  572. "province_id": province_id,
  573. "shipment_city_name": "",
  574. "shipment_city_id": "",
  575. "shipment_province_name": "",
  576. "shipment_province_id": "",
  577. "area_info": company_address or "",
  578. "factory_name": item.get("factoryName"),
  579. "scrape_date": time.strftime("%Y-%m-%d"),
  580. "price": price,
  581. "sales": "",
  582. "stock_count": inventory,
  583. "snapshot_url": snap_url,
  584. "approval_num": item.get("approvalNum"),
  585. "produced_time": item.get("productionTime"),
  586. "deadline": item.get("deadLine"),
  587. "update_time": now,
  588. "insert_time": now,
  589. "number": 1,
  590. "product_brand": self.brand or "",
  591. "collect_task_id": self.collect_task_id,
  592. "search_name": self.product,
  593. "company_name": company_name,
  594. "collect_config_info": json.dumps(
  595. {
  596. "sampling_cycle": self.sampling_cycle,
  597. "sampling_start_time": self.sampling_start_time,
  598. "sampling_end_time": self.sampling_end_time,
  599. }
  600. ),
  601. "account_id": self.account_id,
  602. "collect_region_id": self.collect_region_id,
  603. "collect_round": self.collect_round,
  604. "is_sold_out": is_sold_out,
  605. }
  606. def search(self):
  607. self.driver.get("https://mall.yaoex.com/", timeout=15)
  608. self.driver.wait.doc_loaded(timeout=10)
  609. if not self._is_logged_in():
  610. if not self.login():
  611. logger.error("登录失败")
  612. return False
  613. keyword = self.product
  614. if self.brand:
  615. keyword = (self.brand + " " + self.product).strip()
  616. if self.product_desc:
  617. keyword = (keyword + " " + self.product_desc).strip()
  618. for page in range(self.start_page, self.end_page + 1):
  619. if self.scheduler.end:
  620. logger.warning("收到停止信号,终止采集")
  621. break
  622. logger.info("正在爬取 %s %s,第%s页", self.brand, self.product, page)
  623. page_items, api_total_pages = self.fetch_list_page(keyword=keyword, page=page)
  624. if not page_items:
  625. logger.info("第%s页无数据,停止", page)
  626. break
  627. for item in page_items:
  628. if self.scheduler.end:
  629. logger.warning("收到停止信号,终止当前页面采集")
  630. break
  631. item, detail_url, spu_code, seller_code = self._build_detail_url(item)
  632. name_part = (item.get("productName") or "").strip()
  633. short_part = (item.get("shortName") or "").strip()
  634. product_name = f"{name_part} {short_part}".strip()
  635. if self.product not in product_name:
  636. self.is_not_product += 1
  637. continue
  638. if self.brand not in product_name:
  639. self.is_not_product += 1
  640. continue
  641. self.is_not_product = 0
  642. if not self._goto_detail_page(detail_url, spu_code, seller_code):
  643. logger.warning(
  644. "详情页跳转失败,跳过 spu=%s seller=%s url=%s",
  645. spu_code, seller_code, detail_url,
  646. )
  647. continue
  648. upload_key = hashlib.md5(detail_url.encode("utf-8")).hexdigest()
  649. snap_url = self._take_snapshot(upload_key)
  650. product = self.parse_product(item, detail_url, snap_url)
  651. if not product.get("item_id"):
  652. continue
  653. try:
  654. self.pipeline.storge_data(product)
  655. logger.info("%s", json.dumps(product, ensure_ascii=False, default=str))
  656. except Exception as e:
  657. logger.exception("写入数据库失败: %s", e)
  658. time.sleep(random.uniform(1, 2))
  659. if self.is_not_product > NOT_PRODUCT_BREAK:
  660. logger.info("连续不匹配商品过多,停止搜索")
  661. break
  662. # 每页结束上报进度,检查限额
  663. self._page_no = page
  664. prog_status = self._report_progress(page, total_pages=api_total_pages)
  665. if prog_status == "limit_reached":
  666. logger.warning("⛔ 进度上报检测到限额,通知停止")
  667. self.scheduler.set_flag(True)
  668. time.sleep(random.uniform(1, 3))
  669. return True
  670. def run(self):
  671. self._page_no = 0
  672. try:
  673. self.scheduler.start()
  674. self._report_start()
  675. self.init_browser()
  676. self.search()
  677. except Exception as e:
  678. logger.exception("运行异常: %s", e)
  679. self.success = False
  680. finally:
  681. self.scheduler.stop()
  682. try:
  683. self._report_end(
  684. success=self.success,
  685. real_count=self.pipeline.crawl_count,
  686. page_no=self._page_no,
  687. )
  688. except Exception:
  689. pass
  690. self._quit_browser()
  691. logger.info("爬取总数%s", self.pipeline.crawl_count)
  692. return self.pipeline.crawl_count, self.success