python-qlv-helper 0.2.0__py3-none-any.whl

qlv_helper/po/main_page.py

@@ -0,0 +1,71 @@
+# -*- coding: utf-8 -*-
+"""
+# ---------------------------------------------------------------------------------------------------------
+# ProjectName:  qlv-helper
+# FileName:     main_page.py
+# Description:  首页页面对象
+# Author:       ASUS
+# CreateDate:   2025/11/25
+# Copyright ©2011-2025. Hunan xxxxxxx Company limited. All rights reserved.
+# ---------------------------------------------------------------------------------------------------------
+"""
+from typing import Tuple, Union
+from qlv_helper.po.base_po import BasePo
+from qlv_helper.utils.ocr_helper import get_image_text
+from playwright.async_api import Page, TimeoutError as PlaywrightTimeoutError, Locator
+
+
+class MainPage(BasePo):
+    url: str = "/"
+    __page: Page
+
+    def __init__(self, page: Page, url: str = "/") -> None:
+        super().__init__(page, url)
+        self.url = url
+        self.__page = page
+
+    async def get_confirm_btn_with_system_notice_dialog(self, timeout: float = 5.0) -> Tuple[bool, Union[Locator, str]]:
+        """
+        获取系统通知弹框中的确认按钮，注意这个地方，存在多个叠加的弹框，因此用last()方法，只需定位到最上面的那个弹框就行
+        :return:
+        """
+        selector: str = "//div[@class='CommonAlert'][last()]//a[@class='CommonAlertBtnConfirm']"
+        try:
+            locator = self.__page.locator(selector)
+            if locator:
+                await locator.wait_for(state='visible', timeout=timeout * 1000)
+                return True, locator
+            else:
+                return False, '没有找到首页中的【系统提醒-确定】按钮'
+        except PlaywrightTimeoutError:
+            return False, f"元素 '{selector}' 未在 {timeout} 秒内找到"
+        except Exception as e:
+            return False, f"检查元素时发生错误: {str(e)}"
+
+    async def get_level1_menu_order_checkout(self) -> Tuple[bool, Union[Locator, str]]:
+        selector: str = "//span[contains(normalize-space(), '订单出票')]"
+        try:
+            locator = self.__page.locator(selector)
+            if locator:
+                await locator.wait_for(state='visible', timeout=timeout * 1000)
+                return True, locator
+            else:
+                return False, '没有找到首页中的【订单出票】左侧一级导航菜单'
+        except PlaywrightTimeoutError:
+            return False, f"元素 '{selector}' 未在 {timeout} 秒内找到"
+        except Exception as e:
+            return False, f"检查元素时发生错误: {str(e)}"
+
+    async def get_level2_menu_order_checkout(self) -> Tuple[bool, Union[Locator, str]]:
+        selector: str = "//a[@menuname='国内活动订单']"
+        try:
+            locator = self.__page.locator(selector)
+            if locator:
+                await locator.wait_for(state='visible', timeout=timeout * 1000)
+                return True, locator
+            else:
+                return False, '没有找到首页中的【国内活动订单】左侧二级导航菜单'
+        except PlaywrightTimeoutError:
+            return False, f"元素 '{selector}' 未在 {timeout} 秒内找到"
+        except Exception as e:
+            return False, f"检查元素时发生错误: {str(e)}"

qlv_helper/po/wechat_auth_page.py

