byzh-core 0.0.1__py3-none-any.whl

byzh_core/Bterminal/cmd.py

@@ -0,0 +1,116 @@
+import os
+from pathlib import Path
+import subprocess
+import time
+
+from ..Bbasic import B_Color
+
+
+def args_process(args: tuple) -> list:
+    lst = []
+    for x in args:
+        if type(x) is str:
+            lst.append(x)
+        elif type(x) is list:
+            lst.extend(x)
+    return lst
+
+def b_run_cmd(
+        *args: str,
+        show: bool = True,
+):
+    '''
+    可传入多个字符串, 在cmd中运行
+    :param args:
+    :param show: 若show=True, 则会单开一个cmd, 在cmd中运行
+    :return:
+    '''
+    command = ''
+    for i in range(len(args)):
+        if i == len(args) - 1:
+            command += str(args[i])
+            break
+        command += str(args[i]) + ' && '
+    if show:
+        command = f'start cmd /K "{command}"'
+    # print(command)
+    subprocess.run(command, shell=True)
+
+def b_run_python(
+    *args: str|list[str],
+    limit_time: int|float|None = None,
+    log_path: Path|None = None
+):
+    '''
+    可传入多个字符串, 在当前python环境下运行
+    :param args: 以python开头, 用于运行.py文件
+    :param show:
+    :return:
+    '''
+    def run_log(content=''):
+        if log_path is not None:
+            parent = Path(log_path).parent
+            os.makedirs(parent, exist_ok=True)
+            with open(log_path, 'w', encoding='utf-8') as f:
+                f.write("=====================\n")
+                for string in str_lst:
+                    f.write("\t" + string + '\n')
+                f.write("=====================\n")
+                f.write(content)
+
+    str_lst = args_process(args)
+
+    print(f"{B_Color.GREEN}=====================")
+    print("BRunPython 将在3秒后开始:")
+    for string in str_lst:
+        print("\t" + string)
+    print(f"====================={B_Color.RESET}")
+    time.sleep(3)
+
+    for string in str_lst:
+        try:
+            command_lst = string.split(' ')
+            run_log("正在执行: " + string)
+            start = time.time()
+            result = subprocess.run(command_lst, timeout=limit_time)
+            end = time.time()
+            delta_t = end - start
+            h_t = int(delta_t // 3600)
+            m_t = int((delta_t % 3600) // 60)
+            s_t = int(delta_t % 60)
+            if result.returncode != 0: # 报错
+                index = str_lst.index(string)
+                str_lst[index] = string + f"\t[!!!Error!!!] [{h_t}h {m_t}m {s_t}s]"
+            else:
+                index = str_lst.index(string)
+                str_lst[index] = string + f"\t[~Success~] [{h_t}h {m_t}m {s_t}s]"
+        except subprocess.TimeoutExpired:
+            print(f"程序运行超过 {limit_time} 秒，已被强制终止")
+            h_t = int(limit_time // 3600)
+            m_t = int((limit_time % 3600) // 60)
+            s_t = int(limit_time % 60)
+            index = str_lst.index(string)
+            str_lst[index] = string + f"\t[!!!Time limit!!!] [{h_t}h {m_t}m {s_t}s]"
+
+    print(f"{B_Color.GREEN}====================={B_Color.RESET}")
+    print(f"{B_Color.GREEN}BRunPython 结束:{B_Color.RESET}")
+    for string in str_lst:
+        if 'Time limit' in string:
+            print(f"\t{B_Color.YELLOW}" + string + f"{B_Color.RESET}")
+        elif 'Error' in string:
+            print(f"\t{B_Color.RED}" + string + f"{B_Color.RESET}")
+        else:
+            print(f"\t{B_Color.GREEN}" + string + f"{B_Color.RESET}")
+    print(f"{B_Color.GREEN}====================={B_Color.RESET}")
+
+    run_log('结束')
+
+
+
+if __name__ == '__main__':
+    b_run_cmd("echo hello", "echo world", "echo awa", show=True)
+    # b_run_python(
+    #     r"python E:\byzh_workingplace\byzh-rc-to-pypi\test1.py",
+    #     r"python E:\byzh_workingplace\byzh-rc-to-pypi\test2.py",
+    #     limit_time=3,
+    # )

byzh_core/Btqdm/__init__.py

@@ -0,0 +1,4 @@
+
+from .b_tqdm import B_Tqdm
+
+__all__ = ['B_Tqdm']

byzh_core/Btqdm/b_tqdm.py

@@ -0,0 +1,76 @@
+import sys
+import time
+
+from ..Bbasic import B_Color
+
+
+class B_Tqdm:
+    def __init__(
+            self,
+            total: int|None = None,
+            prefix: str = 'Processing',
+            suffix: str = '',
+            length: int = 20,
+            fill: str = '█',
+    ):
+        """
+        类似tqdm的进度条
+        :param total: 总数
+        :param prefix: 前缀
+        :param suffix: 后缀
+        :param length: 进度条长度(字符)
+        :param fill: 填充字符
+        """
+        super().__init__()
+        self.total = total
+        self.prefix = prefix
+        self.suffix = suffix
+        self.length = length
+        self.fill = fill
+        self.start_time = 0
+        self.current = 0
+
+    def _format_time(self, seconds):
+        """将秒数转换为mm:ss格式"""
+        minutes = int(seconds // 60)
+        seconds = int(seconds % 60)
+        return f'{minutes:02}:{seconds:02}'
+
+    def update(self, step=1, color=B_Color.BLUE, prefix=None, suffix=None):
+        if self.current == 0:
+            self.start_time = time.time()
+        if prefix is not None:
+            self.prefix = prefix
+        if suffix is not None:
+            self.suffix = suffix
+
+        # 更新进度
+        self.current += step
+
+        # 计算已用时间
+        elapsed_time = time.time() - self.start_time
+        elapsed_str = self._format_time(elapsed_time)
+        # 预估剩余时间
+        if self.total is not None:
+            estimated_time = elapsed_time / self.current * (self.total - self.current) if self.current > 0 else 0
+            estimated_str = self._format_time(estimated_time)
+        # 计算每秒处理的项数
+        speed = self.current / elapsed_time if elapsed_time > 0 else 0
+
+        # 更新进度条
+        if self.total is not None:
+            filled_length = int(self.length * self.current // self.total)
+            bar = self.fill * filled_length + '-' * (self.length - filled_length)
+
+            sys.stdout.write(f'\r{color}{self.prefix} |{bar}|'
+                             f' {self.current}/{self.total} -> {elapsed_str}<{estimated_str} | {speed:.1f} it/s |'
+                             f' {self.suffix}{B_Color.RESET}')
+        else:
+            sys.stdout.write(f'\r{color}{self.prefix} | {self.current} iters -> {elapsed_str} | {speed:.1f} it/s |'
+                             f' {self.suffix}{B_Color.RESET}')
+        sys.stdout.flush()
+
+        # 补回车
+        if self.current == self.total:
+            sys.stdout.write('\n')
+            sys.stdout.flush()

byzh_core/Bwriter/__init__.py

@@ -0,0 +1,5 @@
+
+from .writer import B_Writer
+from .globalwriter import B_GlobalWriter
+
+__all__ = ['B_Writer', 'B_GlobalWriter']

byzh_core/Bwriter/globalwriter.py

@@ -0,0 +1,62 @@
+import threading
+import time
+import atexit
+
+
+class B_GlobalWriter():
+    _instance = None
+    _lock = threading.Lock()
+
+    def __new__(cls):
+        if cls._instance is None:
+            with cls._lock:
+                if cls._instance is None:
+                    cls._instance = super().__new__(cls)
+                    # 初始化
+                    cls._instance.file = None
+                    cls._instance.f = None
+                    cls._instance.ifTime = False
+                    atexit.register(cls._instance.closeFile)  # 注册关闭方法
+        return cls._instance
+
+    @classmethod
+    def setFile(cls, file, ifTime=False):
+        cls()
+        cls._instance.ifTime = ifTime
+        cls._instance.path = file
+        cls._instance.f = open(cls._instance.path, "a", encoding="utf-8")
+
+    @classmethod
+    def clearFile(cls):
+        cls()
+        assert cls._instance.f is not None, "请先调用setFile方法"
+        cls._instance.f.close()
+        cls._instance.f = open(cls._instance.path, 'w', encoding="utf-8")
+
+    @classmethod
+    def closeFile(cls):
+        if cls._instance.f:
+            cls._instance.f.close()
+            cls._instance.f = None
+
+    @classmethod
+    def toCmd(cls, string):
+        cls()
+        print(string)
+
+    @classmethod
+    def toFile(cls, string):
+        cls()
+        assert cls._instance.f is not None, "请先调用setFile方法"
+        if  cls._instance.ifTime:
+            t = time.strftime("%Y-%m-%d %H:%M:%S ##### ", time.localtime())
+            cls._instance.f.write(t)
+        cls._instance.f.write(string)
+        cls._instance.f.write("\n")
+        cls._instance.f.flush()
+
+    @classmethod
+    def toBoth(cls, string):
+        cls()
+        cls.toFile(string)
+        cls.toCmd(string)

byzh_core/Bwriter/writer.py

@@ -0,0 +1,138 @@
+from pathlib import Path
+from typing import Literal
+import time
+import os
+
+from ..Bbasic import B_Color
+
+COLOR_DICT = {
+    "default": B_Color.RESET,
+
+    "black": B_Color.BLACK,
+    "red": B_Color.RED,
+    "green": B_Color.GREEN,
+    "yellow": B_Color.YELLOW,
+    "blue": B_Color.BLUE,
+    "purple": B_Color.PURPLE,
+    "cyan": B_Color.CYAN,
+    "silver": B_Color.SILVER
+}
+Color_Literal = Literal["default", "black", "red", "green", "yellow", "blue", "purple", "cyan", "silver"]
+
+class B_Writer:
+    def __init__(
+            self,
+            path: Path,
+            ifTime: bool = False,
+            color: Color_Literal = 'default',
+    ):
+        '''
+        :param path: 日志保存路径
+        :param ifTime: 是否输出时间
+        '''
+        super().__init__()
+        self.path = Path(path)
+        self.ifTime = ifTime
+        self.toWant_file = False
+        self.toWant_cmd = False
+
+        self.__checkColor(color)
+        self.color = color
+
+        self.f = None
+        self.setFile(self.path, self.ifTime)
+
+    def setFile(self, file: Path, ifTime=False):
+        '''
+        设置 file的path 以及 writer的ifTime
+        :param file: 设置log路径
+        :param ifTime:
+        :return:
+        '''
+        if self.f is not None:
+            self.f.close()
+        self.path = Path(file)
+        self.ifTime = ifTime
+        self.__createDir(self.path)
+        self.f = open(self.path, "a", encoding="utf-8")
+
+    def clearFile(self):
+        '''
+        清空内容
+        '''
+        assert self.f is not None, "请先调用setFile方法"
+        self.f.close()
+        self.f = open(self.path, 'w', encoding="utf-8")
+
+    def closeFile(self):
+        '''
+        关闭log
+        '''
+        if self.f:
+            self.f.close()
+            self.f = None
+
+    def toCmd(self, string, color: Color_Literal = None):
+        '''
+        打印到terminal
+        '''
+        # 检查color是否在字典中
+        if color is None:
+            print(COLOR_DICT.get(self.color) + str(string) + B_Color.RESET)
+        else:
+            assert color in COLOR_DICT, f"color参数错误，请输入{COLOR_DICT.keys()}"
+            print(COLOR_DICT.get(color) + str(string) + B_Color.RESET)
+
+    def toFile(self, string, ifTime=None):
+        '''
+        写入到文件内
+        '''
+        assert self.f is not None, "请先调用setFile方法"
+
+        if ifTime == False:
+            pass
+        elif ifTime==True or self.ifTime==True:
+            t = time.strftime("%Y-%m-%d %H:%M:%S ##### ", time.localtime())
+            self.f.write(t)
+
+        self.f.write(str(string))
+        self.f.write("\n")
+        self.f.flush()
+
+    def toBoth(self, string, ifTime=False, color: Color_Literal = None):
+        '''
+        同时写入到文件和terminal
+        :param string:
+        :param color:
+        :return:
+        '''
+        self.toFile(str(string), ifTime)
+        self.toCmd(str(string), color)
+
+    def setWant(self, toCmd=False, toFile=False):
+        '''
+        toWant的全局选项设置
+        '''
+        self.toWant_cmd = toCmd
+        self.toWant_file = toFile
+
+    def toWant(self, string, ifTime=False, color: Color_Literal = None):
+        '''
+        使用前先调用setWant方法
+        '''
+        if self.toWant_cmd and self.toWant_file:
+            self.toBoth(string, ifTime, color)
+        elif self.toWant_cmd:
+            self.toCmd(string, color)
+        elif self.toWant_file:
+            self.toFile(string, ifTime)
+        else:
+            raise Exception("请先调用setWant方法, 设置toCmd或toFile为True")
+
+    def __checkColor(self, color):
+        assert color in COLOR_DICT, f"color参数错误，请输入{list(COLOR_DICT.keys())}"
+
+    def __createDir(self, path):
+        # 获取到该文件的文件夹
+        dir = path.parents[0]
+        os.makedirs(dir, exist_ok=True)

byzh_core/__init__.py

@@ -0,0 +1,5 @@
+# package    以"B"开头
+# class      以"B_"开头
+# function   以"b_"开头
+
+__version__ = '0.0.1'

byzh_core-0.0.1.dist-info/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) [2024] [zhengyu]
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

byzh_core-0.0.1.dist-info/METADATA

@@ -0,0 +1,18 @@
+Metadata-Version: 2.1
+Name: byzh_core
+Version: 0.0.1
+Summary: byzh_core是byzh系列的核心库，包含了一些常用的工具函数和类。
+Author: byzh_rc
+License: MIT
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: wcwidth
+
+# progressBar.py
+just like tqdm
+
+# log.py
+jsut like loguru
+
+# train.py
+provide Trainer

byzh_core-0.0.1.dist-info/RECORD

@@ -0,0 +1,28 @@
+byzh_core/__init__.py,sha256=UkNGyvELAvNTWs8-DnCFe4RDLkKKHYzpUKoEVT--V48,108
+byzh_core/Barchive/__init__.py,sha256=wUdz646VS0Uhq9lwMkd3YQHCvQhLDqgADoDjFGMqIn0,65
+byzh_core/Barchive/archive.py,sha256=yIQgef1APUcZdHM_7Pa9rCam-P3PA3pbcSJSaySoUj4,2498
+byzh_core/Bbasic/__init__.py,sha256=wr7Y7YJINhgpV5X6BT5DaGJKcHUOZYNerCGXoi2Egac,116
+byzh_core/Bbasic/text_style.py,sha256=JelgL3AgcH607Mr-Bvr4u8Svb0twmNmqwXvOl4hHOSs,3161
+byzh_core/Bconfig/__init__.py,sha256=7lAp7wNetW0AyJcike7ekdY_8wrQQtFjBsi6684t_lU,54
+byzh_core/Bconfig/config.py,sha256=P9C4kAkxKnIFaJCErBXdABfOyWgFgVHpR-i-fiZXCNs,10956
+byzh_core/Bmath/__init__.py,sha256=G3AQug6izkEX3UfzO_mqUNJW88ZKlmYb4Q8CAactK4w,105
+byzh_core/Bmath/divides.py,sha256=dr85IqGSE9NvugQu7az29GLcjRiLjXU_hZTwtNifIXg,1132
+byzh_core/Bmath/get_norm.py,sha256=y1QsCTo7qhtGTXIxpc6JtamLWloC7ENRuVdNtK5yi58,596
+byzh_core/Btable/__init__.py,sha256=wpfHMPD4awPRDJVYlaiGtkkpjq2srWB7d7LrWhoX5R0,161
+byzh_core/Btable/auto_table.py,sha256=JREmn6KIlZKkv1Vsc23Ow3fDiJAgfMppbGKDUY8rfyk,14002
+byzh_core/Btable/row_table.py,sha256=rIX0-Yvb3RnO0RJBkA4m18gux1zYSnEKTy6uQGcWilc,5999
+byzh_core/Btable/xy_table.py,sha256=-KHK8EUlkJj3Rh0sp_KHI0QFt29AlMASMmR8q8ykUBM,6784
+byzh_core/Btable/obsolete/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+byzh_core/Btable/obsolete/b_auto_table.py,sha256=Cmblv_pknQeMew_KLO8vGdeSRtAykwbMh30QVhgS3UI,10422
+byzh_core/Bterminal/__init__.py,sha256=B6p6mHVhi5LAHKlvqfrv20E6XNqUsfU2xD753V6-Zf4,83
+byzh_core/Bterminal/cmd.py,sha256=iK0fwI0D3peEwUYYZWwnl9X8ImFfrBsEq6ySd3DcRdE,3839
+byzh_core/Btqdm/__init__.py,sha256=wyVbj7X1YpMeGJlriwISV-blZj4hfAk4ihpPMrUR8a0,52
+byzh_core/Btqdm/b_tqdm.py,sha256=692mfNqXhCwXunIpIMNVDt-5GRz4XMVCbcbiM4F9WGM,2620
+byzh_core/Bwriter/__init__.py,sha256=KSsbCJZ-1j5wOr-ZbTg8K3A8Du73d22DQVLdmq-3CBo,116
+byzh_core/Bwriter/globalwriter.py,sha256=tSjWxzLylAeZep67n5jbRsjQkXkBKRZLvKXUyGSY92Q,1824
+byzh_core/Bwriter/writer.py,sha256=px0ZNoSuXS_RbwVnYsBx2lYohyhHACfHM8-HBNEflhU,4006
+byzh_core-0.0.1.dist-info/LICENSE,sha256=-nRwf0Xga4AX5bsWBXXflpDpgX_U23X06oAMcdf0dSY,1089
+byzh_core-0.0.1.dist-info/METADATA,sha256=sjh9E-FrWF2_82hv2RqEbVsy1lM3OyH-mKUy_JzVjYc,365
+byzh_core-0.0.1.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+byzh_core-0.0.1.dist-info/top_level.txt,sha256=Xv-pzvl6kPdIbi5UehQcUdGhLtb8-4WhS5dRK1LINwM,10
+byzh_core-0.0.1.dist-info/RECORD,,

byzh_core-0.0.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (72.1.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

byzh_core-0.0.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+byzh_core