gomyck-tools 1.4.1__py3-none-any.whl → 1.4.3__py3-none-any.whl

ctools/aspect.py

@@ -0,0 +1,65 @@
+#!/usr/bin/env python
+# -*- coding: UTF-8 -*-
+__author__ = 'haoyang'
+__date__ = '2025/7/23 08:33'
+
+import functools
+import sys
+
+from ctools.pools import thread_pool
+
+def _ensure_list(funcs):
+  if callable(funcs):
+    return [funcs]
+  if isinstance(funcs, (list, tuple, set)):
+    return list(funcs)
+  raise TypeError("必须是可调用对象或可迭代对象")
+
+def before(before_funcs):
+  """
+  用于将无参函数注入目标函数的调用前
+  支持多个函数
+  """
+  before_funcs = _ensure_list(before_funcs)
+  def decorator(func, sync=True):
+    @functools.wraps(func)
+    def wrapper(*args, **kwargs):
+      for bf in before_funcs:
+        if callable(bf):
+          if sync:
+            bf()
+          else:
+            thread_pool.submit(bf)
+      return func(*args, **kwargs)
+    _replace_func_binding(func, wrapper)
+    return wrapper
+  return decorator
+
+def after(after_funcs, sync=True):
+  """
+  用于将无参函数注入目标函数的调用后
+  支持多个函数
+  """
+  after_funcs = _ensure_list(after_funcs)
+  def decorator(func):
+    @functools.wraps(func)
+    def wrapper(*args, **kwargs):
+      result = func(*args, **kwargs)
+      for af in after_funcs:
+        if callable(af):
+          if sync:
+            af()
+          else:
+            thread_pool.submit(af)
+      return result
+    _replace_func_binding(func, wrapper)
+    return wrapper
+  return decorator
+
+def _replace_func_binding(old_func, new_func):
+  """
+  替换函数在其模块中的绑定，确保所有使用点都生效
+  """
+  mod = sys.modules.get(old_func.__module__)
+  if mod and hasattr(mod, old_func.__name__):
+    setattr(mod, old_func.__name__, new_func)

ctools/auto/browser_element.py

@@ -23,6 +23,7 @@ g_result = []
 g_quite_flag = False
 picture_path = ""
 
+
 def get_hovered_element_html(url, explore, callback):
   global g_driver, g_callback, g_result
   g_callback = callback
@@ -73,10 +74,10 @@ def get_hovered_element_html(url, explore, callback):
         match_dom = page_dom_tree.xpath('//*[@ck-flag="ck"]')
         for ck_element in match_dom:
           ck_xpath = page_dom_tree.getpath(ck_element)
-          #print('XPATH IS {}'.format(ck_xpath))
+          # print('XPATH IS {}'.format(ck_xpath))
           try:
             ele = driver.find_element(By.XPATH, ck_xpath)
-            #print('XPATH_HTML: {}'.format(ele.get_attribute('outerHTML')))
+            # print('XPATH_HTML: {}'.format(ele.get_attribute('outerHTML')))
           except Exception:
             pass
           g_result.append(ck_xpath)
@@ -99,10 +100,12 @@ def get_hovered_element_html(url, explore, callback):
   except Exception as e:
     pass
 
+
 def break_func():
   keyboard_listener.stop()
   g_callback(None)
 
+
 def on_press(key):
   global keyboard_listener, ctrl_pressed, g_driver, g_quite_flag, picture_path
   if key == keyboard.Key.ctrl_l and not ctrl_pressed:
@@ -117,7 +120,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-"+cid.get_uuid())
+        picture_path = "%s.png" % os.path.join(application.Server.screenshotPath, "element-" + cid.get_uuid())
         select_element.screenshot(picture_path)
       except Exception:
         pass
@@ -126,11 +129,13 @@ def on_press(key):
     g_quite_flag = True
     g_callback(g_result)
 
+
 def on_release(key):
   global ctrl_pressed
   if key == keyboard.Key.ctrl_l:
     ctrl_pressed = False
 