@@ -0,0 +1,68 @@
+# -*- coding: utf-8 -*-
+"""
+# ---------------------------------------------------------------------------------------------------------
+# ProjectName:  qlv-helper
+# FileName:     wechat_auth_page.py
+# Description:  微信认证页面对象
+# Author:       ASUS
+# CreateDate:   2025/11/27
+# Copyright ©2011-2025. Hunan xxxxxxx Company limited. All rights reserved.
+# ---------------------------------------------------------------------------------------------------------
+"""
+import os
+import asyncio
+from typing import Tuple, Union
+from qlv_helper.po.base_po import BasePo
+from qlv_helper.utils.file_handle import get_caller_dir
+from qlv_helper.utils.windows_utils import gen_allow_btn_image, windows_on_click
+from playwright.async_api import Page, TimeoutError as PlaywrightTimeoutError, Locator
+
+
+class WechatAuthPage(BasePo):
+    __page: Page
+
+    def __init__(self, page: Page, url: str = "/connect/qrconnect") -> None:
+        super().__init__(page, url)
+        self.__page = page
+
+    async def get_wechat_quick_login_btn(self, timeout: float = 5.0) -> Tuple[bool, Union[Locator, str]]:
+        selector: str = '//*[@id="tpl_for_page"]//button[text()="微信快捷登录"]'
+        try:
+            locator = self.__page.locator(selector)
+            if locator:
+                await locator.wait_for(state='visible', timeout=timeout * 1000)
+                return True, locator
+            else:
+                return False, '没有找到登录页面中的【微信快捷登录】按钮'
+        except PlaywrightTimeoutError:
+            return False, f"元素 '{selector}' 未在 {timeout} 秒内找到"
+        except Exception as e:
+            return False, f"检查元素时发生错误: {str(e)}"
+
+    async def get_dialog_allow_btn(self, timeout: float = 5.0) -> Tuple[bool, Union[Locator, str]]:
+        selector: str = '//div[@id="dialog_allow_btn"]'
+        try:
+            locator = self.__page.locator(selector)
+            if locator:
+                await locator.wait_for(state='visible', timeout=timeout * 1000)
+                return True, locator
+            else:
+                return False, '，没有找到申请登录弹框中的【允许】按钮'
+        except PlaywrightTimeoutError:
+            return False, f"元素 '{selector}' 未在 {timeout} 秒内找到"
+        except Exception as e:
+            return False, f"检查元素时发生错误: {str(e)}"
+
+    async def on_click_allow_btn(self, timeout: int = 5) -> Tuple[bool, str]:
+        try:
+            image_name = os.path.join(get_caller_dir(), "allow_btn_image.png")
+            gen_allow_btn_image(image_name=image_name)
+            await asyncio.sleep(delay=timeout)
+            windows_on_click(image_name=image_name, windows_title="微信", is_delete=True)
+            await asyncio.sleep(delay=timeout)
+            if self.is_current_page():
+                return False, "【微信快捷登录】登录失败，需要重新登录"
+            else:
+                return True, "【微信快捷登录】登录成功"
+        except Exception as e:
+            return False, str(e)

qlv_helper/utils/__init__.py

@@ -0,0 +1,11 @@
+# -*- coding: utf-8 -*-
+"""
+# ---------------------------------------------------------------------------------------------------------
+# ProjectName:  qlv-helper
+# FileName:     __init__.py
+# Description:  工具包
+# Author:       ASUS
+# CreateDate:   2025/11/25
+# Copyright ©2011-2025. Hunan xxxxxxx Company limited. All rights reserved.
+# ---------------------------------------------------------------------------------------------------------
+"""

qlv_helper/utils/browser_utils.py

@@ -0,0 +1,25 @@
+# -*- coding: utf-8 -*-
+"""
+# ---------------------------------------------------------------------------------------------------------
+# ProjectName:  qlv-helper
+# FileName:     browser_utils.py
+# Description:  浏览器工具模块
+# Author:       ASUS
+# CreateDate:   2025/11/27
+# Copyright ©2011-2025. Hunan xxxxxxx Company limited. All rights reserved.
+# ---------------------------------------------------------------------------------------------------------
+"""
+import asyncio
+from typing import Optional
+from playwright.async_api import BrowserContext, Page
+
+
+async def switch_for_table_window(browser: BrowserContext, url_keyword: str, wait_time: int = 10) -> Optional[Page]:
+    # 最多等待 wait_time 秒
+    for _ in range(wait_time):
+        await asyncio.sleep(1)
+        for page in browser.pages:
+            if url_keyword.lower() in page.url.lower():
+                await page.bring_to_front()
+                return page
+    return None

qlv_helper/utils/datetime_utils.py

@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+"""
+# ---------------------------------------------------------------------------------------------------------
+# ProjectName:  qlv-helper
+# FileName:     datetime_utils.py
+# Description:  时间处理工具模块
+# Author:       ASUS
+# CreateDate:   2025/11/28
+# Copyright ©2011-2025. Hunan xxxxxxx Company limited. All rights reserved.
+# ---------------------------------------------------------------------------------------------------------
+"""
+from datetime import datetime
+
+
+def get_current_dtstr() -> str:
+    return datetime.now().strftime("%Y%m%d%H%M%S")

qlv_helper/utils/file_handle.py

