gomyck-tools 1.3.7__py3-none-any.whl → 1.3.9__py3-none-any.whl

ctools/application.py

@@ -8,7 +8,9 @@ from configparser import ConfigParser
 
 from PIL import Image
 
-from ctools import work_path, call, sign, thread_pool
+from ctools import path_info, call
+from ctools.cipher import sign
+from ctools.pools import thread_pool
 from ctools.sys_info import get_os_architecture, get_os_architecture4x
 
 """
@@ -23,9 +25,9 @@ class Server:
   metricsPort = 8011  # 获取指标信息端口
   wsPort = 8012  # websocket服务端口
   startTime = time.time()  # 系统启动时间
-  baseWorkPath = work_path.get_user_work_path('rpa')  # 基础的工作目录
+  baseWorkPath = path_info.get_user_path_info('rpa')  # 基础的工作目录
   sysLogPath = os.path.join(baseWorkPath, ".logs")  # 系统日志存储目录
-  pythonHome = os.path.join(work_path.get_Users_path(), 'Public/python-3.8.9')  # python根目录
+  pythonHome = os.path.join(path_info.get_Users_path(), 'Public/python-3.8.9')  # python根目录
   indicatorsPath = os.path.join(baseWorkPath, "indicators")  # 指标信息存储目录
   screenshotPath = os.path.join(baseWorkPath, "screenshot")  # 截图存储目录
   controlServerAddr = None
@@ -37,9 +39,9 @@ class Authorization:
 
 
 class Database:
-  url_biz = os.path.join(work_path.get_user_work_path(), "AppData/Local/SystemWin32Core", 'systemRDB')  # 数据库文件地址
-  url_func = os.path.join(work_path.get_user_work_path(), "AppData/Local/SystemWin32Core", 'explorerSearch')  # 函数库地址
-  url_auth = os.path.join(work_path.get_user_work_path(), "AppData/Local/SystemWin32Core", 'tmp')  # 授权库地址
+  url_biz = os.path.join(path_info.get_user_path_info(), "AppData/Local/SystemWin32Core", 'systemRDB')  # 数据库文件地址
+  url_func = os.path.join(path_info.get_user_path_info(), "AppData/Local/SystemWin32Core", 'explorerSearch')  # 函数库地址
+  url_auth = os.path.join(path_info.get_user_path_info(), "AppData/Local/SystemWin32Core", 'tmp')  # 授权库地址
   pool_size = 5
   max_overflow = 25
 
@@ -52,14 +54,14 @@ class Upload:
 
 
 class Schedule:
-  rpa_script_path = os.path.join(work_path.get_user_work_path(), ".cache/.tmp")  # 执行流程存储目录
+  rpa_script_path = os.path.join(path_info.get_user_path_info(), ".cache/.tmp")  # 执行流程存储目录
   template_output_path = os.path.join(Server.baseWorkPath, 'document')  # 使用模板生成的文档存储目录
   play_wright_slow_mo = 0  # play wright每一步执行延时时间， 单位毫秒
 
 
 @call.once
 def sync_config():
-  path = os.path.join(work_path.get_app_path(), "application.ini")
+  path = os.path.join(path_info.get_app_path(), "application.ini")
   conf = ConfigParser()
   if os.path.exists(path):
     conf.read(path)
@@ -69,7 +71,7 @@ def sync_config():
       server_conf = conf['Server']
       Server.version = server_conf['version']
       Server.port = int(server_conf['port'])
-      Server.baseWorkPath = work_path.get_user_work_path(server_conf['baseWorkPath'])
+      Server.baseWorkPath = path_info.get_user_path_info(server_conf['baseWorkPath'])
       Server.sysLogPath = os.path.join(Server.baseWorkPath, ".logs")
     ######### Server 必须放在第一个初始化的位置 ########
     if "Database" in sections:
@@ -106,18 +108,18 @@ def sync_config():
 
 
 def check_database():
-  db_file = os.path.join(work_path.get_user_work_path(), "AppData/Local/SystemWin32Core", 'systemRDB')
+  db_file = os.path.join(path_info.get_user_path_info(), "AppData/Local/SystemWin32Core", 'systemRDB')
   if os.path.exists(db_file):
     db_size = os.path.getsize(db_file)
     if db_size < 2000 * 1024: os.remove(db_file)
-  db_file = os.path.join(work_path.get_user_work_path(), "AppData/Local/SystemWin32Core", 'tmp')
+  db_file = os.path.join(path_info.get_user_path_info(), "AppData/Local/SystemWin32Core", 'tmp')
   if os.path.exists(db_file):
     db_size = os.path.getsize(db_file)
     if db_size < 2000 * 1024: os.remove(db_file)
 
 def validate_sign():
-  sign_app_val = sign.generate_signature(work_path.get_install_path('rpa-server.exe'))
-  with open(work_path.get_install_path('rpa-server.sign')) as sign_file:
+  sign_app_val = sign.generate_signature(path_info.get_install_path('rpa-server.exe'))
+  with open(path_info.get_install_path('rpa-server.sign')) as sign_file:
     sign_file_val = sign_file.readline()
   if 'rpa-dev-sign' == sign_file_val.strip(): return
   if sign_app_val != sign_file_val: raise Exception('程序签名验证失败!!!')
@@ -126,9 +128,9 @@ def validate_sign():
 def path_config():
   print('start config path')
   driverPath = Upload.driver_path
-  pythonPath = os.path.join(work_path.get_Users_path(), 'Public/python-3.8.9')
-  hplPath = os.path.join(work_path.get_user_work_path(), 'AppData/Local/ms-playwright')
-  taguiPath = os.path.join(work_path.get_user_work_path(), 'AppData/Roaming/tagui')
+  pythonPath = os.path.join(path_info.get_Users_path(), 'Public/python-3.8.9')
+  hplPath = os.path.join(path_info.get_user_path_info(), 'AppData/Local/ms-playwright')
+  taguiPath = os.path.join(path_info.get_user_path_info(), 'AppData/Roaming/tagui')
   os.environ['PATH'] = os.pathsep.join([pythonPath, os.path.join(pythonPath, 'Scripts'), driverPath, os.path.join(hplPath, 'chromium-920619\chrome-win'), taguiPath, os.environ['PATH']])
   print('end config path')
 
@@ -136,9 +138,9 @@ def path_config():
 def sync_version(callFunc):
   destFilePath = os.path.join(Server.baseWorkPath, "version")
   driverPath = Upload.driver_path
-  pythonPath = os.path.join(work_path.get_Users_path(), 'Public/python-3.8.9')
-  msPlayPath = os.path.join(work_path.get_user_work_path(), 'AppData/Local/ms-playwright')
-  taguiPath = os.path.join(work_path.get_user_work_path(), 'AppData/Roaming/tagui')
+  pythonPath = os.path.join(path_info.get_Users_path(), 'Public/python-3.8.9')
+  msPlayPath = os.path.join(path_info.get_user_path_info(), 'AppData/Local/ms-playwright')
+  taguiPath = os.path.join(path_info.get_user_path_info(), 'AppData/Roaming/tagui')
   if not os.path.exists(destFilePath):
     try:
       shutil.rmtree(pythonPath)
@@ -156,7 +158,7 @@ def sync_version(callFunc):
       shutil.rmtree(taguiPath)
     except Exception:
       pass
-    from ctools.pacth import Patch
+    from ctools.auto.pacth import Patch
     patch = Patch(oldVersion='V1.0.0', newVersion=Server.version, pythonPath=pythonPath, playwrightPath=msPlayPath, driverPath=driverPath)
     patch.apply_patch()
     if callFunc: callFunc()
@@ -171,7 +173,7 @@ def sync_version(callFunc):
       oldVersion.close()
     if oldV == Server.version and '-snapshot' not in oldV: return
     print('开始升级本地程序..')
-    from ctools.pacth import Patch
+    from ctools.auto.pacth import Patch
     patch = Patch(oldVersion=oldV, newVersion=Server.version, pythonPath=pythonPath, playwrightPath=msPlayPath, driverPath=driverPath)
     patch.apply_patch()
     if callFunc: callFunc()
@@ -183,8 +185,8 @@ def sync_version(callFunc):
 
 def sync_database():
   for db in [['simpleYWI', 'systemRDB', False], ['simpleHSI', 'explorerSearch', True], ['simpleSQI', 'tmp', False]]:
-    dbPath = os.path.join(work_path.get_app_path(), "extension", db[0])
-    destDBPath = os.path.join(work_path.get_user_work_path(), "AppData/Local/SystemWin32Core")
+    dbPath = os.path.join(path_info.get_app_path(), "extension", db[0])
+    destDBPath = os.path.join(path_info.get_user_path_info(), "AppData/Local/SystemWin32Core")
     os.makedirs(destDBPath, exist_ok=True)
     destFilePath = os.path.join(destDBPath, db[1])
     try:
@@ -199,9 +201,9 @@ def sync_database():
 
 def sync_python_env():
   print('SDK 开始安装...')
-  sourcePath = os.path.join(work_path.get_install_path(), "sources", 'python-3.8.9-{}.zip'.format(get_os_architecture()))
-  destPLPath = os.path.join(work_path.get_Users_path(), 'Public/python-3.8.9')
-  unzipPLPath = os.path.join(work_path.get_Users_path(), 'Public')
+  sourcePath = os.path.join(path_info.get_install_path(), "sources", 'python-3.8.9-{}.zip'.format(get_os_architecture()))
+  destPLPath = os.path.join(path_info.get_Users_path(), 'Public/python-3.8.9')
+  unzipPLPath = os.path.join(path_info.get_Users_path(), 'Public')
   if not os.path.exists(destPLPath):
     with zipfile.ZipFile(sourcePath, 'r') as zip_ref:
       zip_ref.extractall(unzipPLPath)
@@ -220,7 +222,7 @@ def sync_python_module():
       m_path = os.path.join(Upload.module_path, file_name)
       module_list.append(m_path)
 
-  upgrade_module_path = os.path.join(work_path.get_install_path(), "sources/UpgradeModule")
+  upgrade_module_path = os.path.join(path_info.get_install_path(), "sources/UpgradeModule")
   init_flag = os.path.join(upgrade_module_path, "init")
   upgrade_module_path = os.path.join(upgrade_module_path, get_os_architecture4x())
   if not os.path.exists(init_flag) and os.path.exists(upgrade_module_path):
@@ -267,12 +269,12 @@ def sync_python_module():
 
 
 def sync_python_interpreter():
-  destPLPath = os.path.join(work_path.get_Users_path(), 'Public/python-3.8.9')
-  enhancePath = os.path.join(work_path.get_Users_path(), "Public/enhance", 'enhance-{}.zip'.format(get_os_architecture()))
-  originPath = os.path.join(work_path.get_Users_path(), "Public/enhance", 'origin-{}.zip'.format(get_os_architecture()))
-  initFlag = os.path.join(work_path.get_Users_path(), "Public/enhance", 'init')
+  destPLPath = os.path.join(path_info.get_Users_path(), 'Public/python-3.8.9')
+  enhancePath = os.path.join(path_info.get_Users_path(), "Public/enhance", 'enhance-{}.zip'.format(get_os_architecture()))
+  originPath = os.path.join(path_info.get_Users_path(), "Public/enhance", 'origin-{}.zip'.format(get_os_architecture()))
+  initFlag = os.path.join(path_info.get_Users_path(), "Public/enhance", 'init')
   flagValue = 'can not find dest dir'
-  os.makedirs(os.path.join(work_path.get_Users_path(), "Public/enhance"), exist_ok=True)
+  os.makedirs(os.path.join(path_info.get_Users_path(), "Public/enhance"), exist_ok=True)
   if not os.path.exists(initFlag):
     try:
       if os.path.exists(enhancePath):
@@ -300,7 +302,7 @@ def sync_python_interpreter():
 
 def sync_driver():
   print('DRIVER 开始安装...')
-  sourcePath = os.path.join(work_path.get_install_path(), "sources", 'uploadDriver.zip')
+  sourcePath = os.path.join(path_info.get_install_path(), "sources", 'uploadDriver.zip')
   destPLPath = Upload.driver_path
   unzipPLPath = Upload.driver_path
   if not os.path.exists(os.path.join(destPLPath)):
@@ -315,9 +317,9 @@ def sync_driver():
 
 def sync_tagui():
   print('HTG ENGINE 开始安装...')
-  plPath = os.path.join(work_path.get_install_path(), "sources", 'htg-source.zip')
-  destPLPath = os.path.join(work_path.get_user_work_path(), 'AppData/Roaming/tagui')
-  unzipPLPath = os.path.join(work_path.get_user_work_path(), 'AppData/Roaming/')
+  plPath = os.path.join(path_info.get_install_path(), "sources", 'htg-source.zip')
+  destPLPath = os.path.join(path_info.get_user_path_info(), 'AppData/Roaming/tagui')
+  unzipPLPath = os.path.join(path_info.get_user_path_info(), 'AppData/Roaming/')
   if not os.path.exists(os.path.join(destPLPath)):
     with zipfile.ZipFile(plPath, 'r') as zip_ref:
       zip_ref.extractall(unzipPLPath)
@@ -328,9 +330,9 @@ def sync_tagui():
 
 def sync_playwright():
   print('HPL ENGINE 开始安装...')
-  plPath = os.path.join(work_path.get_install_path(), "sources", 'ms-playwright.zip')
-  destPLPath = os.path.join(work_path.get_user_work_path(), 'AppData/Local/ms-playwright')
-  unzipPLPath = os.path.join(work_path.get_user_work_path(), 'AppData/Local/')
+  plPath = os.path.join(path_info.get_install_path(), "sources", 'ms-playwright.zip')
+  destPLPath = os.path.join(path_info.get_user_path_info(), 'AppData/Local/ms-playwright')
+  unzipPLPath = os.path.join(path_info.get_user_path_info(), 'AppData/Local/')
   if not os.path.exists(os.path.join(destPLPath)):
     with zipfile.ZipFile(plPath, 'r') as zip_ref:
       zip_ref.extractall(unzipPLPath)
@@ -339,8 +341,8 @@ def sync_playwright():
     print('HPL ENGINE 依赖安装成功(1)')
 
 def sync_paddleocr_model():
-  model_path = os.path.join(work_path.get_install_path(), "sources", 'paddleocr-model.zip')
-  paddleocr_path = work_path.get_user_work_path(".paddleocr")
+  model_path = os.path.join(path_info.get_install_path(), "sources", 'paddleocr-model.zip')
+  paddleocr_path = path_info.get_user_path_info(".paddleocr")
   if not os.path.exists(paddleocr_path) and os.path.exists(model_path) and get_os_architecture() == "64":
     print('安装paddleocr模型文件开始')
     with zipfile.ZipFile(model_path, 'r') as zip_ref:
@@ -349,13 +351,13 @@ def sync_paddleocr_model():
 
 
 def clean_temp_dir(target_file='rpa_ident'):
-  root_folder = work_path.get_user_temp_path()
+  root_folder = path_info.get_user_temp_path()
   print("Start clear cache...")
   for folder_name in os.listdir(root_folder):
     folder_path = os.path.join(root_folder, folder_name)
     if os.path.isdir(folder_path):
       file_path = os.path.join(folder_path, target_file)
-      if os.path.exists(file_path) and os.path.normpath(folder_path) != os.path.normpath(work_path.get_app_path()):
+      if os.path.exists(file_path) and os.path.normpath(folder_path) != os.path.normpath(path_info.get_app_path()):
         try:
           shutil.rmtree(folder_path)
         except Exception:
@@ -376,7 +378,7 @@ def system_tray(app_server):
       thread_pool.submit(app_server.stop)
       app_icon.stop()
 
-    image = Image.open(os.path.join(work_path.get_app_path(), "images/ico.png"))
+    image = Image.open(os.path.join(path_info.get_app_path(), "images/ico.png"))
     menu = (
       pystray.MenuItem(text='退出', action=on_quit_clicked),
     )

ctools/auto/__init__.py

@@ -0,0 +1,4 @@
+#!/usr/bin/env python
+# -*- coding: UTF-8 -*-
+__author__ = 'haoyang'
+__date__ = '2025/6/11 09:27'

ctools/{browser_element_tools.py → auto/browser_element.py}

@@ -12,7 +12,8 @@ from selenium.webdriver.chrome.service import Service
 from selenium.webdriver.common.by import By
 from urllib3.exceptions import MaxRetryError
 
-from ctools import application, string_tools
+from ctools import application
+from ctools.util import cid
 
 keyboard_listener = None
 ctrl_pressed = None
@@ -116,7 +117,7 @@ def on_press(key):
     if g_result:
       try:
         select_element = g_driver.find_element(By.XPATH, g_result[0])
-        picture_path = "%s.png" % os.path.join(application.Server.screenshotPath, "element-"+string_tools.get_uuid())
+        picture_path = "%s.png" % os.path.join(application.Server.screenshotPath, "element-"+cid.get_uuid())
         select_element.screenshot(picture_path)
       except Exception:
         pass

ctools/{plan_area_tools.py → auto/plan_area.py}

@@ -95,3 +95,6 @@ def get_area():
   pat = PlanAreaTools()
   pat.start()
   return tuple(pat.area)
+
+# if __name__ == '__main__':
+#     get_area()

ctools/{resource_bundle_tools.py → auto/resource_bundle.py}

@@ -11,7 +11,7 @@ from Crypto.Hash import MD5
 from Crypto.PublicKey import RSA
 from Crypto.Signature import pkcs1_15
 
-from ctools import sys_log, work_path, application
+from ctools import sys_log, application, path_info
 
 log = sys_log.flog
 
@@ -39,7 +39,7 @@ def signature(manifest: dict):
   :param manifest: 元数据字典
   :return:
   """
