import logging import os import sys import time import random import shutil import cv2 import numpy as np from playwright.sync_api import sync_playwright, Page from playwright.sync_api import TimeoutError as PlaywrightTimeoutError from playwright.sync_api import Error as PlaywrightError # 将 SakuraFrpQiandao 加入搜索路径，复用其模块 sys.path.insert(0, os.path.join(os.path.dirname(__file__), "SakuraFrpQiandao")) from config import Config from captcha_handler import CaptchaHandler, ModelApiError from human_simulator import HumanSimulator logging.basicConfig( level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s', handlers=[ logging.StreamHandler(), logging.FileHandler('checkin.log', encoding='utf-8'), ], ) logger = logging.getLogger(__name__) def get_gap_offset(bg_path: str, fullbg_path: str) -> int: """OpenCV 滑块缺口检测""" bg = cv2.imread(bg_path, 0) fullbg = cv2.imread(fullbg_path, 0) diff = cv2.absdiff(fullbg, bg) _, thresh = cv2.threshold(diff, 25, 255, cv2.THRESH_BINARY) contours, _ = cv2.findContours(thresh, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE) if len(contours) < 2: raise ValueError("轮廓数量不足，滑块识别失败") sorted_contours = sorted(contours, key=cv2.contourArea, reverse=True)[:2] rects = [cv2.boundingRect(c) for c in sorted_contours] target = max(rects, key=lambda r: r[0]) x, y, w, h = target logger.info(f"滑块缺口坐标: x={x}, y={y}, w={w}, h={h}") return x def drag_slider(page: Page, slider, canvas_bg, gap_x: int): """模拟人类拖动滑块，根据 canvas 缩放比例校正距离""" # 计算 canvas 显示宽度与实际截图宽度的比例 canvas_box = canvas_bg.bounding_box() bg_img = cv2.imread("bg.png", 0) img_w = bg_img.shape[1] # 截图宽度（像素） display_w = canvas_box['width'] # CSS 显示宽度 scale = display_w / img_w if img_w > 0 else 1.0 distance = int(gap_x * scale) logger.info(f"Canvas 缩放比例: {scale:.3f} (img={img_w}px, display={display_w:.1f}px), 校正距离: {distance}px") slider_box = slider.bounding_box() start_x = slider_box['x'] + slider_box['width'] / 2 start_y = slider_box['y'] + slider_box['height'] / 2 logger.info(f"滑块起点: ({start_x:.0f}, {start_y:.0f})") page.mouse.move(start_x, start_y) page.mouse.down() page.wait_for_timeout(random.randint(200, 400)) # 分段拖动到目标位置过一点，再回退微调（模拟人类） page.mouse.move(start_x + distance + random.randint(2, 8), start_y + random.uniform(-1, 1)) page.wait_for_timeout(random.randint(300, 600)) page.mouse.move(start_x + distance - random.randint(2, 5), start_y) page.wait_for_timeout(random.randint(200, 400)) page.mouse.move(start_x + distance - random.randint(5, 8), start_y) page.mouse.up() logger.info("滑块拖动完成") def solve_slider_captcha(page: Page) -> bool: """处理滑块验证码（OpenCV 方案）""" logger.info("检测到滑块验证码，使用 OpenCV 方案处理...") try: # 截图含缺口的背景 bg = page.query_selector('.geetest_canvas_bg') if not bg: logger.error("未找到滑块背景 canvas") return False bg.screenshot(path="bg.png") # 显示完整背景并截图 page.evaluate( "document.querySelector('.geetest_canvas_fullbg').style.display = 'block';" ) fullbg = page.query_selector('.geetest_canvas_fullbg') fullbg.screenshot(path="fullbg.png") page.evaluate( "document.querySelector('.geetest_canvas_fullbg').style.display = 'none';" ) gap_x = get_gap_offset("bg.png", "fullbg.png") slider = page.query_selector('.geetest_slider_button') if not slider: logger.error("未找到滑块按钮") return False drag_slider(page, slider, bg, gap_x) time.sleep(2) # 检查成功文案或 slider 消失 try: page.wait_for_selector('.geetest_slider_button', state='detached', timeout=3000) logger.info("滑块已消失，验证通过") return True except PlaywrightTimeoutError: logger.warning("滑块未消失，可能未通过") return False except Exception as e: logger.error(f"滑块验证处理出错: {e}", exc_info=True) return False def detect_captcha_type(page: Page) -> str: """检测验证码类型: 'slider' | 'grid' | 'none'""" try: slider = page.query_selector('.geetest_slider_button') if slider and slider.is_visible(): return 'slider' except Exception: pass try: grid = page.query_selector('.geetest_tip_img') if grid and grid.is_visible(): return 'grid' except Exception: pass return 'none' class CheckInBot: def __init__(self, config: Config): self.config = config self.captcha_handler = CaptchaHandler(config) self.simulator = HumanSimulator() self.max_retries = config.max_retries def run(self): headless = os.getenv('CI') == 'true' or os.getenv('HEADLESS', 'false').lower() == 'true' with sync_playwright() as playwright: browser = None try: launch_opts = { "headless": headless, "args": [ "--window-size=1280,800", "--disable-blink-features=AutomationControlled", "--no-proxy-server", "--lang=zh-CN", "--disable-gpu", "--no-sandbox", "--disable-dev-shm-usage", ], } chrome_path = self._resolve_chrome() if chrome_path: launch_opts["executable_path"] = chrome_path browser = playwright.chromium.launch(**launch_opts) context = browser.new_context( viewport={"width": 1280, "height": 800}, locale="zh-CN", 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" ), ) context.add_init_script( "Object.defineProperty(navigator, 'webdriver', { get: () => undefined });" ) page = context.new_page() page.set_default_timeout(20000) self._login(page) self._navigate_dashboard(page) self._do_checkin(page) logger.info("签到流程完成") except ModelApiError: logger.error("模型 API 调用失败，终止") except Exception as e: logger.error(f"执行出错: {e}", exc_info=True) finally: if browser and headless: browser.close() def _resolve_chrome(self): if self.config.chrome_binary_path and os.path.exists(self.config.chrome_binary_path): return self.config.chrome_binary_path for exe in ("google-chrome", "google-chrome-stable", "chrome", "chromium"): found = shutil.which(exe) if found: return found return None def _login(self, page: Page): url = "https://www.natfrp.com/user/" logger.info(f"导航到登录页: {url}") page.goto(url, wait_until="domcontentloaded") user_input = page.locator("#username") pass_input = page.locator("#password") user_input.wait_for(state="visible") pass_input.wait_for(state="visible") logger.info("输入登录凭据...") user_input.fill("") self.simulator.type_text(user_input, self.config.sakurafrp_user) pass_input.fill("") self.simulator.type_text(pass_input, self.config.sakurafrp_pass) login_btn = page.locator("#login") login_btn.wait_for(state="visible") login_btn.click() logger.info("登录成功") def _navigate_dashboard(self, page: Page): self.simulator.random_sleep(3, 5) self._dismiss_age_check(page) def _dismiss_age_check(self, page: Page) -> bool: """尝试多种方式解除年龄确认弹窗""" # 方式1: 检测 adult-check 弹窗，点击"是"按钮 try: adult_check = page.locator(".adult-check") if adult_check.count() > 0 and adult_check.first.is_visible(): logger.info("检测到年龄确认弹窗...") yes_btn = page.locator(".adult-check button").first yes_btn.wait_for(state="visible", timeout=3000) yes_btn.click(force=True) logger.info("已点击年龄确认按钮（force click）") time.sleep(1) return True except Exception: pass # 方式2: 匹配文本"是，我已满18岁" try: age = page.locator("div.yes a", has_text="是，我已满18岁") age.wait_for(state="visible", timeout=3000) age.click() logger.info("已处理年龄确认（文本匹配）") time.sleep(1) return True except PlaywrightTimeoutError: pass logger.info("无需年龄确认") return False def _already_signed(self, page: Page) -> bool: try: page.wait_for_selector("text=今天已经签到过啦", timeout=3000) return True except PlaywrightTimeoutError: return False def _do_checkin(self, page: Page): for attempt in range(1, self.max_retries + 1): logger.info(f"签到尝试 {attempt}/{self.max_retries}") # 先检查是否已签到 if self._already_signed(page): logger.info("今日已签到") return btn = None try: btn = page.locator("button", has_text="点击这里签到") btn.wait_for(state="visible", timeout=5000) except PlaywrightTimeoutError: # 等久一点再检查一次 time.sleep(2) if self._already_signed(page): logger.info("今日已签到（延迟检测）") return logger.error("未找到签到按钮，也未检测到已签到标识") return logger.info("点击签到按钮...") self._dismiss_age_check(page) # 二次确认，防止弹窗阻挡点击 btn.click(force=True) self.simulator.random_sleep(2, 4) captcha_type = detect_captcha_type(page) logger.info(f"验证码类型: {captcha_type}") if captcha_type == 'slider': if solve_slider_captcha(page): page.reload(wait_until="domcontentloaded") time.sleep(3) if self._already_signed(page): logger.info("滑块验证 + 签到成功") return else: page.reload(wait_until="domcontentloaded") time.sleep(3) continue elif captcha_type == 'grid': try: self.captcha_handler.handle_geetest_captcha(page) page.reload(wait_until="domcontentloaded") time.sleep(3) if self._already_signed(page): logger.info("九宫格验证 + 签到成功") return except ModelApiError: raise except Exception as e: logger.error(f"九宫格处理出错: {e}") elif captcha_type == 'none': time.sleep(3) if self._already_signed(page): logger.info("签到成功（无验证码）") return page.reload(wait_until="domcontentloaded") time.sleep(3) logger.warning("已达最大重试次数") def main(): try: config = Config.from_env() logger.info(f"使用账户: {config.sakurafrp_user}") bot = CheckInBot(config) bot.run() except ValueError as e: logger.error(f"配置错误: {e}") except Exception as e: logger.error(f"程序执行失败: {e}", exc_info=True) if __name__ == "__main__": main()