+
 def start_keyboard_listener():
   global keyboard_listener
   keyboard_listener = keyboard.Listener(on_press=on_press, on_release=on_release)
@@ -183,6 +188,7 @@ if (shade) {
 }
 """
 
+
 class IE:
 
   @staticmethod
@@ -287,6 +293,7 @@ class Chrome:
 def callback(xpath):
   print("Hovered Element XPATH IS :", xpath)
 
+
 def get_element(url: str = None, explore: str = "chrome"):
   global keyboard_listener, ctrl_pressed, g_driver, g_callback, g_result, g_quite_flag, picture_path
   keyboard_listener = None
@@ -309,6 +316,7 @@ def get_element(url: str = None, explore: str = "chrome"):
 
   return g_result, picture_path
 
+
 if __name__ == "__main__":
   # from explore_record_core import get_hovered_element_html, Chrome
   g_result, picture_path = get_element("weibo.com")

ctools/auto/plan_area.py

@@ -63,8 +63,8 @@ class PlanAreaTools:
 
   def mouse_left_move(self, event):
     self.x, self.y = event.x, event.y
-    self.xcenter = self.xstart + int((self.x-self.xstart)/2)-2
-    self.ycenter = self.ystart + int((self.y-self.ystart)/2)-2
+    self.xcenter = self.xstart + int((self.x - self.xstart) / 2) - 2
+    self.ycenter = self.ystart + int((self.y - self.ystart) / 2) - 2
 
     self.pos_label.config(text=self.pos_div_text % (self.x, self.y))
     self.pos_div.geometry(f"+{self.x + 10}+{self.y + 10}")

ctools/auto/pty_process.py

@@ -70,4 +70,3 @@ class PtyTools:
 
   def close(self):
     self._process.close(force=True)
-

ctools/auto/screenshot.py

@@ -65,7 +65,7 @@ class ScreenshotTools:
   def change_center_point(self, event):
     offset_x = event.x + self.xstart - self.xcenter - 2
     offset_y = event.y + self.ystart - self.ycenter - 2
-    self.center_div.geometry(f"+{self.xstart+event.x-2}+{self.ystart+event.y-2}")
+    self.center_div.geometry(f"+{self.xstart + event.x - 2}+{self.ystart + event.y - 2}")
     self.center_offset = (offset_x, offset_y)
 
   def mouse_left_down(self, event):
@@ -86,8 +86,8 @@ class ScreenshotTools:
 
   def mouse_left_move(self, event):
     self.x, self.y = event.x, event.y
-    self.xcenter = self.xstart + int((self.x-self.xstart)/2)-2
-    self.ycenter = self.ystart + int((self.y-self.ystart)/2)-2
+    self.xcenter = self.xstart + int((self.x - self.xstart) / 2) - 2
+    self.ycenter = self.ystart + int((self.y - self.ystart) / 2) - 2
 
     self.pos_label.config(text=self.pos_div_text % (self.x, self.y))
     self.pos_div.geometry(f"+{self.x + 10}+{self.y + 10}")
@@ -125,4 +125,3 @@ def screenshot():
   st = ScreenshotTools()
   st.start()
   return st.screenshot_path, st.center_offset
-

ctools/auto/win_canvas.py

@@ -4,12 +4,14 @@ from multiprocessing import Process, Queue
 data_queue: Queue = None
 canvas_process: Process = None
 
+
 class MSGType:
-  BOTTOM = 'bottom'       #置于底部
-  DRAW = 'draw'           #绘制矩形
+  BOTTOM = 'bottom'  # 置于底部
+  DRAW = 'draw'  # 绘制矩形
+
 
 class CanvasInfo:
-  def __init__(self,title: str='', points: list=None, msg_type: MSGType=MSGType.DRAW):
+  def __init__(self, title: str = '', points: list = None, msg_type: MSGType = MSGType.DRAW):
     self.msg_type = msg_type
     # 标题
     self.title = title
@@ -17,9 +19,11 @@ class CanvasInfo:
     # [(100, 100), (200, 100), (200, 150), (100, 150)]
     self.points = points
 
+
 def on_ctrl_click(event):
   print("Ctrl+Left Click detected")
 
+
 class RecorderTool(tk.Tk):
   def __init__(self, data: Queue):
     super().__init__()
@@ -65,10 +69,12 @@ class RecorderTool(tk.Tk):
         self.draw_red_border(canvas_info)
     self.after(10, self.check_queue)
 
+
 def _init_recorder(data):
   canvas_app = RecorderTool(data)
   canvas_app.mainloop()
 
+
 def start():
   global data_queue, canvas_process
   data_queue = Queue()
@@ -76,8 +82,8 @@ def start():
   canvas_process.start()
   return canvas_process
 
+
 def stop():
   canvas_process.terminate()
   canvas_process.join()
   data_queue.close()
-

ctools/auto/win_control.py

@@ -6,9 +6,9 @@ import uiautomation as auto
 from pynput import keyboard
 
 from ctools import application
+from ctools import cid
 from ctools.auto import win_canvas
 from ctools.pools import thread_pool
-from ctools import cid
 
 current_control = None
 ctrl_pressed = False
@@ -16,13 +16,14 @@ keyboard_listener = None
 control_json = {}
 screenshot_path = ""
 
+
 class ControlInfo:
   def __init__(self, ControlType, ClassName, AutomationId, Name, Depth):
     self.ControlType = ControlType
     self.ClassName = ClassName
     self.AutomationId = AutomationId
     self.Name = Name
-    #self.Depth = Depth
+    # self.Depth = Depth
 
 
 def get_control_from_cursor():
@@ -36,10 +37,10 @@ def get_control_from_cursor():
       print("No control found {}".format(e))
       return
     if control:
-      #当前控件信息
+      # 当前控件信息
       global current_control
       current_control = control
-      #绘制矩形
+      # 绘制矩形
       canvas_info = win_canvas.CanvasInfo(control.Name, [
         (control.BoundingRectangle.left, control.BoundingRectangle.top),
         (control.BoundingRectangle.right, control.BoundingRectangle.top),
@@ -65,6 +66,7 @@ def get_control_from_cursor():
       # time.sleep(2)
       # xx.Click()
 
+
 def on_press(key):
   global ctrl_pressed, keyboard_listener
   if key == keyboard.Key.ctrl_l and not ctrl_pressed:
@@ -78,6 +80,7 @@ def on_press(key):
     keyboard_listener.stop()
     # pg.alert('采集成功!')
 
+
 def on_release(key):
   global ctrl_pressed
   if key == keyboard.Key.ctrl_l:
@@ -104,6 +107,7 @@ def get_control_json():
     time.sleep(0.5)
   return control_json, screenshot_path
 
+
 if __name__ == '__main__':
   a = get_control_json()
   print(a)

ctools/call.py

@@ -1,71 +1,56 @@
+import os
 import sched
 import threading
 import time
 from functools import wraps
 
+
 # annotation
+_global_once_cache = {}
 def once(func):
-  """
-  decorator to initialize a function once
-  :param func: function to be initialized
-  :return: the real decorator for return the result
-  """
-  initialized = False
-  res = None
-
+  code = func.__code__
+  key = f"{os.path.abspath(code.co_filename)}:{code.co_firstlineno}"
   def wrapper(*args, **kwargs):
-    nonlocal initialized, res
-    if not initialized:
-      res = func(*args, **kwargs)
-      initialized = True
-      return res
-    else:
-      return res
-
+    if key not in _global_once_cache:
+      _global_once_cache[key] = func(*args, **kwargs)
+    return _global_once_cache[key]
   return wrapper
 
 # annotation
+_cache = {}
 def init(func):
-  """
-  decorator to initialize a function automic
-  :param func: function to be initialized
-  :return: the real decorator for return the result
-  """
-  res = func()
-
+  code = func.__code__
+  key = f"{os.path.abspath(code.co_filename)}:{code.co_firstlineno}"
+  if key not in _cache:
+    _cache[key] = func()
   def wrapper():
-    return res
-
+    return _cache[key]
   return wrapper
 
 # annotation
-def schd(interval_seconds, start_by_call: bool = False):
-  start_flag = False
-  run_flag = False
-  scheduler = sched.scheduler(time.time, time.sleep)
-
+_scheduler_cache = {}
+def schd(interval_seconds, start_by_call=False, run_now=False):
   def decorator(func):
+    key = f"{os.path.abspath(func.__code__.co_filename)}:{func.__code__.co_firstlineno}"
+    lock = threading.Lock()
     @wraps(func)
     def wrapper(*args, **kwargs):
-      nonlocal start_flag
-      if start_by_call and not start_flag:
-        start_flag = True
-        threading.Thread(target=wrapper, args=args, kwargs=kwargs, daemon=True).start()
-        return
-
+      if key in _scheduler_cache:
+        return  # 已经调度过
+      scheduler = sched.scheduler(time.time, time.sleep)
       def job():
         func(*args, **kwargs)
         scheduler.enter(interval_seconds, 1, job)
-
-      nonlocal run_flag
-      if not run_flag:
-        scheduler.enter(interval_seconds, 1, job)
-        run_flag = True
-        scheduler.run()
-      else:
-        func(*args, **kwargs)
-
-    if not start_by_call: threading.Thread(target=wrapper, daemon=True).start()
+      def start_scheduler():
+        with lock:
+          if _scheduler_cache.get(key): return
+          _scheduler_cache[key] = True
+          if run_now:
+            func(*args, **kwargs)
+          scheduler.enter(interval_seconds, 1, job)
+          scheduler.run()
+      threading.Thread(target=start_scheduler, daemon=True).start()
+    if not start_by_call:
+      wrapper()
     return wrapper
-
   return decorator

ctools/cdate.py

@@ -1,39 +1,52 @@
 import time
 from datetime import datetime, timedelta
 
+def get_month(date: str=None):
+  if date: return time.strftime('%Y-%m', time.strptime(date, '%Y-%m-%d'))
+  return time.strftime('%Y-%m', time.localtime(time.time()))
+
 def get_date():
   return time.strftime('%Y-%m-%d', time.localtime(time.time()))
 
+
 def get_time():
   return time.strftime('%H-%M-%S', time.localtime(time.time()))
 
+
 def get_date_time(fmt="%Y-%m-%d %H:%M:%S"):
   return time.strftime(fmt, time.localtime(time.time()))
 
+
 def str_to_datetime(val: str, fmt="%Y-%m-%d %H:%M:%S"):
   return time.strptime(val, fmt)
 
+
 def str_to_timestamp(val: str, fmt="%Y-%m-%d %H:%M:%S"):
   return time.mktime(time.strptime(val, fmt))
 
-def timestamp_to_str(timestamp: int=time.time(), fmt="%Y-%m-%d %H:%M:%S"):
+
+def timestamp_to_str(timestamp: int = time.time(), fmt="%Y-%m-%d %H:%M:%S"):
   return time.strftime(fmt, time.localtime(timestamp))
 
+
 def get_today_start_end(now: datetime.now()):
   start = datetime(now.year, now.month, now.day, 0, 0, 0)
   end = datetime(now.year, now.month, now.day, 23, 59, 59)
   return start.strftime("%Y-%m-%d %H:%M:%S"), end.strftime("%Y-%m-%d %H:%M:%S")
 
+
 def get_week_start_end(now: datetime.now()):
   start = now - timedelta(days=now.weekday())  # 本周一
   end = start + timedelta(days=6)  # 本周日
   return start.strftime("%Y-%m-%d 00:00:00"), end.strftime("%Y-%m-%d 23:59:59")
 
-def time_diff_in_seconds(sub_head: str=get_date_time(), sub_end: str=get_date_time()):
+
+def time_diff_in_seconds(sub_head: str = get_date_time(), sub_end: str = get_date_time()):
   start_ts = str_to_timestamp(sub_head)
   end_ts = str_to_timestamp(sub_end)
   return int(start_ts - end_ts)
 
+
 def opt_time(base_time=None, days=0, hours=0, minutes=0, seconds=0, weeks=0, fmt="%Y-%m-%d %H:%M:%S"):
   if base_time is None:
     base_time = datetime.now()
@@ -41,3 +54,31 @@ def opt_time(base_time=None, days=0, hours=0, minutes=0, seconds=0, weeks=0, fmt
     base_time = datetime.strptime(base_time, fmt)
   new_time = base_time + timedelta(days=days, hours=hours, minutes=minutes, seconds=seconds, weeks=weeks)
   return new_time.strftime(fmt)
+
+def get_years_range(start_year: str, end_year: str=datetime.now().strftime("%Y")) -> set[int]:
+  if int(start_year) > int(end_year):
+    raise ValueError("起始年份不能大于结束年份")
+  return list(range(int(start_year), int(end_year) + 1))
+
+def get_month_range(start: str, end: str=datetime.now().strftime("%Y-%m")) -> set[str]:
+  start_date = datetime.strptime(start, "%Y-%m")
+  end_date = datetime.strptime(end, "%Y-%m")
+  if start_date > end_date:
+    raise ValueError("起始月份不能晚于结束月份")
+  result = []
+  current = start_date
+  while current <= end_date:
+    result.append(current.strftime("%Y-%m"))
+    if current.month == 12:
+      current = current.replace(year=current.year + 1, month=1)
+    else:
+      current = current.replace(month=current.month + 1)
+  return result
+
+def get_day_range(start: str, end: str=datetime.now().strftime("%Y-%m-%d")) -> set[str]:
+  start_date = datetime.strptime(start, "%Y-%m-%d")
+  end_date = datetime.strptime(end, "%Y-%m-%d")
+  if start_date > end_date:
+    raise ValueError("起始日期不能晚于结束日期")
+  delta = end_date - start_date
+  return [(start_date + timedelta(days=i)).strftime("%Y-%m-%d") for i in range(delta.days + 1)]

ctools/cid.py

@@ -2,17 +2,19 @@ from ctools.util.snow_id import SnowId
 
 idWorker = SnowId(1, 2, 0)
 
+
 def get_snowflake_id():
   return idWorker.get_id()
 
+
+def get_snowflake_id_str():
+  return str(get_snowflake_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/aes_util.py

@@ -1,5 +1,6 @@
 from cryptography.fernet import Fernet
 
+
 def generate_key():
   """
   生成 AES key