-  with open(work_path.get_app_path() + '/keys/resource_private_key.pem', 'r') as pri:
+  with open(path_info.get_app_path() + '/keys/resource_private_key.pem', 'r') as pri:
     private_key = RSA.import_key(pri.read())
     digest = MD5.new(data_layout(manifest).encode('utf-8'))
     return b64encode(pkcs1_15.new(private_key).sign(digest)).decode()
@@ -52,7 +52,7 @@ def sign_verify(manifest: dict, sign_val: str):
   :param sign_val: 对比的签名值
   :return:
   """
-  with open(work_path.get_app_path() + '/keys/resource_public_key.pem', 'r') as pub:
+  with open(path_info.get_app_path() + '/keys/resource_public_key.pem', 'r') as pub:
     public_key = RSA.import_key(pub.read())
     digest = MD5.new(data_layout(manifest).encode('utf-8'))
     try:

ctools/{screenshot_tools.py → auto/screenshot.py}

@@ -4,7 +4,8 @@ import tkinter as tk
 
 import pyautogui
 
-from ctools import application, string_tools
+from ctools import application
+from ctools import cid
 
 """
 截屏工具类, 按回车截图
@@ -108,7 +109,7 @@ class ScreenshotTools:
       time.sleep(0.3)
       img = pyautogui.screenshot(region=[self.xstart, self.ystart, self.xend - self.xstart, self.yend - self.ystart])
       self.screenshot_path = os.path.join(application.Server.screenshotPath,
-                                          "screenshot-%s.png" % string_tools.get_snowflake_id())
+                                          "screenshot-%s.png" % cid.get_snowflake_id())
       img.save(self.screenshot_path)
     except Exception:
       pass

