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

qrpa/RateLimitedSender.py

@@ -6,15 +6,13 @@ from datetime import datetime
 import traceback
 
 class RateLimitedSender:
-    def __init__(self, sender_func, interval=60, max_batch_size=10):
+    def __init__(self, sender_func, interval=60):
         """
         :param sender_func: 实际的发送函数，参数是字符串消息
-        :param interval: 发送间隔（秒）
-        :param max_batch_size: 队列消息条数超过此值时立即发送
+        :param interval: 最短发送间隔（秒）
         """
         self.sender_func = sender_func
         self.interval = interval
-        self.max_batch_size = max_batch_size
         self.queue = deque()
         self.lock = threading.Lock()
         self.last_send_time = 0
@@ -26,9 +24,6 @@ class RateLimitedSender:
         """添加消息到队列（非阻塞）"""
         with self.lock:
             self.queue.append(message)
-            # 如果超过批量上限，立即发送
-            if len(self.queue) >= self.max_batch_size:
-                self._flush()
 
     def _flush(self):
         """立即发送队列消息（内部调用）"""
@@ -47,4 +42,4 @@ class RateLimitedSender:
             with self.lock:
                 if self.queue and (time.time() - self.last_send_time >= self.interval):
                     self._flush()
-            time.sleep(1)
+            time.sleep(1)

qrpa/__init__.py

@@ -10,4 +10,5 @@ from .time_utils import TimeUtils
 from .fun_file import read_dict_from_file, read_dict_from_file_ex, write_dict_to_file, write_dict_to_file_ex
 from .fun_file import get_progress_json_ex, check_progress_json_ex, done_progress_json_ex
 
-from .fun_web import fetch, fetch_via_iframe, find_all_iframe, full_screen_shot
+from .fun_web import fetch, fetch_via_iframe, find_all_iframe, full_screen_shot
+from .fun_win import *

qrpa/fun_base.py

@@ -49,8 +49,7 @@ def send_exception(msg=None):
 
         send_exception._wx_sender = RateLimitedSender(
             sender_func=wxwork_bot_send,
-            interval=30,  # 60 秒发一次
-            max_batch_size=5  # 累积 5 条立即发
+            interval=30,  # 10 秒发一次
         )
 
     # 构造异常消息

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-1.0.10.dist-info → qrpa-1.0.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: qrpa
-Version: 1.0.10
+Version: 1.0.11
 Summary: qsir's rpa library
 Author: QSir
 Author-email: QSir <1171725650@qq.com>

{qrpa-1.0.10.dist-info → qrpa-1.0.11.dist-info}/RECORD

@@ -1,16 +1,16 @@
-qrpa/RateLimitedSender.py,sha256=7-wDoVJbv-y3Z3dX85conBYNdUfpidZ7QgssdAKNlFU,1698
-qrpa/__init__.py,sha256=HW_Aqi4nHtb-tnPlPRq4vx5M8yr-hKGxxItGRS1Zn5A,608
+qrpa/RateLimitedSender.py,sha256=hqvb1qspDFaW4RsLuVufylOrefkMgixANKeBaGEqYb4,1421
+qrpa/__init__.py,sha256=OSN1T20XRX3uadTfmYrTbqETA6Y_tOca2bDodllnSkc,632
 qrpa/db_migrator.py,sha256=2VmhzcMsU0MKpl-mNCwKyV8tLTqyEysSpP27-S_rQZ8,21862
-qrpa/fun_base.py,sha256=j5EcC-1l6n9ZEb-xGj1sTGgQ-1zCaoSHl919iDku3Wo,3363
+qrpa/fun_base.py,sha256=W_owEa8-yuGG18n9kX3remm9YTzym69ztQjtYCNMTw4,3308
 qrpa/fun_excel.py,sha256=aVSLcE3UJYnxvGi1XULM9qTbziUT7xlOcd2grnP6F3I,103615
 qrpa/fun_file.py,sha256=BInN-Iuxi8sYiJ031gXI_DzO_n170RsOf7fnpkl_etM,7100
 qrpa/fun_web.py,sha256=5QLQorAhRzMOGMRh4eCJ2UH8ZhVHvxkHwobWhmgU5qM,6286
-qrpa/fun_win.py,sha256=mPvojBYYvAcoXIGh_IXaNcA6x1qZAPGTRzwRvDb-WSU,1223
+qrpa/fun_win.py,sha256=-LnTeocdTt72NVH6VgLdpAT9_C5oV9okeudXG6CftMA,8034
 qrpa/shein_ziniao.py,sha256=nSqqcEPh4nVQtUxUnIRzeZfTLyXywGPjPZn5pP-w57U,18309
 qrpa/time_utils.py,sha256=ef0hhbN_6b-gcnz5ETIVOoxemIMvcxGVGGIRnHnGaBo,29564
 qrpa/time_utils_example.py,sha256=shHOXKKF3QSzb0SHsNc34M61wEkkLuM30U9X1THKNS8,8053
 qrpa/wxwork.py,sha256=IeJq-1LwKCsCt_AdMSLvYrQJWNoQp_wqVtk6iqwJqf4,9200
-qrpa-1.0.10.dist-info/METADATA,sha256=wRmrV9leWSw8ap06Z7aIQ4ynlUte1xVm2m8B6XOhs_Q,231
-qrpa-1.0.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-qrpa-1.0.10.dist-info/top_level.txt,sha256=F6T5igi0fhXDucPPUbmmSC0qFCDEsH5eVijfVF48OFU,5
-qrpa-1.0.10.dist-info/RECORD,,
+qrpa-1.0.11.dist-info/METADATA,sha256=TYJZ4HTibjVUL7-V67k_waJYRlohGYGVSbCWRVU3n-U,231
+qrpa-1.0.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+qrpa-1.0.11.dist-info/top_level.txt,sha256=F6T5igi0fhXDucPPUbmmSC0qFCDEsH5eVijfVF48OFU,5
+qrpa-1.0.11.dist-info/RECORD,,