@@ -9,6 +10,7 @@ def generate_key():
   key = Fernet.generate_key()
   return key.decode()
 
+
 def aes_encrypt(sec_key, plaintext):
   """
   aes加密
@@ -31,5 +33,3 @@ def aes_decrypt(sec_key, ciphertext):
   cipher = Fernet(sec_key)
   decrypted_data = cipher.decrypt(ciphertext)
   return decrypted_data.decode()
-
-

ctools/cipher/b64.py

@@ -1,7 +1,9 @@
 import base64
 
+
 def encode(param: str):
   return base64.b64encode(param.encode('UTF-8')).decode('UTF-8')
 
+
 def decode(param: str):
   return base64.b64decode(param.encode('UTF-8')).decode('UTF-8')

ctools/cipher/czip.py

@@ -22,6 +22,8 @@ import pyzipper
   output_directory = '/Users/haoyang/Desktop'
   compress_specific_files(files_to_compress, output_directory, zip_password, "my_files")
 """
+
+
 def create_zip_with_files(file_dict, password=None) -> io.BytesIO:
   """Compress multiple files into a single password-protected ZIP archive in memory.
   Args:
@@ -87,7 +89,7 @@ def process_directory_to_single_zip(root_dir, password=None, zip_name=None):
     if 'zip_buffer' in locals(): zip_buffer.close()
 
 
-def compress_specific_files(file_paths:[], output_dir:str, password=None, zip_name=None):
+def compress_specific_files(file_paths: [], output_dir: str, password=None, zip_name=None):
   """Compress multiple specified files into a single ZIP archive.
   Args:
       file_paths: List of absolute file paths to compress