ctools/{win_control.py → auto/win_control.py}

@@ -5,7 +5,10 @@ import pyautogui
 import uiautomation as auto
 from pynput import keyboard
 
-from ctools import thread_pool, win_canvas, application, string_tools
+from ctools import application
+from ctools.auto import win_canvas
+from ctools.pools import thread_pool
+from ctools import cid
 
 current_control = None
 ctrl_pressed = False
@@ -55,7 +58,7 @@ def get_control_from_cursor():
 
       img = pyautogui.screenshot(region=[control.BoundingRectangle.left, control.BoundingRectangle.top,
                                          control.BoundingRectangle.width(), control.BoundingRectangle.height()])
-      screenshot_path = os.path.join(application.Server.screenshotPath, "screenshot-%s.png" % string_tools.get_snowflake_id())
+      screenshot_path = os.path.join(application.Server.screenshotPath, "screenshot-%s.png" % cid.get_snowflake_id())
       img.save(screenshot_path)
       # xx = auto.Control(**control_json)
       # print(control_json)

ctools/cid.py

@@ -0,0 +1,18 @@
+from ctools.util.snow_id import SnowId
+
+idWorker = SnowId(1, 2, 0)
+
+def get_snowflake_id():
+  return idWorker.get_id()
+
+def get_random_str(size: int = 10) -> str:
+  import random
+  return "".join(random.sample('abcdefghjklmnpqrstuvwxyz123456789', size))
+
+def get_uuid() -> str:
+  import uuid
+  return str(uuid.uuid1()).replace("-", "")
+
+
+
+

