qrpa 1.0.9__py3-none-any.whl → 1.0.11__py3-none-any.whl

qrpa/fun_web.py

@@ -0,0 +1,148 @@
+import json
+from typing import Optional, Union
+from playwright.sync_api import Page
+
+from .fun_base import log, send_exception
+from .time_utils import get_current_datetime
+
+
+def fetch(page: Page, url: str, params: Optional[Union[dict, list, str]] = None, headers: Optional[dict] = None, config:
+Optional[dict] = None) -> dict:
+    """
+    发送 HTTP POST 请求，支持自定义 headers。
+
+    :param page: Playwright 的 Page 对象
+    :param url: 请求地址
+    :param params: 请求参数（dict、list、str 或 None）
+    :param headers: 自定义 headers 字典
+    :return: 服务器返回的 JSON 响应（dict）
+    """
+    if params is not None and not isinstance(params, (dict, list, str)):
+        raise ValueError("params 参数必须是 dict、list、str 或 None")
+    if headers is not None and not isinstance(headers, dict):
+        raise ValueError("headers 参数必须是 dict 或 None")
+
+    try:
+        response = page.evaluate("""
+            async ({ url, params, extraHeaders }) => {
+                try {
+                    const defaultHeaders = {
+                        'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/133.0.0.0 Safari/537.36',
+                        'x-requested-with': 'XMLHttpRequest',
+                    };
+
+                    const headers = Object.assign({}, defaultHeaders, extraHeaders || {});
+                    const options = {
+                        method: 'POST',
+                        credentials: 'include',
+                        headers: headers
+                    };
+
+                    if (params !== null) {
+                        if (typeof params === 'string') {
+                            options.headers['Content-Type'] = 'application/x-www-form-urlencoded; charset=UTF-8';
+                            options.body = params;
+                        } else {
+                            options.headers['Content-Type'] = 'application/json';
+                            options.body = JSON.stringify(params);
+                        }
+                    }
+
+                    const response = await fetch(url, options);
+                    if (!response.ok) {
+                        throw new Error(`HTTP ${response.status} - ${response.statusText}`);
+                    }
+                    return await response.json();
+                } catch (error) {
+                    return { "error": "fetch_failed", "message": error.message };
+                }
+            }
+        """, {"url": url, "params": params, "extraHeaders": headers})
+
+        return response
+    except Exception as e:
+        raise send_exception()
+        # return {"error": "fetch error", "message": str(e)}
+
+
+def fetch_via_iframe(page: Page, target_domain: str, url: str, params: Optional[Union[dict, list, str]] = None, config:
+Optional[dict] = None) -> dict:
+    """
+    方案 2：在 iframe 内部执行 fetch 请求，绕过 CORS 限制
+
+    :param page: Playwright 的 Page 对象
+    :param url: 目标请求的 URL
+    :param target_domain: 目标 iframe 所在的域名（用于匹配 iframe）
+    :param params: 请求参数（dict、list、str 或 None）
+    :return: 服务器返回的 JSON 响应（dict）
+    """
+    if params is not None and not isinstance(params, (dict, list, str)):
+        raise ValueError("params 参数必须是 dict、list、str 或 None")
+    response = None
+    try:
+        # 获取所有 iframe，查找目标域名的 iframe
+        frames = page.frames
+        target_frame = None
+        for frame in frames:
+            if target_domain in frame.url:
+                target_frame = frame
+                break
+
+        if not target_frame:
+            return {"error": "iframe_not_found", "message": f"未找到包含 {target_domain} 的 iframe"}
+
+        response = target_frame.evaluate("""
+            async ({ url, params }) => {
+                try {
+                    const options = {
+                        method: 'POST',
+                        credentials: 'include',
+                        headers: {
+                            'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/133.0.0.0 Safari/537.36',
+                            'x-requested-with': 'XMLHttpRequest',
+                        }
+                    };
+
+                    if (params !== null) {
+                        if (typeof params === 'string') {
+                            options.headers['Content-Type'] = 'application/x-www-form-urlencoded; charset=UTF-8';
+                            options.body = params;
+                        } else {
+                            options.headers['Content-Type'] = 'application/json';
+                            options.body = JSON.stringify(params);
+                        }
+                    }
+
+                    const response = await fetch(url, options);
+                    if (!response.ok) {
+                        throw new Error(`HTTP ${response.status} - ${response.statusText}`);
+                    }
+                    return await response.json();
+                } catch (error) {
+                    return { "error": "iframe_fetch_failed", "message": error.message };
+                }
+            }
+        """, {"url": url, "params": params})
+
+        return response
+    except Exception as e:
+       raise send_exception()
+       # return {"error": "iframe_exception", "message": str(e)}
+
+
+# 找到一个页面里面所有的iframe
+def find_all_iframe(page: Page):
+    frames = page.frames
+    for frame in frames:
+        log("找到 iframe:", frame.url)
+    return [frame.url for frame in frames]
+
+
+# 全屏幕截图
+def full_screen_shot(web_page: Page, config):
+    # 设置页面的视口大小为一个较大的值，确保截图高清
+    web_page.set_viewport_size({"width": 1920, "height": 1080})
+    # 截取全页面的高清截图
+    full_screenshot_image_path = f'{config.auto_dir}/screenshot/{get_current_datetime()}.png'
+    web_page.screenshot(path=full_screenshot_image_path, full_page=True)
+    return full_screenshot_image_path