ctools/cipher/rsa.py

@@ -10,6 +10,7 @@ from ctools import cjson, path_info
 ENCRYPT_CHUNK_SIZE = 245
 decrypt_CHUNK_SIZE = 512
 
+
 def generate_rsa_keypair(bits=2048):
   key = RSA.generate(bits)
   private_key = key.export_key()  # 导出私钥
@@ -19,11 +20,13 @@ def generate_rsa_keypair(bits=2048):
   with open("public_key.pem", "wb") as f:
     f.write(public_key)
 
+
 def loadLicenseInfo(auth_code):
   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)
 
+
 # 加密函数
 def encrypt(msg, public_key):
   parts = b''
@@ -34,6 +37,7 @@ def encrypt(msg, public_key):
   encrypted_base64 = base64.b64encode(parts)
   return encrypted_base64.decode()
 
+
 # 解密函数
 def decrypt(msg, private_key):
   parts = b''
@@ -44,6 +48,7 @@ def decrypt(msg, private_key):
     parts += cipher.decrypt(encrypted_bytes[i:i + decrypt_CHUNK_SIZE])
   return parts.decode()
 
+
 # 验签
 def verify_sign(msg, public_key, sign):
   public_key = RSA.import_key(public_key)
@@ -55,6 +60,7 @@ def verify_sign(msg, public_key, sign):
     print('签名验证失败: {}'.format(e))
     return False
 
