halib 0.2.30__py3-none-any.whl

halib/research/wandb_op.py

@@ -0,0 +1,116 @@
+import glob
+from rich.pretty import pprint
+import os
+import subprocess
+import argparse
+import wandb
+from tqdm import tqdm
+from rich.console import Console
+console = Console()
+
+def sync_runs(outdir):
+    outdir = os.path.abspath(outdir)
+    assert os.path.exists(outdir), f"Output directory {outdir} does not exist."
+    sub_dirs = [name for name in os.listdir(outdir) if os.path.isdir(os.path.join(outdir, name))]
+    assert len(sub_dirs) > 0, f"No subdirectories found in {outdir}."
+    console.rule("Parent Directory")
+    console.print(f"[yellow]{outdir}[/yellow]")
+
+    exp_dirs = [os.path.join(outdir, sub_dir) for sub_dir in sub_dirs]
+    wandb_dirs = []
+    for exp_dir in exp_dirs:
+        wandb_dirs.extend(glob.glob(f"{exp_dir}/wandb/*run-*"))
+    if len(wandb_dirs) == 0:
+        console.print(f"No wandb runs found in {outdir}.")
+        return
+    else:
+        console.print(f"Found [bold]{len(wandb_dirs)}[/bold] wandb runs in {outdir}.")
+        for i, wandb_dir in enumerate(wandb_dirs):
+            console.rule(f"Syncing wandb run {i + 1}/{len(wandb_dirs)}")
+            console.print(f"Syncing: {wandb_dir}")
+            process = subprocess.Popen(
+                ["wandb", "sync", wandb_dir],
+                stdout=subprocess.PIPE,
+                stderr=subprocess.STDOUT,
+                text=True,
+            )
+
+            for line in process.stdout:
+                console.print(line.strip())
+                if " ERROR Error while calling W&B API" in line:
+                    break
+            process.stdout.close()
+            process.wait()
+            if process.returncode != 0:
+                console.print(f"[red]Error syncing {wandb_dir}. Return code: {process.returncode}[/red]")
+            else:
+                console.print(f"Successfully synced {wandb_dir}.")
+
+def delete_runs(project, pattern=None):
+    console.rule("Delete W&B Runs")
+    confirm_msg = f"Are you sure you want to delete all runs in"
+    confirm_msg += f" \n\tproject: [red]{project}[/red]"
+    if pattern:
+        confirm_msg += f"\n\tpattern: [blue]{pattern}[/blue]"
+
+    console.print(confirm_msg)
+    confirmation = input(f"This action cannot be undone. [y/N]: ").strip().lower()
+    if confirmation != "y":
+        print("Cancelled.")
+        return
+
+    print("Confirmed. Proceeding...")
+    api = wandb.Api()
+    runs = api.runs(project)
+
+    deleted = 0
+    console.rule("Deleting W&B Runs")
+    if len(runs) == 0:
+        print("No runs found in the project.")
+        return
+    for run in tqdm(runs):
+        if pattern is None or pattern in run.name:
+            run.delete()
+            console.print(f"Deleted run: [red]{run.name}[/red]")
+            deleted += 1
+
+    console.print(f"Total runs deleted: {deleted}")
+
+
+def valid_argument(args):
+    if args.op == "sync":
+        assert os.path.exists(args.outdir), f"Output directory {args.outdir} does not exist."
+    elif args.op == "delete":
+        assert isinstance(args.project, str) and len(args.project.strip()) > 0, "Project name must be a non-empty string."
+    else:
+        raise ValueError(f"Unknown operation: {args.op}")
+
+def parse_args():
+    parser = argparse.ArgumentParser(description="Operations on W&B runs")
+    parser.add_argument("-op", "--op", type=str, help="Operation to perform", default="sync", choices=["delete", "sync"])
+    parser.add_argument("-prj", "--project", type=str, default="fire-paper2-2025", help="W&B project name")
+    parser.add_argument("-outdir", "--outdir", type=str, help="arg1 description", default="./zout/train")
+    parser.add_argument("-pt", "--pattern",
+        type=str,
+        default=None,
+        help="Run name pattern to match for deletion",
+    )
+
+    return parser.parse_args()
+
+
+def main():
+    args = parse_args()
+    # Validate arguments, stop if invalid
+    valid_argument(args)
+
+    op = args.op
+    if op == "sync":
+        sync_runs(args.outdir)
+    elif op == "delete":
+        delete_runs(args.project, args.pattern)
+    else:
+        raise ValueError(f"Unknown operation: {op}")
+
+if __name__ == "__main__":
+    main()