qrpa/fun_win.py

@@ -1,7 +1,58 @@
 import os
 import win32com.client
+import winreg
 
-from .fun_base import log
+import requests, subprocess, time
+from contextlib import contextmanager
+
+from .fun_base import log, create_file_path
+
+default_chrome_user_data = 'D:\chrome_user_data'
+
+def set_chrome_system_path():
+    path = os.path.dirname(find_software_install_path('chrome'))
+    add_to_system_path(path)
+
+def add_to_system_path(path: str, scope: str = "user"):
+    """
+    将指定路径添加到系统环境变量 Path 中
+    :param path: 要添加的路径（应为绝对路径）
+    :param scope: 'user' 表示用户变量，'system' 表示系统变量（需要管理员权限）
+    """
+    if not os.path.isabs(path):
+        raise ValueError("必须提供绝对路径")
+
+    path = os.path.normpath(path)
+
+    if scope == "user":
+        root = winreg.HKEY_CURRENT_USER
+        subkey = r"Environment"
+    elif scope == "system":
+        root = winreg.HKEY_LOCAL_MACHINE
+        subkey = r"SYSTEM\CurrentControlSet\Control\Session Manager\Environment"
+    else:
+        raise ValueError("scope 参数必须是 'user' 或 'system'")
+
+    try:
+        with winreg.OpenKey(root, subkey, 0, winreg.KEY_READ | winreg.KEY_WRITE) as key:
+            current_path, _ = winreg.QueryValueEx(key, "Path")
+            paths = current_path.split(";")
+
+            if path in paths:
+                print("路径已存在于 Path 中，无需添加: ", path)
+                return False
+
+            new_path = current_path + ";" + path
+            winreg.SetValueEx(key, "Path", 0, winreg.REG_EXPAND_SZ, new_path)
+            print("✅ 路径已成功添加到Path中: ", new_path)
+            return True
+
+    except PermissionError:
+        print("❌ 权限不足，系统变量修改需要管理员权限")
+        return False
+    except Exception as e:
+        print(f"❌ 添加失败: {e}")
+        return False
 
 def find_software_install_path(app_keyword: str):
     """从开始菜单或桌面查找指定软件的安装路径"""
@@ -29,3 +80,119 @@ def find_software_install_path(app_keyword: str):
 
     log(f'未能查找到{str}安装位置')
     return None
