byzh-core 0.0.2.14__py3-none-any.whl → 0.0.3.0__py3-none-any.whl

byzh_core/B_os.py

@@ -1,9 +1,13 @@
 import shutil
 import os
 from pathlib import Path
-from typing import Literal
 
-def get_parent_dir(path: Literal['__file__']) -> Path:
+from .Butils import b_validate_params
+
+@b_validate_params({
+    "path": __file__
+})
+def get_parent_dir(path) -> Path:
     '''
     获取 该py文件 所在的文件夹
     :param path: __file__
@@ -52,6 +56,14 @@ def makedirs(path):
     if is_file(path):
         os.makedirs(path.parent, exist_ok=True)
 
+def makefile(path):
+    path = Path(path)
+
+    parent_dir = path.parent
+    makedirs(parent_dir)
+
+    path.touch() # 创建文件
+
 def rm(path):
     if os.path.isdir(path):
         shutil.rmtree(path)
@@ -86,6 +98,59 @@ def get_filepaths_in_dir(root_dir_path, exclude_name=[], exclude_suffix=['.pyc']
             file_paths.append(file_path)
     return file_paths
 
+@b_validate_params({
+    "black_dirnames": {"examples": ["__pycache__", ".git"]},
+    "black_filenames": {"examples": ["__init__.py", "test.py"]},
+    "black_stems": {"examples": ["test", "example"]},
+    "black_exts": {"examples": [".pyc"]}
+})
+def walk(
+    root: str,
+    black_dirnames: list[str] = None,
+    black_filenames: list[str] = None,
+    black_stems: list[str] = None,
+    black_exts: list[str] = None,
+):
+    """
+    遍历目录, 类似os.walk, 但可以指定黑名单
+    :param root:
+    :param black_dirnames:
+    :param black_stems: 文件名
+    :param black_stems: 文件名, 不包括后缀
+    :param black_exts: 文件名的后缀(含.)
+    :return: root, dirs, files
+    """
+
+    def get_lst(lst):
+        return [] if lst is None else lst
+    black_dirnames = get_lst(black_dirnames)
+    black_filenames = get_lst(black_filenames)
+    black_stems = get_lst(black_stems)
+    black_exts = get_lst(black_exts)
+
+    dirs = []
+    files = []
+    for entry in os.listdir(root):
+        path = os.path.join(root, entry)
+        # 是文件夹
+        if os.path.isdir(path):
+            dirs.append(entry)
+        # 是文件
+        elif os.path.isfile(path):
+            files.append(entry)
+        else:
+            continue
+    dirs = [d for d in dirs if d not in black_dirnames]
+    files = [f for f in files if f not in black_filenames]
+    files = [f for f in files if os.path.splitext(f)[0] not in black_stems]
+    files = [f for f in files if os.path.splitext(f)[1] not in black_exts]
+
+    yield root, dirs, files
+
+    for dirname in dirs:
+        new_path = os.path.join(root, dirname)
+        yield from walk(new_path, black_dirnames, black_filenames, black_stems, black_exts)
+
 if __name__ == '__main__':
     # print(get_dirpaths_in_dir(r'E:\byzh_workingplace\byzh-rc-to-pypi'))
     a = get_filepaths_in_dir(r'E:\byzh_workingplace\byzh-rc-to-pypi')

byzh_core/Barchive/archive.py

@@ -1,16 +1,29 @@
 import os
 import zipfile
+import time
 from pathlib import Path
 
+from .. import B_os
 from ..Btqdm import B_Tqdm
+from ..Butils import b_validate_params
 
+@b_validate_params({
+    "black_dirnames": {"examples": ["__pycache__", ".git", ".idea"]},
+    "black_filenames": {"examples": [".gitignore", "README.md"]},
+    "black_stems": {"examples": ["README"]},
+    "black_exts": {"examples": [".csv", ".npy", ".pt", ".pth"]},
+    "white_filepath": {"examples": ["./.gitignore", "./README.md"]},
+    "white_dirpath": {"examples": [r"E:\byzh_workingplace\...\文件夹1"]},
+})
 def b_archive_zip(
     source_path,
     output_path:str|Path=None,
-    exclude_dirs:list[str]=None,
-    exclude_files:list[str]=None,
-    exclude_exts:list[str]=None,
-    while_list:list[str]=None,
+    black_dirnames:list[str]=None,
+    black_filenames:list[str]=None,
+    black_stems:list[str]=None,
+    black_exts:list[str]=None,
+    white_filepath:list[str]=None,
+    white_dirpath:list[str]=None,
     contain_empty_folder:bool=True,
 ):
     '''