halib/rich_color.py

@@ -0,0 +1,285 @@
+from rich.console import Console
+from rich.pretty import pprint
+from rich.table import Table
+from rich.text import Text
+from rich.panel import Panel
+
+# List of colors
+# ! https://rich.readthedocs.io/en/stable/appendix/colors.html
+all_colors = [
+    "black",
+    "red",
+    "green",
+    "yellow",
+    "blue",
+    "magenta",
+    "cyan",
+    "white",
+    "bright_black",
+    "bright_red",
+    "bright_green",
+    "bright_yellow",
+    "bright_blue",
+    "bright_magenta",
+    "bright_cyan",
+    "bright_white",
+    "grey0",
+    "navy_blue",
+    "dark_blue",
+    "blue3",
+    "blue1",
+    "dark_green",
+    "deep_sky_blue4",
+    "dodger_blue3",
+    "dodger_blue2",
+    "green4",
+    "spring_green4",
+    "turquoise4",
+    "deep_sky_blue3",
+    "dodger_blue1",
+    "dark_cyan",
+    "light_sea_green",
+    "deep_sky_blue2",
+    "deep_sky_blue1",
+    "green3",
+    "spring_green3",
+    "cyan3",
+    "dark_turquoise",
+    "turquoise2",
+    "green1",
+    "spring_green2",
+    "spring_green1",
+    "medium_spring_green",
+    "cyan2",
+    "cyan1",
+    "purple4",
+    "purple3",
+    "blue_violet",
+    "grey37",
+    "medium_purple4",
+    "slate_blue3",
+    "royal_blue1",
+    "chartreuse4",
+    "pale_turquoise4",
+    "steel_blue",
+    "steel_blue3",
+    "cornflower_blue",
+    "dark_sea_green4",
+    "cadet_blue",
+    "sky_blue3",
+    "chartreuse3",
+    "sea_green3",
+    "aquamarine3",
+    "medium_turquoise",
+    "steel_blue1",
+    "sea_green2",
+    "sea_green1",
+    "dark_slate_gray2",
+    "dark_red",
+    "dark_magenta",
+    "orange4",
+    "light_pink4",
+    "plum4",
+    "medium_purple3",
+    "slate_blue1",
+    "wheat4",
+    "grey53",
+    "light_slate_grey",
+    "medium_purple",
+    "light_slate_blue",
+    "yellow4",
+    "dark_sea_green",
+    "light_sky_blue3",
+    "sky_blue2",
+    "chartreuse2",
+    "pale_green3",
+    "dark_slate_gray3",
+    "sky_blue1",
+    "chartreuse1",
+    "light_green",
+    "aquamarine1",
+    "dark_slate_gray1",
+    "deep_pink4",
+    "medium_violet_red",
+    "dark_violet",
+    "purple",
+    "medium_orchid3",
+    "medium_orchid",
+    "dark_goldenrod",
+    "rosy_brown",
+    "grey63",
+    "medium_purple2",
+    "medium_purple1",
+    "dark_khaki",
+    "navajo_white3",
+    "grey69",
+    "light_steel_blue3",
+    "light_steel_blue",
+    "dark_olive_green3",
+    "dark_sea_green3",
+    "light_cyan3",
+    "light_sky_blue1",
+    "green_yellow",
+    "dark_olive_green2",
+    "pale_green1",
+    "dark_sea_green2",
+    "pale_turquoise1",
+    "red3",
+    "deep_pink3",
+    "magenta3",
+    "dark_orange3",
+    "indian_red",
+    "hot_pink3",
+    "hot_pink2",
+    "orchid",
+    "orange3",
+    "light_salmon3",
+    "light_pink3",
+    "pink3",
+    "plum3",
+    "violet",
+    "gold3",
+    "light_goldenrod3",
+    "tan",
+    "misty_rose3",
+    "thistle3",
+    "plum2",
+    "yellow3",
+    "khaki3",
+    "light_yellow3",
+    "grey84",
+    "light_steel_blue1",
+    "yellow2",
+    "dark_olive_green1",
+    "dark_sea_green1",
+    "honeydew2",
+    "light_cyan1",
+    "red1",
+    "deep_pink2",
+    "deep_pink1",
+    "magenta2",
+    "magenta1",
+    "orange_red1",
+    "indian_red1",
+    "hot_pink",
+    "medium_orchid1",
+    "dark_orange",
+    "salmon1",
+    "light_coral",
+    "pale_violet_red1",
+    "orchid2",
+    "orchid1",
+    "orange1",
+    "sandy_brown",
+    "light_salmon1",
+    "light_pink1",
+    "pink1",
+    "plum1",
+    "gold1",
+    "light_goldenrod2",
+    "navajo_white1",
+    "misty_rose1",
+    "thistle1",
+    "yellow1",
+    "light_goldenrod1",
+    "khaki1",
+    "wheat1",
+    "cornsilk1",
+    "grey100",
+    "grey3",
+    "grey7",
+    "grey11",
+    "grey15",
+    "grey19",
+    "grey23",
+    "grey27",
+    "grey30",
+    "grey35",
+    "grey39",
+    "grey42",
+    "grey46",
+    "grey50",
+    "grey54",
+    "grey58",
+    "grey62",
+    "grey66",
+    "grey70",
+    "grey74",
+    "grey78",
+    "grey82",
+    "grey85",
+    "grey89",
+    "grey93",
+]
+
+basic_colors = [
+    "black",
+    "red",
+    "green",
+    "yellow",
+    "blue",
+    "magenta",
+    "cyan",
+    "white",
+    "bright_black",
+    "bright_red",
+    "bright_green",
+    "bright_yellow",
+    "bright_blue",
+    "bright_magenta",
+    "bright_cyan",
+    "bright_white",
+]
+
+def rcolor_all_str():
+    pprint(all_colors)
+
+def rcolor_basic_str():
+    pprint(basic_colors)
+
+def rcolor_str(in_str, color="white"):
+    assert color in all_colors, f"color must be one of {all_colors}"
+    return f"[{color}]{in_str}[/{color}]"
+
+def rcolor_palette(color_list):
+    # make sure all colors are valid (in all_colors)
+    for color in color_list:
+        assert (
+            color in all_colors
+        ), f"color must be a valid color. call <rcolor_all_str()> or <rcolor_palette_all()> to see all valid colors"
+    # Initialize console
+    console = Console()
+
+    # Create a table with horizontal lines and six columns
+    table = Table(show_header=True, header_style="bold magenta", show_lines=True)
+
+    # Define the columns
+    table.add_column("Color Name 1", style="bold")
+    table.add_column("Sample 1", style="bold")
+    table.add_column("Color Name 2", style="bold")
+    table.add_column("Sample 2", style="bold")
+    table.add_column("Color Name 3", style="bold")
+    table.add_column("Sample 3", style="bold")
+
+    # Adjust the number of rows needed for the table
+    num_colors = len(color_list)
+    num_rows = (num_colors + 2) // 3  # Ceiling division to ensure all colors fit
+
+    # Add rows to the table
+    for i in range(num_rows):
+        color1 = color_list[i * 3] if i * 3 < num_colors else ""
+        color2 = color_list[i * 3 + 1] if i * 3 + 1 < num_colors else ""
+        color3 = color_list[i * 3 + 2] if i * 3 + 2 < num_colors else ""
+        filled_rect1 = Text(" " * 10, style=f"on {color1}") if color1 else ""
+        filled_rect2 = Text(" " * 10, style=f"on {color2}") if color2 else ""
+        filled_rect3 = Text(" " * 10, style=f"on {color3}") if color3 else ""
+        table.add_row(color1, filled_rect1, color2, filled_rect2, color3, filled_rect3)
+
+    # Print the table
+    console.print(table)
+
+def rcolor_palette_basic():
+    rcolor_palette(basic_colors)
+
+def rcolor_palette_all():
+    rcolor_palette(all_colors)