@@ -0,0 +1,33 @@
+# -*- coding: utf-8 -*-
+"""
+# ---------------------------------------------------------------------------------------------------------
+# ProjectName:  qlv-helper
+# FileName:     file_handle.py
+# Description:  文件处理工具模块
+# Author:       ASUS
+# CreateDate:   2025/11/25
+# Copyright ©2011-2025. Hunan xxxxxxx Company limited. All rights reserved.
+# ---------------------------------------------------------------------------------------------------------
+"""
+import os
+import inspect
+
+
+def get_caller_dir() -> str:
+    # 获取调用者的 frame
+    frame = inspect.stack()[1]
+    caller_file = frame.filename  # 调用者文件的完整路径
+    return os.path.dirname(os.path.abspath(caller_file))
+
+
+def save_image(file_name: str, img_bytes: bytes) -> None:
+    """
+    保存验证码图片到本地。
+    若文件已存在，会自动覆盖。
+    """
+    # 确保目录存在
+    os.makedirs(os.path.dirname(file_name), exist_ok=True)
+
+    # "wb" 会覆盖已有文件
+    with open(file_name, "wb") as f:
+        f.write(img_bytes)

qlv_helper/utils/html_utils.py

@@ -0,0 +1,59 @@
+# -*- coding: utf-8 -*-
+"""
+# ---------------------------------------------------------------------------------------------------------
+# ProjectName:  qlv-helper
+# FileName:     html_utils.py
+# Description:  html处理工具模块
+# Author:       zhouhanlin
+# CreateDate:   2025/12/01
+# Copyright ©2011-2025. Hunan xxxxxxx Company limited. All rights reserved.
+# ---------------------------------------------------------------------------------------------------------
+"""
+import re
+from bs4 import BeautifulSoup
+from typing import Dict, Any
+
+
+def parse_pagination_info(html: str) -> Dict[str, Any]:
+    """
+    解析分页信息
+    """
+    soup = BeautifulSoup(html, 'html.parser')
+
+    pagination_info = {
+        "current_page": 1,
+        "pages": 1,
+        "is_next_page": False,
+        "is_pre_page": False,
+        "total": 0,
+        "page_size": 20
+    }
+
+    # 找到分页的div
+    pagination_div = soup.find('div', {'class': 'redirect'})
+    if not pagination_div:
+        return pagination_info
+
+    # 解析两个label
+    labels = pagination_div.find_all('label')
+
+    if len(labels) >= 2:
+        # 当前页码
+        pagination_info["current_page"] = int(labels[0].get_text(strip=True))
+        # 总页数
+        pagination_info["pages"] = int(labels[1].get_text(strip=True))
+
+    # 解析显示记录信息
+    span_text = pagination_div.find('span').get_text(strip=True) if pagination_div.find('span') else ''
+    # 提取数字
+    numbers = re.findall(r'\d+', span_text)
+    if len(numbers) >= 3:
+        pagination_info["page_size"] = int(numbers[1])
+        pagination_info["total"] = int(numbers[2])
+
+    if pagination_info["current_page"] == 1:
+        pagination_info["is_pre_page"] = False
+    if pagination_info["pages"] > 1 and pagination_info["current_page"] < pagination_info["pages"]:
+        pagination_info["is_next_page"] = True
+
+    return pagination_info

qlv_helper/utils/ocr_helper.py

