byzh-core 0.0.2.13__tar.gz → 0.0.2.15__tar.gz

{byzh_core-0.0.2.13 → byzh_core-0.0.2.15}/PKG-INFO

@@ -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.13 → byzh_core-0.0.2.15}/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-0.0.2.15/byzh_core/Barchive/archive.py

@@ -0,0 +1,149 @@
+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,
+    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,
+):
+    '''
+    压缩文件夹，排除 指定文件夹and指定后缀文件
+
+    name代表前缀, ext代表后缀(包括.)
+
+    :param source_path:
+    :param output_path: 如果不传入, 则默认为source_path同名同路径的zip文件
+    :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
+    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='[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 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)
+
+        # 白名单-文件夹:
+        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'根文件夹',
+        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-0.0.2.13 → byzh_core-0.0.2.15}/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-0.0.2.13 → byzh_core-0.0.2.15}/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-0.0.2.15/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-0.0.2.15/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-0.0.2.13 → byzh_core-0.0.2.15}/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-0.0.2.13 → byzh_core-0.0.2.15}/byzh_core/__init__.py

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

byzh_core-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-0.0.2.13 → byzh_core-0.0.2.15}/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 → byzh_core-0.0.2.15}/byzh_core.egg-info/PKG-INFO

@@ -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.13 → byzh_core-0.0.2.15}/byzh_core.egg-info/SOURCES.txt

@@ -3,21 +3,24 @@ README.md
 setup.py
 byzh_core/B_os.py
 byzh_core/__init__.py
+byzh_core/__main__.py
 byzh_core.egg-info/PKG-INFO
 byzh_core.egg-info/SOURCES.txt
 byzh_core.egg-info/dependency_links.txt
+byzh_core.egg-info/entry_points.txt
 byzh_core.egg-info/requires.txt
 byzh_core.egg-info/top_level.txt
 byzh_core/Barchive/__init__.py
 byzh_core/Barchive/archive.py
-byzh_core/Bbasic/__init__.py
-byzh_core/Bbasic/text_style.py
 byzh_core/Btable/__init__.py
 byzh_core/Btable/auto_table.py
 byzh_core/Bterminal/__init__.py
 byzh_core/Bterminal/cmd.py
 byzh_core/Btqdm/__init__.py
 byzh_core/Btqdm/my_tqdm.py
+byzh_core/Butils/__init__.py
+byzh_core/Butils/text_style.py
+byzh_core/Butils/timer.py
 byzh_core/Bwriter/__init__.py
 byzh_core/Bwriter/writer.py
 byzh_core/obsolete/__init__.py

byzh_core-0.0.2.15/byzh_core.egg-info/entry_points.txt

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

{byzh_core-0.0.2.13 → byzh_core-0.0.2.15}/setup.py

@@ -12,4 +12,9 @@ setup(
     install_requires=[
         'wcwidth',
     ],
+    entry_points={
+        "console_scripts": [
+            "byzh=byzh_core.__main__:main"
+        ]
+    },
 )

byzh_core-0.0.2.13/byzh_core/Barchive/archive.py

@@ -1,80 +0,0 @@
-import os
-import zipfile
-from pathlib import Path
-
-from ..Btqdm import B_Tqdm
-
-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,
-    contain_empty_folder:bool=True,
-):
-    '''
-    压缩文件夹，排除 指定文件夹and指定后缀文件
-
-    name代表前缀, ext代表后缀(包括.)
-
-    :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 contain_empty_folder: 是否包含空文件夹
-    :return:
-    '''
-    output_path = source_path + '.zip' if output_path is None else 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)
-
-
-    my_tqdm = B_Tqdm(prefix='Archive')
-
-    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 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)
-
-                # 若是空文件夹，则压缩文件夹:
-                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)
-                    my_tqdm.update(1)
-
-if __name__ == '__main__':
-    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']
-    )

byzh_core-0.0.2.13/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']