halib/sys/cmd.py

@@ -0,0 +1,8 @@
+import subprocess
+
+
+# @link: https://geekflare.com/learn-python-subprocess/
+# subprocess.run(['ls', '-la'])
+def run(args):
+    process = subprocess.run(args, capture_output=True, text=True)
+    return process.returncode

halib/sys/filesys.py

@@ -0,0 +1,124 @@
+import glob
+import os
+import shutil
+from distutils.dir_util import copy_tree
+
+
+def is_exist(path):
+    return os.path.exists(path)
+
+
+def is_directory(path):
+    return os.path.isdir(path)
+
+
+def get_current_dir():
+    return os.getcwd()
+
+
+def change_current_dir(new_dir):
+    if is_directory(new_dir):
+        os.chdir(new_dir)
+
+
+def get_dir_name(directory):
+    return os.path.basename(os.path.normpath(directory))
+
+
+def get_parent_dir(directory, return_full_path=False):
+    if not return_full_path:
+        return os.path.basename(os.path.dirname(directory))
+    else:
+        return os.path.dirname(directory)
+
+
+def make_dir(directory):
+    if not os.path.exists(directory):
+        os.makedirs(directory)
+
+
+def copy_dir(src_dir, dst_dir, dirs_exist_ok=True, ignore_patterns=None):
+    shutil.copytree(
+        src_dir, dst_dir, dirs_exist_ok=dirs_exist_ok, ignore=ignore_patterns
+    )
+
+
+def delete_dir(directory):
+    shutil.rmtree(directory)
+
+
+def list_dirs(directory):
+    folders = list(
+        filter(
+            lambda x: os.path.isdir(os.path.join(directory, x)), os.listdir(directory)
+        )
+    )
+    return folders
+
+
+def list_files(directory):
+    files = list(
+        filter(
+            lambda x: os.path.isfile(os.path.join(directory, x)), os.listdir(directory)
+        )
+    )
+    return files
+
+
+def filter_files_by_extension(directory, ext, recursive=True):
+    if is_directory(directory):
+        result_files = []
+        if isinstance(ext, list):
+            ext_list = ext
+        else:
+            ext_list = [ext]
+        if not recursive:
+            filter_pattern = f"{directory}/*"
+        else:
+            filter_pattern = f"{directory}/**/*"
+
+        for ext_item in ext_list:
+            ext_filter = f"{filter_pattern}.{ext_item}"
+            files = glob.glob(filter_pattern, recursive=True)
+            files = [f for f in files if is_file(f) and f.endswith(ext_item)]
+            result_files.extend(files)
+        return result_files
+    else:
+        raise OSError("Directory not exists")
+
+
+def is_file(path):
+    return os.path.isfile(path)
+
+
+def get_file_name(file_path, split_file_ext=False):
+    if is_file(file_path):
+        if split_file_ext:
+            filename, file_extension = os.path.splitext(os.path.basename(file_path))
+            return filename, file_extension
+        else:
+            return os.path.basename(file_path)
+    else:
+        raise OSError("Not a file")
+
+
+def get_absolute_path(file_path):
+    return os.path.abspath(file_path)
+
+
+# dest can be a directory
+def copy_file(source, dest):
+    shutil.copy2(source, dest)
+
+
+def delete_file(path):
+    if is_file(path):
+        os.remove(path)
+
+
+def rename_dir_or_file(old, new):
+    os.renames(old, new)
+
+
+def move_dir_or_file(source, destination):
+    shutil.move(source, destination)

halib/system/_list_pc.csv

@@ -0,0 +1,6 @@
+pc_name;abbr;working_disk
+DESKTOP-JQD9K01;MainPC;E:
+DESKTOP-5IRHU87;MSI_Laptop;D:
+DESKTOP-96HQCNO;4090_SV;E:
+DESKTOP-Q2IKLC0;4GPU_SV;E:
+DESKTOP-QNS3DNF;1GPU_SV;D:

halib/system/cmd.py

@@ -0,0 +1,8 @@
+import subprocess
+
+
+# @link: https://geekflare.com/learn-python-subprocess/
+# subprocess.run(['ls', '-la'])
+def run(args):
+    process = subprocess.run(args, capture_output=True, text=True)
+    return process.returncode