+
+def init_chrome_env(account_list):
+    target = find_software_install_path('chrome')
+    for account in account_list:
+        store_key, port, *rest = account
+        user_data = rest[0] if rest and rest[0] else fr'{default_chrome_user_data}\{port}'
+        create_file_path(user_data)
+        args = fr'--remote-debugging-port={port} --user-data-dir="{user_data}"'
+        shortcut_name = f'{port}_{store_key}.lnk'
+        create_shortcut_on_desktop(target_path=target, arguments=args, shortcut_name=shortcut_name)
+
+def create_shortcut_on_desktop(target_path, arguments='', shortcut_name='MyShortcut.lnk', icon_path=None):
+    """
+    在桌面上创建快捷方式，若已存在指向相同 target + arguments 的快捷方式，则跳过创建。
+    """
+    # 获取当前用户桌面路径
+    desktop_path = os.path.join(os.environ['USERPROFILE'], 'Desktop')
+
+    shell = win32com.client.Dispatch('WScript.Shell')
+
+    # 检查是否已有相同目标的快捷方式
+    for file in os.listdir(desktop_path):
+        if file.lower().endswith('.lnk'):
+            shortcut_file = os.path.join(desktop_path, file)
+            shortcut = shell.CreateShortCut(shortcut_file)
+            if (os.path.normpath(shortcut.Targetpath) == os.path.normpath(target_path)
+                    and shortcut.Arguments.strip() == arguments.strip()):
+                log("已存在指向该 target + args 的快捷方式，跳过创建")
+                return
+
+    # 创建新的快捷方式
+    shortcut_path = os.path.join(desktop_path, shortcut_name)
+    shortcut = shell.CreateShortCut(shortcut_path)
+    shortcut.Targetpath = target_path
+    shortcut.Arguments = arguments
+    shortcut.WorkingDirectory = os.path.dirname(target_path)
+    if icon_path:
+        shortcut.IconLocation = icon_path
+    shortcut.save()
+    log(f"已创建快捷方式：{shortcut_path}")
+
+def check_chrome_dev(port=3000):
+    try:
+        url = f"http://127.0.0.1:{port}/json"
+        response = requests.get(url, timeout=5)  # 设置超时，避免长时间等待
+        if response.status_code == 200:
+            try:
+                data = response.json()
+                if data:
+                    # print("接口返回了数据：", data)
+                    print("接口返回了数据：")
+                    return True
+                else:
+                    print("接口返回了空数据")
+                    return False
+            except ValueError:
+                print("返回的不是有效的 JSON")
+                return False
+        else:
+            print(f"接口返回了错误状态码: {response.status_code}")
+            return False
+    except requests.RequestException as e:
+        print(f"请求接口时发生错误: {e}")
+        return False
+
+def is_chrome_running():
+    try:
+        output = subprocess.check_output('tasklist', shell=True, text=True)
+        return 'chrome.exe' in output.lower()
+    except subprocess.CalledProcessError:
+        return False
+
+@contextmanager
+def get_chrome_page_v3(p, port=3000, user_data=None):
+    browser = context = page = None
+    is_custom_chrome_opened = False  # 标记是否是程序自己开的浏览器
+
+    try:
+        if not check_chrome_dev(port):
+            set_chrome_system_path()
+            chrome_path = r'"chrome.exe"'
+            debugging_port = fr"--remote-debugging-port={port}"
+            if user_data is not None:
+                chrome_user_data = fr'--user-data-dir="{user_data}"'
+            else:
+                chrome_user_data = fr'--user-data-dir="{create_file_path(default_chrome_user_data)}\{port}"'
+
+            disable_webrtc = "--disable-features=WebRTC"
+            disable_webrtc_hw_encoder = "--disable-features=WebRTC-HW-ENCODER"
+            disable_webrtc_alt = "--disable-webrtc"
+            start_maximized = "--start-maximized"
+
+            command = f"{chrome_path} {debugging_port} {chrome_user_data} {disable_webrtc} {disable_webrtc_hw_encoder} {disable_webrtc_alt}"
+            subprocess.Popen(command, shell=True)
+            is_custom_chrome_opened = True
+            time.sleep(1)
+
+        browser = p.chromium.connect_over_cdp(f"http://127.0.0.1:{port}")
+        context = browser.contexts[0] if browser.contexts else browser.new_context()
+        page = context.pages[0] if context.pages else context.new_page()
+
+        yield browser, context, page
+
+    except Exception as e:
+        # 向上抛出错误，否则主函数感知不到错误
+        raise
+
+    finally:
+        for obj in [("page", page), ("context", context), ("browser", browser)]:
+            name, target = obj
+            try:
+                if target and is_custom_chrome_opened:
+                    log(f'关闭: {name}')
+                    target.close()
+            except Exception:
+                pass  # 你可以在这里加日志记录关闭失败

qrpa/shein_ziniao.py

@@ -24,6 +24,7 @@ from .fun_win import find_software_install_path
 from .fun_base import log, hostname, send_exception
 from .fun_file import check_progress_json_ex, get_progress_json_ex, done_progress_json_ex, write_dict_to_file_ex
 