+
 # 签名
 def sign_msg(msg, private_key):
   private_key = RSA.import_key(private_key)
@@ -62,7 +68,6 @@ def sign_msg(msg, private_key):
   signature = pkcs1_15.new(private_key).sign(hash_message)
   return base64.b64encode(signature).decode()
 
-
 # with open(work_path.get_current_path() + '/private_key.pem', 'r') as key:
 #   key = key.read()
 #   sign = sign_msg(key, key)

ctools/cipher/sign.py

@@ -14,6 +14,7 @@ def generate_signature(file_path, key: str = global_key):
   except:
     return ''
 
+
 def digest(value: str, key: str = global_key):
   val_hash = hashlib.sha256(value.encode()).digest()
   sign_val = hmac.new(key.encode(), val_hash, hashlib.sha256).digest()

ctools/cipher/sm_util.py

@@ -5,6 +5,7 @@ from gmssl.sm4 import CryptSM4, SM4_ENCRYPT, SM4_DECRYPT
 
 sm2_crypt: sm2.CryptSM2 = None
 
+
 def init(private_key: str, public_key: str):
   global sm2_crypt
   if sm2_crypt is not None:
@@ -12,6 +13,7 @@ def init(private_key: str, public_key: str):
     return
   sm2_crypt = sm2.CryptSM2(private_key=private_key, public_key=public_key, asn1=True, mode=1)
 
+
 def sign_with_sm2(sign_data: str) -> str:
   if sm2_crypt is None: raise Exception('sm2 is not init!!!')
   return sm2_crypt.sign_with_sm3(sign_data.encode('UTF-8'))
@@ -25,6 +27,7 @@ def verify_with_sm2(sign_val: str, sign_data: str) -> bool:
     print('签名验证失败: {}'.format(e))
     return False
 
+
 def encrypt_with_sm2(encrypt_data: str) -> str:
   if sm2_crypt is None: raise Exception('sm2 is not init!!!')
   return base64.b64encode(sm2_crypt.encrypt(encrypt_data.encode('UTF-8'))).decode('UTF-8')