from playwright.sync_api import sync_playwright, TimeoutError as PlaywrightTimeoutError import os import json import random from logger_config import logger from config import * import re COOKIE_FILE_PATH = "yjj_cookies.json" # Cookie保存路径 LOGIN_VALIDATE_URL = "https://www.yyjzt.com/goodDetail?ladderNum&itemStoreId=124250306&sourceProdetail=%252Fsearch&is_store=0" TARGET_LOGIN_URL = "https://www.yyjzt.com/goodDetail?ladderNum&itemStoreId=124250306&sourceProdetail=%252Fsearch&is_store=0" def load_cookies(context, cookie_path=COOKIE_FILE_PATH): """从本地JSON文件加载Cookie到浏览器上下文""" if not os.path.exists(cookie_path): # logger.warning(f" Cookie文件不存在：{cookie_path}") return False try: with open(cookie_path, "r", encoding="utf-8") as f: cookies = json.load(f) context.add_cookies(cookies) # logger.info(f"✅ 已从{cookie_path}加载Cookie") return True except Exception as e: # logger.error(f" 加载Cookie失败：{e}") return False def is_login(page): """验证是否已登录（核心：检测登录态）""" try: # 访问需要登录的页面 page.goto(LOGIN_VALIDATE_URL, timeout=5000) page.wait_for_load_state("networkidle") # 检测是否跳转到登录页（URL包含login则未登录） if "login" in page.url.lower(): # logger.warning(" Cookie失效，需要重新登录") return False # 可选：检测登录后的专属元素（比如用户名、个人中心等） # if page.locator("用户中心选择器").count() > 0: # return True # logger.info(" Cookie有效，已保持登录状态") return True except Exception as e: # logger.error(f" 验证登录状态失败：{e}") return False def popup_guard(page, tag=""): """ 全局弹窗/遮罩守卫：多步引导 + 关闭按钮 + 遮罩清理 + 恢复滚动 tag 仅用于日志区分调用位置 """ try: # 给弹窗一点出现时间 page.wait_for_timeout(300) # 1) 连续点“下一步/完成/我知道了/关闭” for _ in range(6): btn = page.locator( "xpath=//button[normalize-space()='下一步' or normalize-space()='完成' or normalize-space()='我知道了' or normalize-space()='关闭']" ).first if btn.count() > 0 and btn.is_visible(): btn.click(timeout=1500) page.wait_for_timeout(250) continue # 2) 常见的 close icon close_btn = page.locator( "css=.el-dialog__headerbtn, .el-message-box__headerbtn, .close, .icon-close, .el-icon-close" ).first if close_btn.count() > 0 and close_btn.is_visible(): close_btn.click(timeout=1200) page.wait_for_timeout(250) continue break # 3) 清遮罩 + 恢复滚动/交互 page.evaluate(r""" () => { // 第一步：精准清理已知的遮罩/弹窗类名（Element UI框架常用） const selectors = [ '.v-modal', '.el-overlay', '.el-overlay-dialog', '.el-dialog__wrapper', '.el-message-box__wrapper', '.el-loading-mask' ]; selectors.forEach(sel => document.querySelectorAll(sel).forEach(e => e.remove())); // 泛化兜底：近似全屏 + 高 z-index 的层直接屏蔽 const all = Array.from(document.querySelectorAll('body *')); for (const el of all) { const s = getComputedStyle(el); // 获取元素的实际样式（含CSS生效的样式） const z = parseInt(s.zIndex || '0', 10); // 取元素的层级（z-index），默认0 // 条件1：元素是固定/绝对定位（弹窗/遮罩常见定位方式）+ 层级≥1000（高优先级遮挡）+ 能拦截鼠标事件 if ((s.position === 'fixed' || s.position === 'absolute') && z >= 1000 && s.pointerEvents !== 'none') { const r = el.getBoundingClientRect(); // 获取元素的尺寸和位置 // 条件2：元素宽度/高度≥屏幕80%（近似全屏遮罩） const nearFull = r.width >= innerWidth * 0.8 && r.height >= innerHeight * 0.8; if (nearFull) { el.style.pointerEvents = 'none'; // 让元素不拦截鼠标点击 el.style.display = 'none'; // 隐藏元素 } } } // 第三步：恢复页面滚动功能（弹窗常把页面设为不可滚动） document.documentElement.style.overflow = 'auto'; // html标签恢复滚动 document.body.style.overflow = 'auto'; // body标签恢复滚动 document.body.classList.remove('el-popup-parent--hidden'); // 移除Element UI的滚动禁用类 } """) # logger.info("杀除弹窗成功") except Exception: pass SEARCH_INPUT_SELECTOR = "input[placeholder*='药名/店铺/品牌/厂家']" def pick_search_input(page): """优先选可见且可用的搜索输入框；第一个不行就尝试第二个""" inputs = page.locator(SEARCH_INPUT_SELECTOR) cnt = inputs.count() # 优先检查前两个（你说只有两个） for i in range(min(cnt, 2)): candidate = inputs.nth(i) try: candidate.wait_for(state="visible", timeout=1500) # 小超时快速试探 if candidate.is_enabled(): return candidate except PlaywrightTimeoutError: continue # 兜底：直接找任意可见的（避免命中 hidden 模板） candidate = page.locator(f"{SEARCH_INPUT_SELECTOR}:visible").first candidate.wait_for(state="visible", timeout=5000) return candidate def type_slow(locator, text: str, min_delay=0.06, max_delay=0.18): """逐字输入，模拟真人打字""" for ch in text: locator.type(ch, delay=int(random.uniform(min_delay, max_delay) * 1000)) SEARCH_BTN_SELECTOR = ".el-button.ph-si-btn" def force_close_popup(page): """关闭新手引导/遮罩（多步：下一步/完成/我知道了），并兜底移除遮罩层""" try: # 1) 尝试连续点“下一步/完成/我知道了/关闭” for _ in range(5): # 最多点5次，足够覆盖多步引导 btn = page.locator( "//button[normalize-space()='下一步' or normalize-space()='完成' or normalize-space()='我知道了' or normalize-space()='关闭']" ).first if btn.count() > 0 and btn.is_visible(): btn.click(timeout=1500) page.wait_for_timeout(300) continue # 有些引导是右上角 X（如果存在就点） close_icon = page.locator( "xpath=//*[contains(@class,'close') or contains(@class,'el-icon-close') or name()='svg' or name()='i'][1]" ).first if close_icon.count() > 0 and close_icon.is_visible(): close_icon.click(timeout=1000) page.wait_for_timeout(300) continue break # 2) 兜底：移除常见遮罩层（element-ui / 通用 mask/overlay） page.evaluate(""" const selectors = [ '.v-modal', '.el-overlay', '.el-overlay-dialog', '.el-dialog__wrapper', '[class*="mask"]', '[class*="overlay"]', '[style*="z-index"]' ]; for (const sel of selectors) { document.querySelectorAll(sel).forEach(el => { const s = window.getComputedStyle(el); // 只移除“覆盖层”倾向的元素：fixed/absolute 且 z-index 很高 if ((s.position === 'fixed' || s.position === 'absolute') && parseInt(s.zIndex || '0', 10) >= 1000) { el.remove(); } }); } """) except Exception: pass def kill_masks(page): """ 强制清理残留遮罩层/覆盖层，并恢复 body 可滚动、可点击状态 """ page.evaluate(r""" () => { const removed = []; const hidden = []; // 1) 先处理已知常见遮罩 const knownSelectors = [ '.v-modal', '.el-overlay', '.el-overlay-dialog', '.el-dialog__wrapper', '.el-message-box__wrapper', '.el-loading-mask', '.el-popup-parent--hidden' ]; for (const sel of knownSelectors) { document.querySelectorAll(sel).forEach(el => { // v-modal / overlay 直接 remove 最省事 removed.push(sel); el.remove(); }); } // 2) 再做一次“泛化兜底”：全屏 fixed/absolute + 高 z-index 的覆盖层 // 注意：不要误删页面正常的固定导航，所以加上“近似全屏”的判断 const all = Array.from(document.querySelectorAll('body *')); for (const el of all) { const s = window.getComputedStyle(el); if (!s) continue; const z = parseInt(s.zIndex || '0', 10); const pos = s.position; const pe = s.pointerEvents; if ((pos === 'fixed' || pos === 'absolute') && z >= 1000 && pe !== 'none') { const r = el.getBoundingClientRect(); const nearFullScreen = r.width >= window.innerWidth * 0.8 && r.height >= window.innerHeight * 0.8 && r.left <= window.innerWidth * 0.1 && r.top <= window.innerHeight * 0.1; // 常见遮罩是半透明背景色，或者透明但拦截点击 const bg = s.backgroundColor || ''; const looksLikeMask = nearFullScreen && (bg.includes('rgba') || bg.includes('rgb') || s.opacity !== '1'); if (nearFullScreen) { // 不管透明不透明，只要近似全屏且高 z-index，就先让它不拦截点击 el.style.pointerEvents = 'none'; el.style.display = 'none'; hidden.push(el.tagName + '.' + (el.className || '')); } } } // 3) 恢复 body / html 的滚动与交互（很多弹窗会锁滚动） document.documentElement.style.overflow = 'auto'; document.body.style.overflow = 'auto'; document.body.style.position = 'static'; document.body.style.width = 'auto'; document.body.style.paddingRight = '0px'; // 4) 去掉 Element-UI 常见的锁定 class document.body.classList.remove('el-popup-parent--hidden'); return { removed, hiddenCount: hidden.length, hidden }; } """) # ==================== 搜索操作函数 ==================== def search_operation(page, keyword, has_opened_new_page=False): """搜索框填充+提交搜索""" try: # 1) 找到“可用”的搜索框（第一个不行就用第二个） search_locator = pick_search_input(page) # 清空并填充搜索框 search_locator.wait_for(timeout=5000) # 2. 清空搜索框（双重保障：先调用locator的clear，再手动全选删除） search_locator.click() # 聚焦 search_locator.fill("") page.keyboard.down("Control") # 按住Control键 page.keyboard.press("a") # 按a键 page.keyboard.up("Control") # 松开Control键 page.keyboard.press("Backspace") # 删除选中内容 # 3) 逐字输入 type_slow(search_locator, keyword, min_delay=0.25, max_delay=0.50) # 3. 输入搜索关键词 # search_locator.fill(keyword) logger.info(f"📝 已输入搜索关键词：{keyword}") # 3) 搜索按钮也建议点可见的那个 btn = page.locator(f"{SEARCH_BTN_SELECTOR}:visible").first btn.wait_for(state="visible", timeout=5000) btn.click() page.wait_for_timeout(600) try: page.wait_for_load_state("networkidle", timeout=10000) except Exception: pass force_close_popup(page) kill_masks(page) logger.info("✅ 已触发搜索") return True # 搜索后等待结果加载 # page.wait_for_timeout(COLLECT_DELAY) # return True except PlaywrightTimeoutError as e: logger.error(f" 搜索失败：元素定位超时 - {str(e)}") return False except Exception as e: logger.error(f" 搜索异常：{str(e)}") return False def main(): with sync_playwright() as p: browser = p.chromium.launch( headless=False, # 不要用无头模式（反爬：无头模式易被识别） channel="chrome", # 使用真实Chrome内核 slow_mo=random.randint(100, 300), # 全局操作延迟（模拟真人慢速操作） args=[ "--disable-blink-features=AutomationControlled", # 禁用webdriver特征（核心！） "--enable-automation=false", # 新增：禁用自动化标识 "--disable-infobars", # 新增：禁用信息栏 "--remote-debugging-port=0", # 新增：随机调试端口 "--start-maximized", # 最大化窗口（模拟真人使用） "--disable-extensions", # 禁用扩展（避免特征） "--disable-plugins-discovery", # 禁用插件发现 "--no-sandbox", # 避免沙箱模式特征 "--disable-dev-shm-usage", # 避免内存限制导致的异常 f"--user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/{random.randint(110, 120)}.0.0.0 Safari/537.36" # 随机Chrome版本的UA ] ) # 创建页面时伪装指纹 context = browser.new_context( locale="zh-CN", # 中文环境 timezone_id="Asia/Shanghai", # 上海时区 geolocation={"latitude": 31.230416, "longitude": 121.473701}, # 模拟上海地理位置（可选） permissions=["geolocation"], # 授予定位权限（模拟真人） user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36", viewport={"width": 1600, "height": 1400}, # 关键：隐藏自动化特征 java_script_enabled=True, bypass_csp=True, # user_data_dir="./temp_user_data" # 模拟真实用户数据目录 ) page = context.new_page() # 关键：移除navigator.webdriver标识（反爬核心） page.add_init_script(""" Object.defineProperty(navigator, 'webdriver', { get: () => undefined }); Object.defineProperty(navigator, 'plugins', { get: () => [1, 2, 3] }); // 新增：模拟插件 Object.defineProperty(navigator, 'mimeTypes', { get: () => [1, 2, 3] }); // 新增：模拟MIME类型 window.chrome = { runtime: {}, loadTimes: () => ({}) }; // 增强Chrome模拟 delete window.navigator.languages; window.navigator.languages = ['zh-CN', 'zh']; // 新增：模拟真实鼠标移动特征 (() => { const originalAddEventListener = EventTarget.prototype.addEventListener; EventTarget.prototype.addEventListener = function(type, listener) { if (type === 'mousemove') { return originalAddEventListener.call(this, type, (e) => { e._automation = undefined; listener(e); }); } return originalAddEventListener.call(this, type, listener); }; })(); """) try: # ========== 核心：Cookie复用逻辑 ========== # 1. 加载本地Cookie load_cookies(context) # 2. 验证登录状态 if not is_login(page): # 3. Cookie失效/不存在，执行登录 page.goto(TARGET_LOGIN_URL) page.wait_for_load_state("networkidle") # logger.info("🔑 开始执行登录流程") # 执行登录操作 # login_success = login_operation(page, USERNAME, PASSWORD) # if not login_success: # logger.error(" 登录失败，程序终止") # return # # 4. 登录成功后保存Cookie # save_cookies(context) # logger.info(" 登录并保存Cookie成功！") KEYWORDS = get_search_keywords_from_db() # get_search_keywords_from_db() # 执行搜索 total_num = 0 for kw in KEYWORDS: popup_guard(page, "before_search") search_success = search_operation(page, kw) if not search_success: print(f"❌ 搜索失败：{kw}") continue popup_guard(page, "after_search") #找不到数据跳过判断和出现杂数据跳过 not_found_keywords = page.locator("span:has-text('新品登记')") if not_found_keywords.count() > 0: logger.warning(f"⚠️ 关键词「{kw}」无匹配商品，直接跳过整个关键词采集") continue TARGET_SELECTOR = page.locator('div[data-v-4c22c8c9].sr-page_turner-pagination-total') total_count = 0 # ⚠️ 每一轮关键词都重置 if TARGET_SELECTOR.count() > 0: nums = TARGET_SELECTOR.inner_text(timeout=5000).strip() print(nums) match = re.search(r'\d+', nums) if match: total_count = int(match.group()) print(total_count) else: itme_boxes = page.locator("div[data-v-4c22c8c9].sr-list-item[data-item_loc]") total_count = itme_boxes.count() print(f"【{kw}】无分页，当前页盒子数：{total_count}") total_num += total_count print(f"截止到这个{kw}关键词有{total_num}条数据") page.wait_for_timeout(10000) print(f"✅ 本次采集总数据量：{total_num}") except Exception as e: print(f" 程序异常：{str(e)}") finally: browser.close() print(" 浏览器已关闭，程序结束") # ==================== 程序入口 ==================== if __name__ == '__main__': main()