+
 class ZiniaoClient:
     """紫鸟客户端管理类"""
 
@@ -52,7 +53,7 @@ class ZiniaoClient:
     def _get_user_info(self) -> Dict[str, str]:
         """获取用户登录信息"""
         return {
-            "company" : self.config.ziniao.company,
+            "company": self.config.ziniao.company,
             "username": self.config.ziniao.username,
             "password": self.config.ziniao.password
         }
@@ -89,7 +90,7 @@ class ZiniaoClient:
     def update_core(self):
         """下载所有内核，打开店铺前调用，需客户端版本5.285.7以上"""
         data = {
-            "action"   : "updateCore",
+            "action": "updateCore",
             "requestId": str(uuid.uuid4()),
         }
         data.update(self.user_info)
@@ -145,6 +146,7 @@ class ZiniaoClient:
         print('@@ get_exit...' + json.dumps(data))
         self.send_http(data)
 
+
 class ZiniaoBrowser:
     """紫鸟浏览器操作类"""
 
@@ -153,22 +155,22 @@ class ZiniaoBrowser:
         self.config = config
 
     def open_store(self, store_info: str, isWebDriverReadOnlyMode: int = 0,
-            isprivacy: int = 0, isHeadless: int = 0,
-            cookieTypeSave: int = 0, jsInfo: str = "") -> Dict[str, Any]:
+                   isprivacy: int = 0, isHeadless: int = 0,
+                   cookieTypeSave: int = 0, jsInfo: str = "") -> Dict[str, Any]:
         """打开店铺"""
         request_id = str(uuid.uuid4())
         data = {
-            "action"                 : "startBrowser",
-            "isWaitPluginUpdate"     : 0,
-            "isHeadless"             : isHeadless,
-            "requestId"              : request_id,
+            "action": "startBrowser",
+            "isWaitPluginUpdate": 0,
+            "isHeadless": isHeadless,
+            "requestId": request_id,
             "isWebDriverReadOnlyMode": isWebDriverReadOnlyMode,
-            "cookieTypeLoad"         : 0,
-            "cookieTypeSave"         : cookieTypeSave,
-            "runMode"                : "1",
-            "isLoadUserPlugin"       : False,
-            "pluginIdType"           : 1,
-            "privacyMode"            : isprivacy
+            "cookieTypeLoad": 0,
+            "cookieTypeSave": cookieTypeSave,
+            "runMode": "1",
+            "isLoadUserPlugin": False,
+            "pluginIdType": 1,
+            "privacyMode": isprivacy
         }
         data.update(self.client.user_info)
 
@@ -194,9 +196,9 @@ class ZiniaoBrowser:
         """关闭店铺"""
         request_id = str(uuid.uuid4())
         data = {
-            "action"      : "stopBrowser",
-            "requestId"   : request_id,
-            "duplicate"   : 0,
+            "action": "stopBrowser",
+            "requestId": request_id,
+            "duplicate": 0,
             "browserOauth": browser_oauth
         }
         data.update(self.client.user_info)
@@ -215,7 +217,7 @@ class ZiniaoBrowser:
         """获取浏览器列表"""
         request_id = str(uuid.uuid4())
         data = {
-            "action"   : "getBrowserList",
+            "action": "getBrowserList",
             "requestId": request_id
         }
         data.update(self.client.user_info)
@@ -265,6 +267,7 @@ class ZiniaoBrowser:
         # 标记完成
         done_progress_json_ex(self.config, task_key, store_name)
 
+
 class ZiniaoTaskManager:
     """紫鸟任务管理类"""
 
@@ -273,10 +276,10 @@ class ZiniaoTaskManager:
         self.config = config
 
     def run_single_store_task(self, browser_info: Dict[str, Any],
-            run_func: Callable, task_key: str,
-            just_store_username: Optional[List[str]] = None,
-            is_skip_store: Optional[Callable] = None
-    ):
+                              run_func: Callable, task_key: str,
+                              just_store_username: Optional[List[str]] = None,
+                              is_skip_store: Optional[Callable] = None
+                              ):
         """运行单个店铺的任务"""
         retry_count = 0
         while True:
@@ -288,7 +291,8 @@ class ZiniaoTaskManager:
                 store_username = browser_info.get("store_username")
 
                 # 记录店铺账号与店铺别名对应关系