@@ -20,61 +33,126 @@ def b_archive_zip(
 
     :param source_path:
     :param output_path: 如果不传入, 则默认为source_path同名同路径的zip文件
-    :param exclude_dirs: 黑名单-文件夹名 >>> ['__pycache__', '.git', '.idea']
-    :param exclude_files: 黑名单-文件全名 >>> ['.gitignore', 'README.md']
-    :param exclude_exts: 黑名单-文件后缀(包括.) >>> ['.csv', '.npy', '.pt', '.pth']
-    :param while_list: 白名单-文件全名 >>> ['.gitignore', 'README.md']
+    :param black_dirnames: 黑名单-文件夹名
+    :param black_filenames: 黑名单-文件全名
+    :param black_stems: 黑名单-文件名(不含后缀)
+    :param black_exts: 黑名单-文件后缀(包括.)
+    :param white_filepath: 白名单-文件全名
+    :param white_dirpath: 白名单-文件夹路径
     :param contain_empty_folder: 是否包含空文件夹
     :return:
     '''
     output_path = source_path + '.zip' if output_path is None else output_path
+    if os.path.exists(output_path):
+        print(f"检测到{output_path}已存在, 将在3秒后删除...")
+        time.sleep(3)
+        B_os.rm(output_path)
+        print(f"{output_path}已删除")
     def get_lst(lst):
         return [] if lst is None else lst
-    exclude_dirs = get_lst(exclude_dirs)
-    exclude_files = get_lst(exclude_files)
-    exclude_exts = get_lst(exclude_exts)
-    while_list = get_lst(while_list)
+    black_dirnames = get_lst(black_dirnames)
+    black_filenames = get_lst(black_filenames)
+    black_stems = get_lst(black_stems)
+    black_exts = get_lst(black_exts)
+    white_filepath = get_lst(white_filepath)
+    white_dirpath = get_lst(white_dirpath)
 
-
-    my_tqdm = B_Tqdm(prefix='Archive')
+    my_tqdm = B_Tqdm(prefix='[b_archive_zip]')
 
     with zipfile.ZipFile(output_path, 'w', zipfile.ZIP_DEFLATED) as zipf:
-        # 压缩文件:
+        # 如果传入的是文件:
         if os.path.isfile(source_path):
             arcname = os.path.basename(source_path)
             zipf.write(source_path, arcname)
             my_tqdm.update(1)
             return
 
-        # 压缩文件夹:
+        # 如果传入的是文件夹:
         if os.path.isdir(source_path):
-            for root, dirs, files in os.walk(source_path):
-                # 排除指定文件夹dirs
-                dirs[:] = [d for d in dirs if d in while_list or d not in exclude_dirs]
-                # 排除指定文件files
-                files = [f for f in files if f in while_list or f not in exclude_files]
-                # 排除指定后缀文件files
-                files = [f for f in files if f in while_list or not any(f.endswith(ext) for ext in exclude_exts)]
+            for root, dirs, files in B_os.walk(source_path,
+                                               black_dirnames=black_dirnames,
+                                               black_filenames=black_filenames,
+                                               black_stems=black_stems,
+                                               black_exts=black_exts):
 
                 # 压缩文件:
                 for file in files:
                     file_path = os.path.join(root, file)
                     arcname = os.path.relpath(file_path, source_path) # 相对于source_path的相对路径
                     zipf.write(file_path, arcname)
-
                     my_tqdm.update(1)
+                # 压缩文件夹:
+                for i, dir in enumerate(dirs):
+                    if len(os.listdir(os.path.join(root, dir))) == 0:
+                        if not contain_empty_folder:
+                            dirs.pop(i)
+                            continue
+                    dir_path = os.path.join(root, dir)
+                    arcname = os.path.relpath(dir_path, source_path) # 相对于source_path的相对路径
+                    zipf.write(dir_path, arcname)
+                    my_tqdm.update(1)
+
+        # 白名单-文件:
+        for filepath in white_filepath:
+            arcname = os.path.relpath(filepath, source_path)
+            zipf.write(filepath, arcname)
+            my_tqdm.update(1)
 
-                # 若是空文件夹，则压缩文件夹:
-                if contain_empty_folder and (len(dirs) == 0 and len(files) == 0):
-                    arcname = os.path.relpath(root, source_path)
-                    folder_path = root
-                    zipf.write(folder_path, arcname)
+        # 白名单-文件夹:
+        for dirpath in white_dirpath:
+            for root, dirs, files in B_os.walk(dirpath):
+                # 压缩文件:
+                for file in files:
+                    file_path = os.path.join(root, file)
+                    arcname = os.path.relpath(file_path, source_path)  # 相对于source_path的相对路径
+                    zipf.write(file_path, arcname)
+                    my_tqdm.update(1)
+                # 压缩文件夹:
+                for i, dir in enumerate(dirs):
+                    if len(os.listdir(os.path.join(root, dir))) == 0:
+                        if not contain_empty_folder:
+                            dirs.pop(i)
+                            continue
+                    dir_path = os.path.join(root, dir)
+                    arcname = os.path.relpath(dir_path, source_path)  # 相对于source_path的相对路径
+                    zipf.write(dir_path, arcname)
                     my_tqdm.update(1)
 
 if __name__ == '__main__':
+    # 不包含空文件夹:`文件夹111`
+    B_os.makedirs('根文件夹/文件夹1/文件夹11/文件夹111')
+    contain_empty_folder = False
+
+    # 不包含`文件夹12`
+    B_os.makefile('根文件夹/文件夹1/文件夹12/file1.txt')
+    B_os.makefile('根文件夹/文件夹1/文件夹12/文件夹123/file2.txt')
+    B_os.makefile('根文件夹/文件夹1/文件夹12/文件夹1234/文件夹12345/file3.txt')
+    black_dirnames = ['文件夹12']
+
+    # 不包含`file4.txt`, 且因此`文件夹21`变为空文件夹, 因此不包含
+    B_os.makefile('根文件夹/文件夹2/文件夹21/file4.txt')
+    black_filenames = ['file4.txt']
+
+    # 不包含`awa.txt`, `awa.html`
+    B_os.makefile('根文件夹/文件夹3/awa.txt')
+    B_os.makefile('根文件夹/文件夹3/文件夹31/awa.html')
+    black_stems = ['awa']
+
+    # 不包含`file5.csv`
+    B_os.makefile('根文件夹/文件夹4/file5.csv')
+    black_exts = ['.csv']
+
+    white_filepath = ['根文件夹/文件夹1/文件夹12/文件夹1234/文件夹12345/file3.txt']
+    white_dirpath = ['根文件夹/文件夹1/文件夹12/文件夹123']
+
     b_archive_zip(
-        source_path=r'E:\byzh_workingplace\byzh-rc-to-pypi\mnist',
-        output_path=r'E:\byzh_workingplace\byzh-rc-to-pypi\awaqwq.zip',
-        exclude_dirs=['__pycache__', 'com'],
-        exclude_exts=['.ppt']
+        source_path=r'根文件夹',
+        output_path=r'awaqwq.zip',
+        black_dirnames=black_dirnames,
+        black_filenames=black_filenames,
+        black_stems=black_stems,
+        black_exts=black_exts,
+        white_filepath=white_filepath,
+        white_dirpath=white_dirpath,
+        contain_empty_folder=contain_empty_folder,
     )

byzh_core/Btable/auto_table.py

@@ -1,5 +1,4 @@
 import copy
-import math
 import os
 from pathlib import Path
 from typing import List, Tuple, Union

byzh_core/Butils/__init__.py

@@ -1,7 +1,9 @@
 from .text_style import B_Color, B_Appearance, B_Background
 from .timer import B_Timer
+from .decorator import b_validate_params
 
 __all__ = [
     'B_Color', 'B_Appearance', 'B_Background',
-    'B_Timer'
+    'B_Timer',
+    'b_validate_params'
 ]

byzh_core/Butils/decorator.py

@@ -0,0 +1,15 @@
+def b_validate_params(specs): # specs是@validate_params()传入的内容
+    '''
+    >>> @b_validate_params({
+    >>>     "my_str": {"literal": "happy" or "sad"},
+    >>>     "my_int": {"example": 114514}
+    >>> })
+    >>> def awa(my_str, my_int):
+    >>>     return "qwq"
+    '''
+    def decorator(func): # func是调用修饰器的函数本身
+        def wrapper(*args, **kwargs): # *args, **kwargs是函数本身收到的参数
+            ...
+            return func(*args, **kwargs) # 在此处真正调用函数本身
+        return wrapper
+    return decorator

byzh_core/Butils/timer.py

@@ -1,6 +1,10 @@
 import time
 class B_Timer:
     def __init__(self):
+        '''
+        >>> with B_Timer():
+        >>>     ...
+        '''
         self.start_time = 0
         self.end_time = 0
 

byzh_core/Bwriter/writer.py

@@ -107,4 +107,7 @@ class B_Writer:
         os.makedirs(dir, exist_ok=True)
 
     def __exit__(self):
-        self.closeFile()
+        self.closeFile()
+
+if __name__ == '__main__':
+    B_Writer()

byzh_core/__init__.py

@@ -2,4 +2,4 @@
 # class      以"B_"开头
 # function   以"b_"开头
 
-__version__ = '0.0.2.14'
+__version__ = '0.0.3.0'

byzh_core/__main__.py

@@ -0,0 +1,9 @@
+import argparse
+
+def b_zip():
+    parser = argparse.ArgumentParser()
+    parser.add_argument("dir_path")
+    args = parser.parse_args()
+
+    from .Barchive import b_archive_zip
+    b_archive_zip(args.dir_path)

{byzh_core-0.0.2.14.dist-info → byzh_core-0.0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: byzh_core
-Version: 0.0.2.14
+Version: 0.0.3.0
 Summary: byzh_core是byzh系列的核心库，包含了一些常用的工具函数和类。
 Author: byzh_rc
 License: MIT

byzh_core-0.0.3.0.dist-info/RECORD

@@ -0,0 +1,29 @@
+byzh_core/B_os.py,sha256=f8XuZZOMnP0Fszg6oPLlo9kxX1ytJBK8cEoHqzVkMVA,4313
+byzh_core/__init__.py,sha256=7C6DIxqXyGDta34Hbsf4gdcM9QWGSeROrONoT2IHeCU,110
+byzh_core/__main__.py,sha256=o7-Mn6j1g-VGWhE13goDOMmxKSlXvtrkYVkwt4DrCjo,217
+byzh_core/Barchive/__init__.py,sha256=wUdz646VS0Uhq9lwMkd3YQHCvQhLDqgADoDjFGMqIn0,65
+byzh_core/Barchive/archive.py,sha256=CYuWUScgDEmo1dHwdAZOfW0gSlXlrCTwX3FB_DdMW7I,6598
+byzh_core/Btable/__init__.py,sha256=NlQBjp_mvObEUm4y6wXbGipkNM2N3uNIUidaJkTwFsw,62
+byzh_core/Btable/auto_table.py,sha256=avMeRWydw9L8s3fU0-BSl8a2dCyreIUzW--csm5eR0g,14734
+byzh_core/Bterminal/__init__.py,sha256=B6p6mHVhi5LAHKlvqfrv20E6XNqUsfU2xD753V6-Zf4,83
+byzh_core/Bterminal/cmd.py,sha256=0I-d7VPHv9eYJoSrAZOnEK-R-hyjWVY0hRRn3Zoq1fg,3839
+byzh_core/Btqdm/__init__.py,sha256=G8pvJYizvdTUI3Xw-yELXY7BbjTghCXJLdqUuDoRNZA,141
+byzh_core/Btqdm/my_tqdm.py,sha256=61FeXKrkteKpCvXM5qU2hoMROqFkZW-bs5Rz86KMQSo,3202
+byzh_core/Butils/__init__.py,sha256=_0aAm72oynvpFf-v9EAMSqhWBHbTt4Tn85jtv3Yf0z8,236
+byzh_core/Butils/decorator.py,sha256=HaB8F80wsqQuS30jlI8C2d3SmEa0mBNfJEr5-pwGZj0,584
+byzh_core/Butils/text_style.py,sha256=34u2-Rsur63iNQ0m7tHxpFWqnQZWybSxBx7vnhOhqDo,3346
+byzh_core/Butils/timer.py,sha256=OLxbWWRauYjp1KwavACjoyrqrCciUiIG2ae7qgtZrVA,511
+byzh_core/Bwriter/__init__.py,sha256=QcHzKs0305Pr78ZvZgcC_4miSYZh8jJQDt5TD0GIeEg,82
+byzh_core/Bwriter/writer.py,sha256=ss9EWo-l7czBUlY7Fu0buvQgO5fRspYKtC2X6eaRa0c,3213
+byzh_core/obsolete/__init__.py,sha256=wpfHMPD4awPRDJVYlaiGtkkpjq2srWB7d7LrWhoX5R0,161
+byzh_core/obsolete/auto_table.py,sha256=nC9eHj_IIGVkS2lmN_1gtOyglLCaGlwdoHzPvNNQDEw,13952
+byzh_core/obsolete/row_table.py,sha256=rIX0-Yvb3RnO0RJBkA4m18gux1zYSnEKTy6uQGcWilc,5999
+byzh_core/obsolete/xy_table.py,sha256=-KHK8EUlkJj3Rh0sp_KHI0QFt29AlMASMmR8q8ykUBM,6784
+byzh_core/obsolete/Bconfig/__init__.py,sha256=7lAp7wNetW0AyJcike7ekdY_8wrQQtFjBsi6684t_lU,54
+byzh_core/obsolete/Bconfig/config.py,sha256=_gUNHfW46mTRecYsz_DQn0LENEPifcF3IMHFVE1prgg,10953
+byzh_core-0.0.3.0.dist-info/LICENSE,sha256=-nRwf0Xga4AX5bsWBXXflpDpgX_U23X06oAMcdf0dSY,1089
+byzh_core-0.0.3.0.dist-info/METADATA,sha256=9yFT3-xX-aLQ9YTpIFLIn6vf5Hqw0Ga5jMXSfU1-JPs,371
+byzh_core-0.0.3.0.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+byzh_core-0.0.3.0.dist-info/entry_points.txt,sha256=swk5zKUZn93OW4lUCK8etsh8yPfFsNUZt4LviEEYpvU,50
+byzh_core-0.0.3.0.dist-info/top_level.txt,sha256=Xv-pzvl6kPdIbi5UehQcUdGhLtb8-4WhS5dRK1LINwM,10
+byzh_core-0.0.3.0.dist-info/RECORD,,

byzh_core-0.0.3.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+byzh = byzh_core.__main__:b_zip

byzh_core-0.0.2.14.dist-info/RECORD

@@ -1,26 +0,0 @@
-byzh_core/B_os.py,sha256=-6KjK2NzEeF8GiQno0BmbVDei9DDxOaXdaKGqFeL_eY,2372
-byzh_core/__init__.py,sha256=wvMVJNqpQJ3SUDdlKmDrKGhsaRa4jE6lea3GayPTu2A,111
-byzh_core/Barchive/__init__.py,sha256=wUdz646VS0Uhq9lwMkd3YQHCvQhLDqgADoDjFGMqIn0,65
-byzh_core/Barchive/archive.py,sha256=sG9m1Hr-d2zxYVmzTVRNXX2U2T5baMYECZVqjt0k8pU,3157
-byzh_core/Btable/__init__.py,sha256=NlQBjp_mvObEUm4y6wXbGipkNM2N3uNIUidaJkTwFsw,62
-byzh_core/Btable/auto_table.py,sha256=nc8RvF86laDnITAzX8sVj3l5tbIGflrsstIbSbGDaAI,14747
-byzh_core/Bterminal/__init__.py,sha256=B6p6mHVhi5LAHKlvqfrv20E6XNqUsfU2xD753V6-Zf4,83
-byzh_core/Bterminal/cmd.py,sha256=0I-d7VPHv9eYJoSrAZOnEK-R-hyjWVY0hRRn3Zoq1fg,3839
-byzh_core/Btqdm/__init__.py,sha256=G8pvJYizvdTUI3Xw-yELXY7BbjTghCXJLdqUuDoRNZA,141
-byzh_core/Btqdm/my_tqdm.py,sha256=61FeXKrkteKpCvXM5qU2hoMROqFkZW-bs5Rz86KMQSo,3202
-byzh_core/Butils/__init__.py,sha256=ZtdKTJ8rcJ9zogD4t5pdmWSF1l-TiDaot_C1bvzHQaE,168
-byzh_core/Butils/text_style.py,sha256=34u2-Rsur63iNQ0m7tHxpFWqnQZWybSxBx7vnhOhqDo,3346
-byzh_core/Butils/timer.py,sha256=QrIQ37LALSfx3wUuILLd953-iLDpq_SyqMqrwN6njBs,435
-byzh_core/Bwriter/__init__.py,sha256=QcHzKs0305Pr78ZvZgcC_4miSYZh8jJQDt5TD0GIeEg,82
-byzh_core/Bwriter/writer.py,sha256=P3TDFKhc_o2LgDumsrgtfhHkAczJ1JO1yYrd2NjdlrY,3167
-byzh_core/obsolete/__init__.py,sha256=wpfHMPD4awPRDJVYlaiGtkkpjq2srWB7d7LrWhoX5R0,161
-byzh_core/obsolete/auto_table.py,sha256=nC9eHj_IIGVkS2lmN_1gtOyglLCaGlwdoHzPvNNQDEw,13952
-byzh_core/obsolete/row_table.py,sha256=rIX0-Yvb3RnO0RJBkA4m18gux1zYSnEKTy6uQGcWilc,5999
-byzh_core/obsolete/xy_table.py,sha256=-KHK8EUlkJj3Rh0sp_KHI0QFt29AlMASMmR8q8ykUBM,6784
-byzh_core/obsolete/Bconfig/__init__.py,sha256=7lAp7wNetW0AyJcike7ekdY_8wrQQtFjBsi6684t_lU,54
-byzh_core/obsolete/Bconfig/config.py,sha256=_gUNHfW46mTRecYsz_DQn0LENEPifcF3IMHFVE1prgg,10953
-byzh_core-0.0.2.14.dist-info/LICENSE,sha256=-nRwf0Xga4AX5bsWBXXflpDpgX_U23X06oAMcdf0dSY,1089
-byzh_core-0.0.2.14.dist-info/METADATA,sha256=DbsOAJVoaiSdeAJGZcDoaCChTYqGq5Ekb2r-S80rtwY,372
-byzh_core-0.0.2.14.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-byzh_core-0.0.2.14.dist-info/top_level.txt,sha256=Xv-pzvl6kPdIbi5UehQcUdGhLtb8-4WhS5dRK1LINwM,10
-byzh_core-0.0.2.14.dist-info/RECORD,,