@@ -0,0 +1,83 @@
+# -*- coding: utf-8 -*-
+"""
+# ---------------------------------------------------------------------------------------------------------
+# ProjectName:  qlv-helper
+# FileName:     ocr_helper.py
+# Description:  OCR模块
+# Author:       ASUS
+# CreateDate:   2025/11/25
+# Copyright ©2011-2025. Hunan xxxxxxx Company limited. All rights reserved.
+# ---------------------------------------------------------------------------------------------------------
+"""
+import ddddocr
+import requests
+from typing import Union, Tuple
+from aiohttp import ClientSession
+from playwright.async_api import Page, TimeoutError as PlaywrightTimeoutError
+
+# 复用 OCR 实例，不用每次都重新加载模型（更快）
+_ocr = ddddocr.DdddOcr(show_ad=False)
+
+
+def fetch_and_ocr_captcha(url: str) -> Tuple[str, bytes]:
+    # 1) 请求验证码图片
+    resp = requests.get(url, timeout=10)
+    img_bytes = resp.content
+
+    # 2) OCR 识别
+    result = _ocr.classification(img_bytes)
+
+    return result, img_bytes
+
+
+async def async_fetch_and_ocr_captcha(url: str) -> Tuple[str, bytes]:
+    async with ClientSession() as session:
+        async with session.get(url, timeout=10) as resp:
+            img_bytes = await resp.read()
+
+    result = _ocr.classification(img_bytes)
+    return result, img_bytes
+
+
+def recognize_captcha(image: Union[str, bytes]) -> str:
+    """
+    识别验证码图片，返回识别文本。
+    参数:
+        image: 图片路径 str，或图片的二进制 bytes
+    返回:
+        识别出的验证码字符串
+    """
+    try:
+        # 如果是路径，读取文件
+        if isinstance(image, str):
+            with open(image, "rb") as f:
+                img_bytes = f.read()
+        else:
+            img_bytes = image
+
+        result = _ocr.classification(img_bytes)
+        return result
+
+    except Exception as e:
+        raise RuntimeError(f"OCR 识别失败: {e}")
+
+
+async def get_image_text(page: Page, selector: str, timeout: float = 5.0) -> Tuple[bool, str]:
+    try:
+        # 找到 img
+        locator = page.locator(selector)
+        if locator:
+            img = await locator.element_handle(timeout=timeout * 1000)
+
+            # 直接截图获取原始图片字节，不刷新图片
+            img_bytes = await img.screenshot(timeout=timeout * 1000)
+
+            # OCR 识别
+            text = _ocr.classification(img_bytes)
+            return True, text.strip()
+        else:
+            return False, f'没有找到当前页面中的【{selector}】图片'
+    except PlaywrightTimeoutError:
+        return False, f"元素 '{selector}' 未在 {timeout} 秒内找到"
+    except Exception as e:
+        return False, f"检查元素时发生错误: {str(e)}"

qlv_helper/utils/po_utils.py

@@ -0,0 +1,113 @@
+# -*- coding: utf-8 -*-
+"""
+# ---------------------------------------------------------------------------------------------------------
+# ProjectName:  qlv-helper
+# FileName:     po_utils.py
+# Description:  po对象处理工具
+# Author:       ASUS
+# CreateDate:   2025/11/25
+# Copyright ©2011-2025. Hunan xxxxxxx Company limited. All rights reserved.
+# ---------------------------------------------------------------------------------------------------------
+"""
+from typing import Literal, Union, Tuple
+from playwright.async_api import Page, Locator
+
+MouseButton = Literal["left", "middle", "right"]
+
+
+async def on_click_locator(locator: Locator, button: MouseButton = "left") -> bool:
+    try:
+        await locator.click(button=button)
+        return True
+    except(Exception,):
+        return False
+
+
+async def on_click_element(page: Page, selector: str, button: MouseButton = "left") -> bool:
+    try:
+        await page.locator(selector).click(button=button)
+        return True
+    except(Exception,):
+        return False
+
+
+async def locator_input_element(locator: Locator, text: str) -> bool:
+    """
+    输入内容，输入前会自动清空原来的内容
+    :param locator: Locator对象
+    :param text: 输入的内容
+    :return:
+    """
+    try:
+        # 填入文字，会清空原内容
+        if isinstance(text, str) is False:
+            text = str(text)
+        await locator.fill(text)
+        return True
+    except (Exception,):
+        return False
+
+
+async def simulation_input_element(locator: Locator, text: str, delay: int = 100) -> bool:
+    """
+    模拟逐字符输入（可用于触发 JS 事件）
+    :param locator: 元素定位器对象
+    :param text: 输入的内容
+    :param int delay: 每个字符延迟输入 100ms
+    :return:
+    """
+    try:
+        if isinstance(text, str) is False:
+            text = str(text)
+        await locator.type(text, delay=delay)
+        return True
+    except (Exception,):
+        return False
+
+
+async def page_screenshot(page: Page, file_name: str = None, is_full_page: bool = False) -> Union[bytes, None]:
+    """
+    保存截图，默认截图整个可见页面（不是整个滚动区域）
+    :param page:
+    :param file_name: 文件名，若没传递，这返回bytes
+    :param is_full_page: true---> 截图整个网页（包括滚动区域）
+    :return:
+    """
+    if is_full_page is True:
+        if file_name is None:
+            return await page.screenshot()
+        await page.screenshot(path=file_name, full_page=True)
+    else:
+        await page.screenshot(path=file_name)
+
+
+async def locator_element_screenshot(locator: Locator, file_name: str = None) -> bytes:
+    """
+    元素截图
+    :param locator: Locator对象
+    :param file_name: 文件名，若没传递，这返回bytes
+    :return:
+    """
+    if file_name is None:
+        return await locator.screenshot(path=file_name)
+
+    else:
+        await locator.screenshot(path=file_name)
+
+
+async def is_exists_value(element: Locator) -> Tuple[bool, Union[str, None]]:
+    """
+    判断一个元素是否存在值
+    :param element:
+    :return:
+    """
+    value = await element.input_value()
+    # value = await element.get_attribute("value")
+    if value:
+        value = value.strip()
+        if len(value) > 0:
+            return True, value
+        else:
+            return False, None
+    else:
+        return False, None

