byzh-core 0.0.2.13__py3-none-any.whl → 0.0.2.15__py3-none-any.whl

byzh_core/B_os.py

@@ -52,6 +52,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 +94,53 @@ def get_filepaths_in_dir(root_dir_path, exclude_name=[], exclude_suffix=['.pyc']
             file_paths.append(file_path)
     return file_paths
 
+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,20 @@
 import os
 import zipfile
+import time
 from pathlib import Path
 
 from ..Btqdm import B_Tqdm
+from .. import B_os
 
 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 +24,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: 黑名单-文件夹名 >>> ['__pycache__', '.git', '.idea']
+    :param black_filenames: 黑名单-文件全名 >>> ['.gitignore', 'README.md']
+    :param black_stems: 黑名单-文件名(不含后缀) >>> ['README']
+    :param black_exts: 黑名单-文件后缀(包括.) >>> ['.csv', '.npy', '.pt', '.pth']
+    :param white_filepath: 白名单-文件全名 >>> ['.gitignore', 'README.md']
+    :param white_dirpath: 白名单-文件夹路径 >>> ['E:\byzh_workingplace\byzh-rc-to-pypi\big_work\文件夹1']
     :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/Bterminal/cmd.py

@@ -3,7 +3,7 @@ from pathlib import Path
 import subprocess
 import time
 
-from ..Bbasic import B_Color
+from ..Butils import B_Color
 
 
 def args_process(args: tuple) -> list:

byzh_core/Btqdm/my_tqdm.py

@@ -2,7 +2,7 @@ import sys
 import time
 import threading
 
-from ..Bbasic import B_Color, B_Appearance, B_Background
+from ..Butils import B_Color, B_Appearance, B_Background
 
 
 class B_Tqdm:

byzh_core/Butils/__init__.py

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

byzh_core/Butils/timer.py

@@ -0,0 +1,17 @@
+import time
+class B_Timer:
+    def __init__(self):
+        self.start_time = 0
+        self.end_time = 0
+
+    def __enter__(self):
+        self.start_time = time.time()
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.end_time = time.time()
+        print(f"[Timer] cost: {self.end_time - self.start_time:.4f}s")
+
+if __name__ == '__main__':
+    with B_Timer():
+        time.sleep(2)

byzh_core/Bwriter/writer.py

@@ -3,7 +3,7 @@ from typing import Literal
 import time
 import os
 
-from ..Bbasic import B_Color
+from ..Butils import B_Color
 
 def color_wrap(string:str, color:B_Color):
     return color.value + string + B_Color.RESET.value

byzh_core/__init__.py

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

byzh_core/__main__.py

@@ -0,0 +1,10 @@
+import argparse
+
+def main():
+    parser = argparse.ArgumentParser(description="打个招呼的命令行工具")
+    parser.add_argument("name", help="你的名字")
+    parser.add_argument("-t", "--times", type=int, default=1, help="打招呼的次数")
+    args = parser.parse_args()
+
+    for _ in range(args.times):
+        print(f"你好，{args.name}！")

byzh_core/obsolete/Bconfig/config.py

@@ -1,4 +1,4 @@
-from ...Bbasic import B_Color
+from ...Butils import B_Color
 
 
 class B_Config:

{byzh_core-0.0.2.13.dist-info → byzh_core-0.0.2.15.dist-info}/METADATA

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

byzh_core-0.0.2.15.dist-info/RECORD

@@ -0,0 +1,28 @@
+byzh_core/B_os.py,sha256=dMQvaeyp5biW8GfLCc6ivfpPshkHJaR3vxc-ef3nuZU,4022
+byzh_core/__init__.py,sha256=63yKymaZaQiHS18yNbasNXZT9eraqiuINSxUV7tbIKQ,111
+byzh_core/__main__.py,sha256=0CXIQY5vma1qV2ykdCGH_9MXukMBFOkA1Y84bVb0LO0,368
+byzh_core/Barchive/__init__.py,sha256=wUdz646VS0Uhq9lwMkd3YQHCvQhLDqgADoDjFGMqIn0,65
+byzh_core/Barchive/archive.py,sha256=LGCHt92wP3hg26dpbPt9IbLByXKgcRGwB_IVCOhSFC8,6351
+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.15.dist-info/LICENSE,sha256=-nRwf0Xga4AX5bsWBXXflpDpgX_U23X06oAMcdf0dSY,1089
+byzh_core-0.0.2.15.dist-info/METADATA,sha256=wDqDKTawSgTTHhfYtbmc99CtPSbHCgx8ujVzaKVMKPo,372
+byzh_core-0.0.2.15.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+byzh_core-0.0.2.15.dist-info/entry_points.txt,sha256=n9TtHWWOcdcCVFqF4FTXLj2KKGHpvjifBDRoufTrgN0,49
+byzh_core-0.0.2.15.dist-info/top_level.txt,sha256=Xv-pzvl6kPdIbi5UehQcUdGhLtb8-4WhS5dRK1LINwM,10
+byzh_core-0.0.2.15.dist-info/RECORD,,

byzh_core-0.0.2.15.dist-info/entry_points.txt

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

byzh_core/Bbasic/__init__.py

@@ -1,3 +0,0 @@
-from .text_style import B_Color, B_Appearance, B_Background
-
-__all__ = ['B_Color', 'B_Appearance', 'B_Background']

byzh_core-0.0.2.13.dist-info/RECORD

@@ -1,25 +0,0 @@
-byzh_core/B_os.py,sha256=-6KjK2NzEeF8GiQno0BmbVDei9DDxOaXdaKGqFeL_eY,2372
-byzh_core/__init__.py,sha256=3MQSx_FHA9m2ns8iWJvgFa0fodxgXK2zeOmKii0WeIU,111
-byzh_core/Barchive/__init__.py,sha256=wUdz646VS0Uhq9lwMkd3YQHCvQhLDqgADoDjFGMqIn0,65
-byzh_core/Barchive/archive.py,sha256=sG9m1Hr-d2zxYVmzTVRNXX2U2T5baMYECZVqjt0k8pU,3157
-byzh_core/Bbasic/__init__.py,sha256=wr7Y7YJINhgpV5X6BT5DaGJKcHUOZYNerCGXoi2Egac,116
-byzh_core/Bbasic/text_style.py,sha256=34u2-Rsur63iNQ0m7tHxpFWqnQZWybSxBx7vnhOhqDo,3346
-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=iK0fwI0D3peEwUYYZWwnl9X8ImFfrBsEq6ySd3DcRdE,3839
-byzh_core/Btqdm/__init__.py,sha256=G8pvJYizvdTUI3Xw-yELXY7BbjTghCXJLdqUuDoRNZA,141
-byzh_core/Btqdm/my_tqdm.py,sha256=OIlT01P6g-rDyDt6wlX9lORAUux9x-LA6EZRGQyJaXU,3202
-byzh_core/Bwriter/__init__.py,sha256=QcHzKs0305Pr78ZvZgcC_4miSYZh8jJQDt5TD0GIeEg,82
-byzh_core/Bwriter/writer.py,sha256=kSfIyGdx-XGoV1TBUY6f7NoFL_40fTx9ISTT8Fic0yE,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=TD8CC1P2Zu95Tf8AGK2IHRXjrCCMBA_tdoZsc1Rr0Gk,10953
-byzh_core-0.0.2.13.dist-info/LICENSE,sha256=-nRwf0Xga4AX5bsWBXXflpDpgX_U23X06oAMcdf0dSY,1089
-byzh_core-0.0.2.13.dist-info/METADATA,sha256=3zXlKKCBBEzdMdoI6-WVmgqsnZ7UqAaB5lzqjjy1ZqM,372
-byzh_core-0.0.2.13.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-byzh_core-0.0.2.13.dist-info/top_level.txt,sha256=Xv-pzvl6kPdIbi5UehQcUdGhLtb8-4WhS5dRK1LINwM,10
-byzh_core-0.0.2.13.dist-info/RECORD,,