jd_auto_crawl.py 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842
  1. import random
  2. import re
  3. import signal
  4. import socket
  5. import sys
  6. import time
  7. from decimal import Decimal, InvalidOperation
  8. from urllib.parse import quote
  9. from DrissionPage import ChromiumPage, ChromiumOptions
  10. import json
  11. import hashlib
  12. from commons.Logger import get_spider_logger
  13. from commons.conn_mysql import MySQLPoolOn2
  14. from pipelines.drug_pipelines import DrugPipeline
  15. from commons.feishu_webhook import send_text
  16. from oss_upload.oss_upload import AliyunOSSUploader
  17. import threading
  18. from commons.config import JD_DEVICE_ID
  19. import requests
  20. logger = get_spider_logger("jd")
  21. jd_log = get_spider_logger("jd_log")
  22. chrome_path = r"C:\Program Files\Google\Chrome\Application\chrome.exe"
  23. FETCH_TIMEOUT_FIRST = 5
  24. FETCH_TIMEOUT_SCROLL = 6
  25. LISTEN_CLEAR_ROUNDS = 3
  26. LISTEN_CLEAR_TIMEOUT = 0.45
  27. # 「下一页」是否在视口内(条件略宽)
  28. _JS_NEXT_BTN_IN_VIEWPORT = """
  29. var el = arguments[0];
  30. if (!el) return false;
  31. var r = el.getBoundingClientRect();
  32. var h = window.innerHeight || document.documentElement.clientHeight || 800;
  33. var w = window.innerWidth || document.documentElement.clientWidth || 1200;
  34. return r.bottom > 80 && r.top < h - 40 && r.right > 0 && r.left < w;
  35. """
  36. class JdCrawlerV2:
  37. def __init__(self, drug_dict=None,scheduler=None):
  38. self.driver = None
  39. self.cumulative_pages = 0
  40. self.cumulative_items = 0
  41. self.cumulative_stored = 0
  42. self.cumulative_skipped = 0
  43. self.page_stored = 0
  44. self.register_signal_handler()
  45. self.db = MySQLPoolOn2()
  46. self.ip = None
  47. self.account_name = None
  48. self.login_username = None
  49. self.login_password = None
  50. self.platform = 2
  51. self.pipeline = DrugPipeline("jd")
  52. self.task_dict = drug_dict or {}
  53. self.ossuploader = AliyunOSSUploader()
  54. self.start_page = 1
  55. self.end_page = 1
  56. self.report_data={}
  57. self.heartbeat_interval = 30
  58. self.scheduler = scheduler
  59. if self.task_dict:
  60. self.get_product_data()
  61. self.success = True
  62. self.is_no_prodcut = 0
  63. def get_product_data(self):
  64. self.task_id = self.task_dict["id"]
  65. self.company_id = self.task_dict["company_id"]
  66. self.product = self.task_dict["product_name"]
  67. self.product_desc = self.task_dict.get("product_specs", "")
  68. self.brand = self.task_dict.get("product_brand", "")
  69. self.product_keyword = self.task_dict.get("product_keyword", "")
  70. self.collect_task_id = self.task_dict.get("collect_task_id", "")
  71. self.sampling_cycle = self.task_dict.get("sampling_cycle", "")
  72. self.sampling_start_time = self.task_dict.get("sampling_start_time", "")
  73. self.sampling_end_time = self.task_dict.get("sampling_end_time", "")
  74. self.collect_equipment_id = self.task_dict.get("collect_equipment_id", "")
  75. self.account_id = self.task_dict.get("collect_equipment_account_id", "")
  76. self.collect_region_id = self.task_dict.get("collect_region_id", "")
  77. self.collect_round = self.task_dict.get("collect_round", 1)
  78. self.start_page = self._parse_page(self.task_dict.get("current_page")+1, 1)
  79. self.end_page = self._parse_page(self.task_dict.get("end_page"), 15)
  80. self.username = self.task_dict.get("username",'')
  81. self.report_data = {'task_id':self.task_id,'platform':self.platform,'username':self.task_dict.get("username",JD_DEVICE_ID)}
  82. @staticmethod
  83. def _parse_page(value, default=1):
  84. try:
  85. page = int(value)
  86. return page if page >= 1 else default
  87. except (TypeError, ValueError):
  88. return default
  89. @staticmethod
  90. def _get_free_port():
  91. """获取一个当前可用的本地端口,供 Chrome 调试使用。"""
  92. with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
  93. s.bind(("127.0.0.1", 0))
  94. return s.getsockname()[1]
  95. def init_browser(self):
  96. co = ChromiumOptions().set_browser_path(chrome_path)
  97. debug_port = self._get_free_port()
  98. co.set_user_data_path(f"./spiders/jd/test/")
  99. co.set_local_port(debug_port)
  100. co.set_argument(f"--remote-debugging-port={debug_port}")
  101. co.set_argument("--remote-debugging-address=127.0.0.1")
  102. # co.set_argument("--disable-blink-features=AutomationControlled")
  103. co.set_argument("--disable-dev-shm-usage")
  104. co.set_argument("--no-first-run") # 避免首次运行弹窗
  105. co.set_argument("--no-default-browser-check") # 避免默认浏览器检查
  106. #co.headless(True)
  107. co.set_user_agent('Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36')
  108. # 3. 添加反检测启动参数
  109. co.set_argument('--disable-blink-features=AutomationControlled') # 禁用自动化控制标志
  110. co.set_argument('--disable-gpu') # 禁用 GPU(无头模式常用)
  111. co.set_argument('--no-sandbox') # 禁用沙箱
  112. co.set_argument('--start-maximized') # 启动时最大化窗口
  113. co.set_argument('--disable-infobars') # 禁用信息栏提示
  114. if self.ip:
  115. proxy = self.ip.strip()
  116. if not proxy.startswith(("http://", "https://")):
  117. proxy = f"http://{proxy}"
  118. co.set_argument(f"--proxy-server={proxy}")
  119. logger.info("启动浏览器: account=%s, debug_port=%s", self.account_name, debug_port)
  120. self.driver = ChromiumPage(co)
  121. self._listen_started = False
  122. def _start_listen(self):
  123. """登录完成后再开监听,避免干扰登录页/验证码拖动。"""
  124. if self._listen_started or not self.driver:
  125. return
  126. self.driver.listen.start("api?appid=search-pc-java")
  127. self._listen_started = True
  128. logger.info("已启动搜索接口监听")
  129. def register_signal_handler(self):
  130. def handler(signum, frame):
  131. print("\n⚠️ 程序退出")
  132. if self.driver:
  133. self.driver.quit()
  134. sys.exit(0)
  135. signal.signal(signal.SIGINT, handler)
  136. if hasattr(signal, "SIGTERM"):
  137. signal.signal(signal.SIGTERM, handler)
  138. def sleep(self, a, b):
  139. time.sleep(random.uniform(a, b))
  140. def _scroll_page_down(self, delta=900):
  141. self.driver.run_js(f"window.scrollBy(0, {int(delta)});")
  142. time.sleep(random.uniform(0.3, 0.6))
  143. def _scroll_next_into_view(self, el):
  144. if not el:
  145. return
  146. try:
  147. self.driver.run_js(
  148. "arguments[0].scrollIntoView({block:'center',behavior:'instant'});",
  149. el,
  150. )
  151. self.sleep(1, 2)
  152. except Exception as e:
  153. logger.warning("滚动到下一页按钮失败: %s", e)
  154. try:
  155. el.scroll.to_see()
  156. except Exception:
  157. pass
  158. def _get_scroll_info(self):
  159. return self.driver.run_js("""
  160. return {
  161. scrollY: window.scrollY || window.pageYOffset || 0,
  162. docH: Math.max(document.body.scrollHeight,
  163. document.documentElement.scrollHeight,
  164. document.body.offsetHeight),
  165. viewH: window.innerHeight || document.documentElement.clientHeight || 800
  166. };
  167. """)
  168. def _find_next_btn(self, timeout=0.3):
  169. try:
  170. return self.driver.ele("text=下一页", timeout=timeout)
  171. except Exception:
  172. return None
  173. def _is_next_btn_visible(self, btn):
  174. if not btn:
  175. return False
  176. try:
  177. return bool(self.driver.run_js(_JS_NEXT_BTN_IN_VIEWPORT, btn))
  178. except Exception:
  179. return False
  180. def _human_click(self, element):
  181. """在目标节点上触发 click,避免 move_to + 无目标 actions.click() 因布局位移点到商品链接触发详情页。"""
  182. if not element:
  183. return False
  184. try:
  185. self.sleep(0.8, 2.0)
  186. try:
  187. self.driver.run_js(
  188. "arguments[0].scrollIntoView({block:'center',behavior:'instant'});",
  189. element,
  190. )
  191. except Exception:
  192. pass
  193. self.sleep(0.2, 0.6)
  194. self.driver.run_js("arguments[0].click();", element)
  195. return True
  196. except Exception as e:
  197. logger.warning("点击失败: %s", e)
  198. try:
  199. element.click()
  200. return True
  201. except Exception:
  202. return False
  203. @staticmethod
  204. def _estimated_price(json_data):
  205. fp = json_data.get("finalPrice")
  206. if isinstance(fp, dict):
  207. return fp.get("estimatedPrice", "") or ""
  208. return ""
  209. def get_heshu(self,full_title):
  210. last_box = None
  211. last_bottle = None
  212. for match in re.finditer(r"(\d+)(盒|瓶)", full_title):
  213. if match.group(2) == '盒':
  214. last_box = match
  215. else: # 瓶
  216. last_bottle = match
  217. if last_box:
  218. return int(last_box.group(1))
  219. elif last_bottle:
  220. return int(last_bottle.group(1))
  221. else:
  222. return 1
  223. def _take_snapshot(self, upload_key, ele):
  224. """在指定标签页截图并上传。"""
  225. time.sleep(1)
  226. try:
  227. jpg_bytes = ele.get_screenshot(as_bytes="jpg")
  228. if not jpg_bytes:
  229. logger.warning("截图为空 upload_key=%s", upload_key)
  230. return ""
  231. img_url = self.ossuploader.upload_from_bytes(jpg_bytes, str(upload_key))
  232. except Exception:
  233. logger.exception("截图或 OSS 上传失败 upload_key=%s", upload_key)
  234. return ""
  235. if not img_url:
  236. logger.warning("OSS 未返回有效地址 upload_key=%s", upload_key)
  237. return ""
  238. logger.info("截图上传完成 upload_key=%s url=%s", upload_key, img_url)
  239. time.sleep(random.uniform(1, 2))
  240. return img_url
  241. def get_heshu(self,full_title):
  242. last_box = None
  243. last_bottle = None
  244. for match in re.finditer(r"(\d+)(盒|瓶)", full_title):
  245. if match.group(2) == '盒':
  246. last_box = match
  247. else: # 瓶
  248. last_bottle = match
  249. if last_box:
  250. return int(last_box.group(1))
  251. elif last_bottle:
  252. return int(last_bottle.group(1))
  253. else:
  254. return 1
  255. def parse(self, ware_list):
  256. for w in ware_list:
  257. title = w.get("wareName", "")
  258. title = re.sub(r"<[^>]*>", "", title).strip()
  259. color = w.get("color", "")
  260. full_title = title + " " + color
  261. logger.info(full_title)
  262. if self.product not in full_title:
  263. self.is_no_prodcut += 1
  264. continue
  265. if self.brand not in full_title:
  266. self.is_no_prodcut += 1
  267. continue
  268. if self.product_desc:
  269. if self.product_desc in full_title:
  270. crawl_product_desc = self.product_desc
  271. else:
  272. crawl_product_desc = ""
  273. title = full_title
  274. else:
  275. crawl_product_desc = ""
  276. title = full_title
  277. if "+[" in title:
  278. continue
  279. self.is_no_prodcut = 0
  280. status = 1
  281. if self.product_keyword:
  282. search_keyword_list = self.product_keyword.split(",")
  283. for search_keyword in search_keyword_list:
  284. if search_keyword.strip() not in title:
  285. status = 0
  286. if status == 0:
  287. continue
  288. logger.info(f"商品名:{title}")
  289. sku_id = w.get("skuId", "")
  290. sales = w.get("totalSales", "")
  291. shop_id = w.get("shopId", "")
  292. shop_name = w.get("shopName", "")
  293. heshu_count = self.get_heshu(full_title)
  294. final_price = self._estimated_price(w)
  295. jd_price = w.get("jdPrice", "")
  296. item_url = f"https://item.jd.com/{sku_id}.html"
  297. low_price = final_price if final_price else jd_price
  298. # 获取列表页快照
  299. ele_xpath = "//div[@id='main_search_conter']//div[contains(@class,'_goodsContainer_')]/div[@data-sku=" + "'" + sku_id + "'" + "]"
  300. ele_screen = self.driver.ele("xpath="+ele_xpath)
  301. upload_key = hashlib.md5(item_url.encode("utf-8")).hexdigest()
  302. #后续开启
  303. #snap_url = self._take_snapshot(upload_key,ele_screen)
  304. snap_url = ''
  305. try:
  306. price = Decimal(str(low_price)).quantize(Decimal("0.00"))
  307. except (InvalidOperation, ValueError):
  308. price = Decimal("0.00")
  309. item_url = f"https://item.jd.com/{sku_id}.html"
  310. mall_url = f"https://mall.jd.com/index-{shop_id}.html?from=pc"
  311. # 字段与 yaofangwang_crawl 对齐;键顺序须与 commons.sql_data.RETRIEVE_SCRAPE_INSERT_COLUMNS 一致
  312. now_ts = time.strftime("%Y-%m-%d %H:%M:%S")
  313. product = {
  314. "platform": self.platform,
  315. "item_id": sku_id,
  316. "enterprise_id": self.company_id,
  317. "product_name": title,
  318. "spec": crawl_product_desc,
  319. "one_price": "",
  320. "detail_url": item_url,
  321. "shop_name": shop_name,
  322. "anonymous_store_name": "",
  323. "shop_url": mall_url,
  324. "city_name": "",
  325. "city_id": "",
  326. "province_name": "",
  327. "province_id": "",
  328. "shipment_city_name": "",
  329. "shipment_city_id": "",
  330. "shipment_province_name": "",
  331. "shipment_province_id": "",
  332. "area_info": "",
  333. "factory_name": "",
  334. "scrape_date": time.strftime("%Y-%m-%d"),
  335. "price": price,
  336. "sales": sales,
  337. "stock_count": "",
  338. "snapshot_url": snap_url,
  339. "approval_num": "",
  340. "produced_time": "",
  341. "deadline": "",
  342. "update_time": now_ts,
  343. "insert_time": now_ts,
  344. "number": heshu_count,
  345. "product_brand": self.brand or "",
  346. "collect_task_id": self.collect_task_id,
  347. "task_id": self.task_id,
  348. "search_name": self.product,
  349. "company_name": "",
  350. "collect_config_info": json.dumps(
  351. {
  352. "sampling_cycle": self.sampling_cycle,
  353. "sampling_start_time": self.sampling_start_time,
  354. "sampling_end_time": self.sampling_end_time,
  355. }
  356. ),
  357. "account_id": self.account_id,
  358. "collect_region_id": self.collect_region_id,
  359. "collect_round": self.collect_round,
  360. "is_sold_out": 0
  361. }
  362. try:
  363. affected_rows = self.pipeline.storge_data(product)
  364. if affected_rows and affected_rows > 0:
  365. self.page_stored += 1
  366. logger.info("%s", json.dumps(product, ensure_ascii=False, default=str))
  367. except Exception as e:
  368. logger.exception("写入数据库失败: %s", e)
  369. @staticmethod
  370. def _response_has_ware_list(data):
  371. if not isinstance(data, dict):
  372. return False
  373. wl = data.get("data", {}).get("wareList")
  374. return bool(wl)
  375. def fetch_items_once(self, timeout=FETCH_TIMEOUT_FIRST):
  376. n = 0
  377. for resp in self.driver.listen.steps(timeout=timeout):
  378. try:
  379. data = resp.response.body
  380. if not self._response_has_ware_list(data):
  381. continue
  382. ware_list = data["data"]["wareList"]
  383. self.parse(ware_list)
  384. n += len(ware_list)
  385. except Exception as e:
  386. logger.warning("解析监听响应失败: %s", e)
  387. return n
  388. def clear_listen_buffer(self, rounds=LISTEN_CLEAR_ROUNDS, timeout=LISTEN_CLEAR_TIMEOUT):
  389. try:
  390. for _ in range(rounds):
  391. resps = list(self.driver.listen.steps(timeout=timeout))
  392. if not resps:
  393. break
  394. logger.debug("监听缓冲已清空")
  395. except Exception as e:
  396. logger.debug("清空监听缓冲失败: %s", e)
  397. def collect_full_page_items(self, max_steps=10):
  398. """单次循环:边滑动边收数据,到底 / 看见「下一页」即停。"""
  399. n = self.fetch_items_once(timeout=FETCH_TIMEOUT_FIRST)
  400. stagnant = 0
  401. last_scroll_y = None
  402. for step in range(max_steps):
  403. next_btn = self._find_next_btn(timeout=0.3)
  404. if self._is_next_btn_visible(next_btn):
  405. n += self.fetch_items_once(timeout=FETCH_TIMEOUT_SCROLL)
  406. return n, next_btn
  407. info = self._get_scroll_info()
  408. scroll_y = info["scrollY"]
  409. doc_h = info["docH"]
  410. view_h = info["viewH"]
  411. at_bottom = (scroll_y + view_h >= doc_h - 20)
  412. if last_scroll_y is not None and abs(scroll_y - last_scroll_y) < 8:
  413. stagnant += 1
  414. else:
  415. stagnant = 0
  416. last_scroll_y = scroll_y
  417. if at_bottom and stagnant >= 2:
  418. n += self.fetch_items_once(timeout=FETCH_TIMEOUT_SCROLL)
  419. next_btn = self._find_next_btn(timeout=2)
  420. if next_btn:
  421. self._scroll_next_into_view(next_btn)
  422. return n, next_btn
  423. logger.info("已到页面底部且未发现下一页,停止滑动")
  424. return n, None
  425. self._scroll_page_down(random.randint(400, 800))
  426. if random.random() < 0.15:
  427. self.driver.run_js(f"window.scrollBy(0, -{random.randint(60, 140)})")
  428. self.sleep(0.5, 1.5)
  429. if step % 3 == 2:
  430. n += self.fetch_items_once(timeout=FETCH_TIMEOUT_SCROLL)
  431. n += self.fetch_items_once(timeout=FETCH_TIMEOUT_SCROLL)
  432. next_btn = self._find_next_btn(timeout=3)
  433. if next_btn and not self._is_next_btn_visible(next_btn):
  434. self._scroll_next_into_view(next_btn)
  435. return n, next_btn
  436. def get_account(self):
  437. sql_account = """
  438. SELECT *
  439. FROM `retrieve_collect_equipment_account`
  440. WHERE `id` = %s
  441. and `status` = 0
  442. """
  443. account_list = self.db.select_data(sql_account, self.account_id)
  444. if not account_list:
  445. return False
  446. account_dict = account_list[0]
  447. print(account_dict)
  448. self.ip = account_dict.get("ip")
  449. self.account_name = account_dict.get("username")
  450. self.login_username = account_dict.get("phone", "")
  451. self.login_password = account_dict.get("password", "")
  452. logger.info("获取到账号: %s, ip: %s", self.account_name, self.ip)
  453. return True
  454. def disable_account(self):
  455. update_sql = f""" UPDATE `retrieve_collect_equipment_account` SET `status`= %s WHERE `name` = %s; """
  456. self.db.execute(update_sql, (1, self.account_name))
  457. def _build_search_keyword(self):
  458. parts = [p for p in (self.brand, self.product, self.product_desc) if p]
  459. return " ".join(parts).strip() or self.product
  460. def _is_logged_out(self):
  461. return bool(self.driver.ele("xpath=//*[@class='link-login']", timeout=2))
  462. def perform_jd_login(self):
  463. """
  464. 使用已有浏览器实例执行京东账号密码登录(含滑块验证码)。
  465. 成功返回 True,失败返回 False。
  466. """
  467. username = self.login_username
  468. password = self.login_password
  469. login_url = "https://passport.jd.com/new/login.aspx"
  470. self.driver.get(login_url)
  471. input_name = self.driver.ele("xpath=//input[@id='loginname']", timeout=15)
  472. if not input_name:
  473. print("未找到用户名输入框")
  474. return False
  475. input_name.input(username)
  476. time.sleep(random.uniform(1.5, 2.5))
  477. input_pass = self.driver.ele("xpath://input[@name='nloginpwd']", timeout=5)
  478. if not input_pass:
  479. print("未找到密码输入框")
  480. return False
  481. input_pass.input(password)
  482. time.sleep(random.uniform(1.5, 2.5))
  483. login_btn = self.driver.ele("xpath://a[@id='loginsubmit']", timeout=5)
  484. if not login_btn:
  485. print("未找到登录按钮")
  486. return False
  487. login_btn.click()
  488. time.sleep(random.uniform(3, 5))
  489. return True
  490. def _ensure_logged_in(self):
  491. """未登录时自动走登录流程(账号密码 + 滑块)。"""
  492. if not self._is_logged_out():
  493. return True
  494. logger.info("检测到未登录,开始自动登录: %s", self.account_name)
  495. ok = self.perform_jd_login()
  496. if ok and not self._is_logged_out():
  497. logger.info("自动登录成功: %s", self.account_name)
  498. return True
  499. logger.error("自动登录失败: %s", self.account_name)
  500. return False
  501. def _check_page_blocked(self):
  502. html = self.driver.html or ""
  503. if "抱歉由于访问频繁导致无法搜索" in html:
  504. logger.error("账号无法搜索(访问频繁)")
  505. self.success = False
  506. return True
  507. return False
  508. def _jump_to_page(self, target_page):
  509. """跳转到指定页码,并清空跳转前的监听残留。"""
  510. to_page_input = self.driver.ele(
  511. "xpath=//div[contains(@class,'_pagination_toPageNum_')]//input[@type='text']",
  512. timeout=3,
  513. )
  514. if not to_page_input:
  515. logger.warning("未找到跳页输入框,无法跳转到第 %s 页", target_page)
  516. return False
  517. self.clear_listen_buffer()
  518. to_page_input.input(str(target_page))
  519. self.sleep(1, 2)
  520. self.driver.actions.key_down("enter").key_up("enter")
  521. self.sleep(3, 5)
  522. self.clear_listen_buffer()
  523. logger.info("已跳转到第 %s 页", target_page)
  524. return True
  525. def _go_next_page(self, next_btn):
  526. self.clear_listen_buffer()
  527. if not self._human_click(next_btn):
  528. logger.warning("点击下一页失败")
  529. return False
  530. self.sleep(2, 4)
  531. return True
  532. def _error_report(self,data):
  533. self.report_data.update(data)
  534. self.scheduler.stop()
  535. self.scheduler.post_report(self.report_data)
  536. def _success_report(self,data):
  537. self.report_data.update(data)
  538. self.scheduler.post_report(self.report_data)
  539. def post_report(self,data):
  540. url = "http://192.168.2.246:8080/api/collect_task/report"
  541. print('传给返回接口的数据',data)
  542. headers = {'X-Crawler-Token':'zhijiayun_crawler_2026'}
  543. response = requests.post(url, json=data,headers =headers)
  544. #记录日志
  545. result = response.json()
  546. if(result.get('code')!='success'):
  547. logger.info(f'翻页回传:{result}')
  548. self.scheduler.stop()
  549. #用守护线程 直接杀死该进程
  550. print(f'任务进度上传 {result}')
  551. def crawl(self):
  552. total = 0
  553. keyword = self._build_search_keyword()
  554. self.driver.get("https://www.jd.com/", timeout=15)
  555. time.sleep(15)
  556. if(self.driver.url == 'https://www.jd.com/?from=pc_search_sd'):
  557. self._error_report({'is_finished':0,'need_reassign':1,'current_page':self.start_page,'exception_type':2})
  558. if self._is_logged_out():
  559. if not self.login_password or not self.login_username:
  560. #这里要报账号异常错误,没登录账号
  561. self._error_report({'is_finished':0,'need_reassign':1,'current_page':self.start_page,'exception_type':1})
  562. return
  563. if not self._ensure_logged_in():
  564. self.disable_account()
  565. send_text(f"京东:{self.account_name}账号登录失败")
  566. jd_log.info(f'出现要求登录:::::{self.driver.html}')
  567. self.success = False
  568. self._error_report({'is_finished':0,'need_reassign':1,'current_page':self.start_page,'exception_type':1})
  569. return
  570. self.driver.get("https://www.jd.com/", timeout=15)
  571. self.sleep(3, 5)
  572. kw = quote(str(keyword or ""), safe="")
  573. self._search_kw = kw
  574. # 必须先监听再打开搜索页,否则首屏 wareList(前约 30 条)在监听开启前就返回了
  575. self._start_listen()
  576. self.driver.get(
  577. f"https://search.jd.com/Search?keyword={kw}&enc=utf-8&wq={kw}", timeout=15
  578. )
  579. self.sleep(5, 8)
  580. if self._check_page_blocked(): #禁止搜索
  581. self._error_report({'is_finished':0,'need_reassign':1,'current_page':self.start_page,'exception_type':2})
  582. return
  583. #这一步捕捉异常:
  584. html = self.driver.html or ""
  585. if "请刷新或切换账户试试" in html:
  586. self._error_report({'is_finished':0,'need_reassign':1,'current_page':self.start_page,'exception_type':2})
  587. return
  588. total_page = int(re.search(r'(\d+)', self.driver.ele('@class:_pagination_total_1gv36_61', timeout=2).text).group(1))
  589. self.report_data.update({'total_pages':total_page})
  590. if self.start_page > 1:
  591. if not self._jump_to_page(self.start_page):
  592. logger.warning("跳页失败,将从第 1 页开始采集")
  593. self.start_page = 1
  594. self._error_report({'is_finished':0,'need_reassign':1,'current_page':self.start_page,'exception_type ':4})
  595. return #说明页数可能出错,直接返回结束任务且汇报
  596. logger.info(
  597. "采集页码范围: %s ~ %s(共 %s 页)",
  598. self.start_page,
  599. total_page,
  600. self.end_page - self.start_page + 1,
  601. )
  602. for page_no in range(self.start_page, total_page+1):
  603. if self._is_logged_out():
  604. if not self._ensure_logged_in():
  605. self._error_report({'is_finished':0,'need_reassign':1,'current_page':page_no,'exception_type':1})
  606. self.success = False
  607. break
  608. self.driver.get(
  609. f"https://search.jd.com/Search?keyword={kw}&enc=utf-8&wq={kw}",
  610. timeout=15,
  611. )
  612. self.sleep(3, 5)
  613. if page_no > 1:
  614. self._jump_to_page(page_no)
  615. if self._check_page_blocked():
  616. break
  617. self._success_report({'is_finished':0,'need_reassign':0,'current_page':page_no})
  618. logger.info("===== 正在爬取第 %s 页 =====", page_no)
  619. self.page_stored = 0
  620. search_ele = self.driver.ele("xpath=//div[@id='search-condition']", timeout=10)
  621. if not search_ele:
  622. self._error_report({'is_finished':0,'need_reassign':1,'current_page':page_no,'exception_type':2})
  623. logger.warning("未找到搜索结果区域,停止采集")
  624. break
  625. page_n, _ = self.collect_full_page_items()
  626. logger.info("本页监听商品条数(含可能重复): %s", page_n)
  627. total += page_n
  628. logger.info("累计监听条数: %s", total)
  629. self.cumulative_pages += 1
  630. self.cumulative_items += page_n
  631. page_skipped = page_n - self.page_stored
  632. self.cumulative_stored += self.page_stored
  633. self.cumulative_skipped += page_skipped
  634. logger.info(
  635. "关键字 %s 第 %s 页获取完成, 本页获取量: %d, 本页已入库: %d, 本页未入库: %d | 账号测试总页数: %d, 账号测试总数据量: %d, 总已入库: %d, 总未入库: %d",
  636. keyword,
  637. page_no,
  638. page_n,
  639. self.page_stored,
  640. page_skipped,
  641. self.cumulative_pages,
  642. self.cumulative_items,
  643. self.cumulative_stored,
  644. self.cumulative_skipped
  645. )
  646. print(f"[{keyword}] 当前页数: {page_no}, 本页获取数据量: {page_n}, 本页已入库: {self.page_stored}, 本页未入库: {page_skipped} | 账号测试总页数: {self.cumulative_pages}, 账号测试总数据量: {self.cumulative_items}, 总已入库: {self.cumulative_stored}, 总未入库: {self.cumulative_skipped}")
  647. if self.scheduler.end ==True:
  648. logger.info('心跳失败')
  649. self._error_report({'is_finished':0,'need_reassign':0,'current_page':page_no})
  650. break
  651. if self.is_no_prodcut > 20:
  652. logger.info("连续无匹配商品过多,停止采集")
  653. self._success_report({'is_finished':1,'need_reassign':0,'current_page':page_no})
  654. break
  655. if page_no >= self.end_page:
  656. self._success_report({'is_finished':1,'need_reassign':0,'current_page':page_no})
  657. break
  658. next_btn = self.driver.ele("text=下一页", timeout=2)
  659. if not next_btn:
  660. logger.info("没有下一页(未找到)")
  661. self._success_report({'is_finished':1,'need_reassign':0,'current_page':page_no})
  662. break
  663. cls_str = next_btn.attr("class") or ""
  664. if "disabled" in cls_str:
  665. logger.info("没有下一页(已禁用)")
  666. self._success_report({'is_finished':1,'need_reassign':0,'current_page':page_no})
  667. break
  668. if not self._go_next_page(next_btn):
  669. self._error_report({'is_finished':0,'need_reassign':1,'current_page':page_no,'exception_type':4})
  670. break
  671. if page_no==total_page: #达到最大页数
  672. self._success_report({'is_finished':1,'need_reassign':0,'current_page':page_no})
  673. break
  674. self.scheduler.stop()
  675. def run(self):
  676. # 检测账号
  677. # 启动守护线程
  678. # self._daemon_thread = threading.Thread(target=self._heartbeat_reporter)
  679. # self._daemon_thread.daemon = True # 守护线程,主线程退出时自动结束
  680. # self._daemon_thread.start()
  681. #if self.get_account():
  682. if 0:
  683. logger.info("==================当前无账号可用==================")
  684. self.success = False
  685. return self.pipeline.crawl_count, self.success
  686. logger.info("获取到账号:%s,代理ip:%s", self.account_name, self.ip)
  687. # # # 每次选取账号,立马账号使用时间
  688. update_sql = f""" UPDATE `retrieve_collect_equipment_account` SET `status`= %s, `update_time`= %s WHERE `username` = %s; """
  689. self.db.execute(update_sql, (0, int(time.time()), self.account_name))
  690. try:
  691. self.init_browser()
  692. self.crawl()
  693. except Exception as e:
  694. self.success = False
  695. logger.exception("爬取异常: %s", e)
  696. self.sleep(3, 5)
  697. self._error_report({'is_finished':0,'need_reassign':1,'current_page':self.start_page,'exception_type':3})
  698. finally:
  699. if self.driver:
  700. self.driver.quit()
  701. self.driver = None
  702. return self.pipeline.crawl_count, self.success