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

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

@@ -3,6 +3,7 @@ import threading
 import time
 from functools import wraps
 
+
 # annotation
 def once(func):
   """
@@ -24,6 +25,7 @@ def once(func):
 
   return wrapper
 
+
 # annotation
 def init(func):
   """
@@ -38,34 +40,28 @@ def init(func):
 
   return wrapper
 
+
 # annotation
-def schd(interval_seconds, start_by_call: bool = False):
-  start_flag = False
-  run_flag = False
+def schd(interval_seconds, start_by_call=False, run_now=False):
   scheduler = sched.scheduler(time.time, time.sleep)
-
+  lock = threading.Lock()
+  started = [False]  # 可变对象，线程可见
+  print("schd delay is: ", interval_seconds)
   def decorator(func):
     @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
-
       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()
-    return wrapper
-
+      def start_scheduler():
+        with lock:
+          if started[0]: return
+          started[0] = 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  # 如果是 start_by_call=True，返回 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,16 @@ 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/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')

ctools/cjson.py

@@ -6,6 +6,7 @@ jsonpickle.set_preferred_backend('json')
 jsonpickle.set_encoder_options('json', ensure_ascii=False)
 jsonpickle.set_decoder_options('json')
 
+
 def dumps(obj, **kwargs) -> str:
   """
   将对象转换为json字符串
@@ -17,6 +18,7 @@ def dumps(obj, **kwargs) -> str:
   if type(obj) == str: return obj
   return f'{jsonpickle.encode(obj, unpicklable=False, make_refs=False, **kwargs)}'
 
+
 def loads(json_str: str, **kwargs) -> dict:
   """
   将json字符串转换为对象
@@ -25,6 +27,7 @@ def loads(json_str: str, **kwargs) -> dict:
   """
   return jsonpickle.decode(json_str, **kwargs)
 
+
 def unify_to_str(json_str: str) -> str:
   if not str_value_keys and len(str_value_keys) == 0: return json_str
   obj = loads(json_str)
@@ -34,6 +37,7 @@ def unify_to_str(json_str: str) -> str:
     _handle_dict(obj)
   return dumps(obj)
 
+
 def _handle_list(data):
   for o in data:
     if isinstance(o, list):
@@ -41,6 +45,7 @@ def _handle_list(data):
     elif isinstance(o, dict):
       _handle_dict(o)
 
+
 def _handle_dict(data):
   for k, v in data.items():
     if isinstance(v, list):

ctools/cron_lite.py

@@ -29,6 +29,7 @@ print(123123)
 cron_lite.start_all()
 """
 
+
 class SchedulerMeta:
   timer_task_name: str = None
   switch: bool = True
@@ -40,7 +41,7 @@ class SchedulerMeta:
 
 scheduler_map: Dict[str, SchedulerMeta] = {}  # {timer_task_name: SchedulerMeta}
 _switch = False
-_info_handler  = print
+_info_handler = print
 _error_handler = print
 _time_zone: Optional[pytz.BaseTzInfo] = pytz.timezone("Asia/Shanghai")
 
@@ -49,6 +50,7 @@ def set_time_zone(time_zone_name: str):
   global _time_zone
   _time_zone = pytz.timezone(time_zone_name)
 
+
 # @annotation
 def cron_task(cron_expr: str, task_name: str = None, till_time_stamp: int = None):
   """
@@ -92,8 +94,9 @@ def reg_cron_task(cron_expr, func, params, timer_task_name=None, till_time_stamp
   :return: the real decorator
   """
   cron_expr = _convert_cron(cron_expr)
-  assert len(cron_expr.split(" ")) in (5, 6),  "Only supported <minute hour day month weekday> and <minute hour day month weekday second>"
+  assert len(cron_expr.split(" ")) in (5, 6), "Only supported <minute hour day month weekday> and <minute hour day month weekday second>"
   task_name = func.__name__ if timer_task_name is None else timer_task_name
+
   @wraps(func)
   def wrapper(*args, **kwargs):
     try:
@@ -107,6 +110,7 @@ def reg_cron_task(cron_expr, func, params, timer_task_name=None, till_time_stamp
 
   _register_next(task_name, wrapper, cron_expr, till_time_stamp, init=True)
 
+
 def start_all(spawn: bool = True, daemon: bool = True, info_handler=None, error_handler=None) -> Optional[threading.Thread]:
   """
   start_all starts all cron tasks registered before.
@@ -144,6 +148,7 @@ def active(timer_task_name):
   if timer_task_name in scheduler_map:
     scheduler_map.get(timer_task_name).status = True
 
+
 def get_switch(timer_task_name):
   switch = True
   if timer_task_name in scheduler_map:
@@ -215,6 +220,7 @@ def _run_sched(scheduler_meta: SchedulerMeta):
         return
       time.sleep(0.5)
 
+
 def _start(taskName: str = None):
   global _switch
   _switch = True
@@ -223,7 +229,7 @@ def _start(taskName: str = None):
   for timer_task_name, scheduler_meta in scheduler_map.items():
     if taskName is not None and timer_task_name != taskName: continue
     print("register job: ", timer_task_name, ", cron: ", scheduler_meta.cron_str)
-    thread = threading.Thread(target=_run_sched, args=(scheduler_meta, ), daemon=True)
+    thread = threading.Thread(target=_run_sched, args=(scheduler_meta,), daemon=True)
     thread.start()
     taskList.append(thread)
   for task in taskList: task.join()
@@ -231,6 +237,7 @@ def _start(taskName: str = None):
   _switch = False
   scheduler_map.clear()
 
+
 def _convert_cron(cron_expr):
   res_cron = ""
   cron_list = cron_expr.split(" ")
@@ -242,4 +249,3 @@ def _convert_cron(cron_expr):
   else:
     res_cron = cron_expr
   return res_cron
-