qlv_helper/utils/stealth_browser.py

@@ -0,0 +1,100 @@
+# -*- coding: utf-8 -*-
+"""
+# ---------------------------------------------------------------------------------------------------------
+# ProjectName:  qlv-helper
+# FileName:     stealth_browser.py
+# Description:  防检测浏览器配置
+# Author:       ASUS
+# CreateDate:   2025/11/27
+# Copyright ©2011-2025. Hunan xxxxxxx Company limited. All rights reserved.
+# ---------------------------------------------------------------------------------------------------------
+"""
+from playwright.async_api import Page
+
+CHROME_STEALTH_ARGS = [
+    # 核心防检测
+    '--disable-blink-features=AutomationControlled',
+    '--disable-automation-controlled-blink-features',
+
+    # 隐藏"Chrome正受到自动测试软件控制"提示
+    '--disable-infobars',
+    '--disable-popup-blocking',
+
+    # 性能优化
+    '--no-first-run',
+    '--no-default-browser-check',
+    '--disable-default-apps',
+    '--disable-translate',
+
+    # 禁用自动化标志
+    '--disable-background-timer-throttling',
+    '--disable-backgrounding-occluded-windows',
+    '--disable-renderer-backgrounding',
+
+    # 网络和安全
+    '--disable-web-security',
+    '--disable-features=VizDisplayCompositor',
+    '--disable-features=RendererCodeIntegrity',
+    '--remote-debugging-port=0',  # 随机端口
+
+    # 硬件相关（减少特征）
+    '--disable-gpu',
+    '--disable-software-rasterizer',
+    '--disable-dev-shm-usage',
+]
+
+IGNORE_ARGS = [
+    '--enable-automation',
+    '--enable-automation-controlled-blink-features',
+    '--password-store=basic',  # 避免密码存储提示
+]
+
+USER_AGENT = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/142.0.0.0 Safari/537.36"
+
+viewport = {'width': 1920, 'height': 1080}
+
+
+async def setup_stealth_page(page: Page):
+    """设置页面为隐身模式"""
+    # 修改 navigator.webdriver
+    await page.add_init_script("""
+        // 进一步修改 navigator 属性
+        Object.defineProperty(navigator, 'webdriver', {
+            get: () => undefined,
+        });
+        Object.defineProperty(navigator, 'plugins', {
+            get: () => [1, 2, 3, 4, 5],
+        });
+        Object.defineProperty(navigator, 'languages', {
+            get: () => ['zh-CN', 'zh', 'en'],
+        });
+        
+        // 删除 webdriver 属性
+        delete navigator.__proto__.webdriver;
+
+        // 修改 plugins
+        Object.defineProperty(navigator, 'plugins', {
+            get: () => [{
+                name: 'Chrome PDF Plugin',
+                filename: 'internal-pdf-viewer'
+            }],
+        });
+
+        // 修改 languages
+        Object.defineProperty(navigator, 'languages', {
+            get: () => ['zh-CN', 'zh', 'en-US', 'en'],
+        });
+
+        // 修改 platform
+        Object.defineProperty(navigator, 'platform', {
+            get: () => 'Win32',
+        });
+
+        // 隐藏 chrome 对象
+        window.chrome = {
+            runtime: {},
+            loadTimes: function(){},
+            csi: function(){},
+            app: {}
+        };
+    """)