ctools/cipher/__init__.py

@@ -0,0 +1,4 @@
+#!/usr/bin/env python
+# -*- coding: UTF-8 -*-
+__author__ = 'haoyang'
+__date__ = '2025/6/11 09:20'

ctools/{rsa.py → cipher/rsa.py}

@@ -5,7 +5,7 @@ from Crypto.Hash import SHA256
 from Crypto.PublicKey import RSA
 from Crypto.Signature import pkcs1_15
 
-from ctools import work_path, cjson
+from ctools import cjson, path_info
 
 ENCRYPT_CHUNK_SIZE = 245
 decrypt_CHUNK_SIZE = 512
@@ -20,7 +20,7 @@ def generate_rsa_keypair(bits=2048):
     f.write(public_key)
 
 def loadLicenseInfo(auth_code):
-  with open(work_path.get_app_path() + '/keys/license.key', 'r') as pri:
+  with open(path_info.get_app_path() + '/keys/license.key', 'r') as pri:
     decrypt_message = decrypt(auth_code.strip(), pri.read())
     return cjson.loads(decrypt_message)
 

ctools/database/__init__.py

@@ -0,0 +1,4 @@
+#!/usr/bin/env python
+# -*- coding: UTF-8 -*-
+__author__ = 'haoyang'
+__date__ = '2025/6/11 09:22'