-                write_dict_to_file_ex(self.config.shein_store_alias, {store_username: store_name}, [store_username])
+                cache_file = f'{self.config.auto_dir}/shein_store_alias.json'
+                write_dict_to_file_ex(cache_file, {store_username: store_name}, [store_username])
 
                 if is_skip_store and is_skip_store(store_username, store_name):
                     return
@@ -341,25 +345,25 @@ class ZiniaoTaskManager:
                         self.browser.close_store(store_id)
                     break
             except:
-                send_exception(self.config, f'第{retry_count}次运行失败，准备重新打开店铺')
+                send_exception(f'第{retry_count}次运行失败，准备重新打开店铺')
                 if retry_count > 5:
                     break
 
     def run_all_stores_task(self, browser_list: List[Dict[str, Any]],
-            run_func: Callable, task_key: str,
-            just_store_username: Optional[List[str]] = None,
-            is_skip_store: Optional[Callable] = None
-    ):
+                            run_func: Callable, task_key: str,
+                            just_store_username: Optional[List[str]] = None,
+                            is_skip_store: Optional[Callable] = None
+                            ):
         """循环运行所有店铺的任务"""
         for browser_info in browser_list:
             self.run_single_store_task(browser_info, run_func, task_key, just_store_username, is_skip_store)
 
     def run_with_thread_pool(self, browser_list: List[Dict[str, Any]],
-            max_threads: int = 3, run_func: Callable = None,
-            task_key: str = None,
-            just_store_username: Optional[List[str]] = None,
-            is_skip_store: Optional[Callable] = None
-    ):
+                             max_threads: int = 3, run_func: Callable = None,
+                             task_key: str = None,
+                             just_store_username: Optional[List[str]] = None,
+                             is_skip_store: Optional[Callable] = None
+                             ):
         """使用线程池控制最大并发线程数运行任务"""
         with ThreadPoolExecutor(max_workers=max_threads) as executor:
             task = partial(self.run_single_store_task,
@@ -369,23 +373,27 @@ class ZiniaoTaskManager:
             log(f'店铺总数: {len(browser_list)}')
             executor.map(task, browser_list)
 
+
 class ZiniaoRunner:
     """紫鸟主运行器类"""
 
     def __init__(self, config):
         self.config = config
+        os.environ['auto_dir'] = self.config.auto_dir
+        os.environ['wxwork_bot_exception'] = self.config.wxwork_bot_exception
+
         self.client = ZiniaoClient(config)
         self.browser = ZiniaoBrowser(self.client, config)
         self.task_manager = ZiniaoTaskManager(self.browser, config)
 
     def execute(self, run_prepare: Optional[Callable] = None,
-            run: Optional[Callable] = None,
-            run_summary: Optional[Callable] = None,
-            run_notify: Optional[Callable] = None,
-            task_key: Optional[str] = None,
-            just_store_username: Optional[List[str]] = None,
-            is_skip_store: Optional[Callable] = None,
-    ):
+                run: Optional[Callable] = None,
+                run_summary: Optional[Callable] = None,
+                run_notify: Optional[Callable] = None,
+                task_key: Optional[str] = None,
+                just_store_username: Optional[List[str]] = None,
+                is_skip_store: Optional[Callable] = None,
+                ):
         """主执行入口"""
         # 前置执行 if run_prepare:
         run_prepare()
@@ -414,10 +422,10 @@ class ZiniaoRunner:
         try_times = 0
         while not check_progress_json_ex(self.config, task_key, just_store_username):
             try_times += 1
-            send_exception(self.config, f'检测到任务未全部完成,再次执行: {try_times}')
+            send_exception(f'检测到任务未全部完成,再次执行: {try_times}')
             self.task_manager.run_with_thread_pool(browser_list, max_threads, run, task_key, just_store_username, is_skip_store)
             if try_times >= 4:
-                send_exception(self.config, f'检测到任务未全部完成,再次执行: {try_times}')
+                send_exception(f'检测到任务未全部完成,再次执行: {try_times}')
                 break
 
         # 数据汇总
@@ -437,5 +445,6 @@ class ZiniaoRunner:
         # 关闭客户端
         self.client.exit()
 
+
 if __name__ == "__main__":
     pass