jd_auto_crawl.py 31 KB

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