ctools/{database.py → database/database.py}

@@ -7,8 +7,9 @@ from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import sessionmaker, Session
 from sqlalchemy.sql import text
 
-from ctools import call, string_tools
-from ctools.thread_pool import thread_local
+from ctools import call
+from ctools.pools.thread_pool import thread_local
+from ctools import cid
 
 """
 class XXXX(BaseMixin):
@@ -82,7 +83,7 @@ def _create_connection(db_url: str, pool_size: int=5, max_overflow: int=25, conn
   return engine, sm
 
 def generate_custom_id():
-  return str(string_tools.get_snowflake_id())
+  return str(cid.get_snowflake_id())
 
 class BaseMixin(Base):
   __abstract__ = True

ctools/geo/__init__.py

@@ -0,0 +1,4 @@
+#!/usr/bin/env python
+# -*- coding: UTF-8 -*-
+__author__ = 'haoyang'
+__date__ = '2025/6/11 09:20'

ctools/metrics.py

@@ -4,7 +4,7 @@ from enum import Enum
 
 from prometheus_client import Counter, Gauge, Summary, Histogram, start_http_server
 
-from ctools import call, cjson, sys_log, work_path
+from ctools import call, cjson, sys_log, path_info
 
 log = sys_log.flog
 
@@ -12,7 +12,7 @@ _metrics_port = 8889
 _persistent_json = {}
 _metrics_initial = {}
 _lock = threading.Lock()
-_metrics_persistent_path = os.path.join(work_path.get_user_work_path('metrics', True), 'persistent.json')
+_metrics_persistent_path = os.path.join(path_info.get_user_work_path('metrics', True), 'persistent.json')
 
 # 认证中间件
 # @app.before_request

ctools/office/__init__.py

@@ -0,0 +1,4 @@
+#!/usr/bin/env python
+# -*- coding: UTF-8 -*-
+__author__ = 'haoyang'
+__date__ = '2025/6/11 09:37'

ctools/{word_fill.py → office/word_fill.py}

@@ -16,7 +16,7 @@ from xlsxtpl import writerx as xlsx_writerx
 from xlutils.copy import copy
 
 from ctools import cjson
-from ctools.word_fill_entity import WordTable, WordCell, Style, Font, Alignment
+from ctools.office.word_fill_entity import WordTable, WordCell, Style, Font, Alignment
 
 EXCEL_WORD_ALIGNMENT = {"left": WD_TABLE_ALIGNMENT.LEFT,
                         "center": WD_TABLE_ALIGNMENT.CENTER,

ctools/pools/__init__.py

@@ -0,0 +1,4 @@
+#!/usr/bin/env python
+# -*- coding: UTF-8 -*-
+__author__ = 'haoyang'
+__date__ = '2025/6/11 09:18'

ctools/similar.py

@@ -0,0 +1,22 @@
+from fuzzywuzzy import fuzz, process
+
+def best_match(query: str, choices: list[str], score_cutoff: int = 70):
+  """
+  获取最接近 query 的匹配项
+  :return: (匹配项, 相似度得分) 或 None
+  """
+  return process.extractOne(query, choices, scorer=fuzz.ratio, score_cutoff=score_cutoff)
+
+def all_matches(query: str, choices: list[str], limit: int = 5):
+  """
+  获取多个相似匹配项
+  :return: [(匹配项, 相似度得分), ...]
+  """
+  return process.extract(query, choices, scorer=fuzz.ratio, limit=limit)
+
+def is_similar(s1: str, s2: str, threshold: int = 85):
+  """
+  判断两个字符串是否相似
+  :return: True / False
+  """
+  return fuzz.ratio(s1, s2) >= threshold

ctools/stream/__init__.py

@@ -0,0 +1,4 @@
+#!/usr/bin/env python
+# -*- coding: UTF-8 -*-
+__author__ = 'haoyang'
+__date__ = '2025/6/11 09:21'

ctools/{credis.py → stream/credis.py}

@@ -6,7 +6,8 @@ __date__ = '2025/2/14 11:09'
 import redis
 from redis import Redis
 
-from ctools import date_utils, thread_pool, string_tools
+from ctools.pools import thread_pool
+from ctools import cdate, cid
 
 
 def init_pool(host: str = 'localhost', port: int = 6379, db: int = 0, password: str = None,
@@ -35,11 +36,11 @@ def init_pool(host: str = 'localhost', port: int = 6379, db: int = 0, password:
 def add_lock(r: Redis, key: str, timeout: int = 30):
   if r.exists(key):
     expire_time = r.get(key)
-    if date_utils.time_diff_in_seconds(expire_time, date_utils.get_date_time()) > 0:
+    if cdate.time_diff_in_seconds(expire_time, cdate.get_date_time()) > 0:
       return True
     else:
       r.delete(key)
-  return r.set(key, date_utils.opt_time(seconds=timeout), nx=True, ex=timeout) is not None
+  return r.set(key, cdate.opt_time(seconds=timeout), nx=True, ex=timeout) is not None
 
 def remove_lock(r: Redis, key: str):
   r.delete(key)
@@ -94,7 +95,7 @@ def stream_subscribe(r: Redis, stream_name, group_name, callback, from_id: str='
         print(f"Consumer group '{group_name}' already exists.")
       else:
         print(f"Error creating consumer group '{group_name}': {e}")
-    consumer_name = 'consumer-{}'.format(string_tools.get_uuid())
+    consumer_name = 'consumer-{}'.format(cid.get_uuid())
     # 处理未确认的消息
     _process_pending_messages(r, stream_name, group_name, consumer_name, callback)
     while True:

ctools/{mqtt_utils.py → stream/mqtt_utils.py}

@@ -5,8 +5,11 @@ from typing import Dict
 from paho.mqtt import client as mqtt
 from paho.mqtt.enums import CallbackAPIVersion
 
-from ctools import sys_log, cjson, string_tools, sys_info, date_utils, sm_tools, thread_pool
+from ctools import sys_log, cjson, sys_info
+from ctools.cipher import sm_util
 from ctools.dict_wrapper import DictWrapper as DictToObj
+from ctools.pools import thread_pool
+from ctools import cid, cdate
 
 
 class MQTTEvent(Enum):
@@ -42,9 +45,9 @@ class MQTTRequest:
 
   def __init__(self, **kwargs):
     self.event: str = kwargs.get('event')
-    self.trace_id: str = kwargs.get('trace_id') if kwargs.get('trace_id') else string_tools.get_uuid()
+    self.trace_id: str = kwargs.get('trace_id') if kwargs.get('trace_id') else cid.get_uuid()
     self.client_id: str = kwargs.get('client_id')
-    self.time: str = kwargs.get('time') if kwargs.get('time') else date_utils.get_date_time()
+    self.time: str = kwargs.get('time') if kwargs.get('time') else cdate.get_date_time()
     self.body = kwargs.get('body')
 
   def __str__(self):
@@ -60,7 +63,7 @@ class MQTTResponse:
     self.event: str = kwargs.get('event')
     self.trace_id: str = kwargs.get('trace_id')
     self.status: int = kwargs.get('status')
-    self.time: str = kwargs.get('time') if kwargs.get('time') else date_utils.get_date_time()
+    self.time: str = kwargs.get('time') if kwargs.get('time') else cdate.get_date_time()
     self.msg: str = kwargs.get('msg')
     self.body = kwargs.get('body')
 
@@ -94,8 +97,8 @@ class MQTTUtils:
       will_msg = {
         "event": "offline",
         "client_id": self.client_id,
-        "trace_id": string_tools.get_uuid(),
-        "time": date_utils.get_date_time(),
+        "trace_id": cid.get_uuid(),
+        "time": cdate.get_date_time(),
         "body": {"type": "will_msg"}
       }
       self.client.will_set(self.publish_topic, cjson.dumps(will_msg), 2, False)
@@ -150,16 +153,16 @@ class MQTTUtils:
   def encrypt_body(self, body):
     if self.broker_encrypt_key:
       try:
-        return sm_tools.encrypt_with_sm4(self.broker_encrypt_key, cjson.dumps(body))
+        return sm_util.encrypt_with_sm4(self.broker_encrypt_key, cjson.dumps(body))
       except Exception:
-        return sm_tools.encrypt_with_sm4(self.broker_encrypt_key, body)
+        return sm_util.encrypt_with_sm4(self.broker_encrypt_key, body)
 
   def decrypt_body(self, body):
     if self.broker_encrypt_key:
       try:
-        return cjson.loads(sm_tools.decrypt_with_sm4(self.broker_encrypt_key, body))
+        return cjson.loads(sm_util.decrypt_with_sm4(self.broker_encrypt_key, body))
       except Exception:
-        sm_tools.decrypt_with_sm4(self.broker_encrypt_key, body)
+        sm_util.decrypt_with_sm4(self.broker_encrypt_key, body)
 
   def connect(self, username=None, password=None, timeout=10):
     if username and password:

ctools/sys_info.py

@@ -2,7 +2,8 @@ import hashlib
 import os
 import platform
 
-from ctools import cjson, sm_tools, work_path
+from ctools import cjson, path_info
+from ctools.cipher import sm_util
 
 MACHINE_KEY = b'EnrGffoorbFyTYoS0902YyT1Fhehj4InpbezIDUuPOg='
 
@@ -15,7 +16,7 @@ def get_user():
 
 def get_machine_code():
   if MachineInfo.machine_code: return MachineInfo.machine_code
-  destPath = os.path.join(work_path.get_user_work_path(), "AppData/Local/machine")
+  destPath = os.path.join(path_info.get_user_work_path(), "AppData/Local/machine")
   machine_file = os.path.join(destPath, 'machine_code.mc')
   origin_machine_code = get_origin_machine_code()
   if os.path.exists(machine_file):
@@ -23,7 +24,7 @@ def get_machine_code():
       file_code = f.readline()
       dec_code = ''
       try:
-        dec_code = cjson.loads(sm_tools.decrypt_with_sm4(MACHINE_KEY, file_code))
+        dec_code = cjson.loads(sm_util.decrypt_with_sm4(MACHINE_KEY, file_code))
         origin_code = dec_code.get("origin_code")
         if origin_code == origin_machine_code:
           MachineInfo.machine_code = dec_code.get("hash_code")
@@ -33,7 +34,7 @@ def get_machine_code():
         print('machine code file is error: {} {}'.format(file_code, dec_code))
   hash_machine_code = get_hash_machine_code(origin_machine_code)
   machine_code = {"origin_code": origin_machine_code, "hash_code": hash_machine_code}
-  enc_code = sm_tools.encrypt_with_sm4(MACHINE_KEY, cjson.dumps(machine_code))
+  enc_code = sm_util.encrypt_with_sm4(MACHINE_KEY, cjson.dumps(machine_code))
   os.makedirs(destPath, exist_ok=True)
   with open(machine_file, 'w') as f:
     f.write(enc_code)