halib 0.1.40__py3-none-any.whl → 0.1.41__py3-none-any.whl

halib/__init__.py

@@ -23,6 +23,10 @@ __all__ = [
     "tcuda",
     "timebudget",
     "tqdm",
+    "rcolor_all_str",
+    "rcolor_palette",
+    "rcolor_str",
+    "rcolor_palette_all"
 ]
 
 import numpy as np
@@ -48,3 +52,4 @@ import arrow
 import omegaconf
 from omegaconf import OmegaConf
 from omegaconf.dictconfig import DictConfig
+from .rich_color import rcolor_str, rcolor_palette, rcolor_palette_all, rcolor_all_str

halib/common.py

@@ -43,7 +43,6 @@ def norm_str(in_str):
     norm_string = norm_string.strip()
     return norm_string
 
-
 def console_rule(msg, do_norm_msg=True, is_end_tag=False):
     msg = norm_str(msg) if do_norm_msg else msg
     if is_end_tag:

halib/rich_color.py

@@ -0,0 +1,259 @@
+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",
+]
+
+def rcolor_all_str():
+    pprint(all_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_all():
+    rcolor_palette(all_colors)

{halib-0.1.40.dist-info → halib-0.1.41.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: halib
-Version: 0.1.40
+Version: 0.1.41
 Summary: Small library for common tasks
 Author: Hoang Van Ha
 Author-email: hoangvanhauit@gmail.com
@@ -42,6 +42,9 @@ Requires-Dist: tqdm
 Requires-Dist: tube-dl
 
 Helper package for coding and automation
+**Version 0.1.41**
++ add <rich_color.py> to display rich color information in <rich> python package (rcolor_str, rcolor_pallet_all, etc.)
+
 **Version 0.1.40**
 
 +  update <csvfile.py> to use `itables` and `pygwalker` to display dataframe in jupyter notebook.

{halib-0.1.40.dist-info → halib-0.1.41.dist-info}/RECORD

@@ -1,5 +1,5 @@
-halib/__init__.py,sha256=fmB0TlY6yyNyKOAFqj2JNp7vTgwc1JD2OYByOZW5cJI,999
-halib/common.py,sha256=TsUHMCTZQ1dYwIDqTHv38NzNNkFCNHNAr92Ghtxfd48,2041
+halib/__init__.py,sha256=KrD3NLEgEE3mWparYxAu2iprtv6Al42zaDmECUpa_dE,1176
+halib/common.py,sha256=28kBN56Ema1A1pIDF2B8BydepXy799L21o2cc01SANc,2039
 halib/csvfile.py,sha256=Eoeni0NIbNG3mB5ESWAvNwhJxOjmCaPd1qqYRHImbvk,1567
 halib/cuda.py,sha256=1bvtBY8QvTWdLaxalzK9wqXPl0Ft3AfhcrebupxGzEA,1010
 halib/dataset.py,sha256=QU0Hr5QFb8_XlvnOMgC9QJGIpwXAZ9lDd0RdQi_QRec,6743
@@ -11,6 +11,7 @@ halib/jsonfile.py,sha256=9XfdFS1wcTdBRmpAGzVu2dVCcJp8RCpsSY16f6KUNts,480
 halib/listop.py,sha256=Vpa8_2fI0wySpB2-8sfTBkyi_A4FhoFVVvFiuvW8N64,339
 halib/plot.py,sha256=4pUJ5GeKQ1LzgiQXbAZlUqSydXSk5tt2_qsw78nSzA4,9445
 halib/projectmake.py,sha256=8dNTxB2jgQlwvovaOnJcCKgHe5vM0nBwzm46Yc8INEI,4048
+halib/rich_color.py,sha256=3048TppVjaJX7P0K3OpJZajH3_wlF6JLFFUKIOeuk3Q,5971
 halib/tele_noti.py,sha256=xL8f889VFR65uVJTj1NNT3qSi3lJ7s_iDjcObwZ0-V0,5926
 halib/textfile.py,sha256=EhVFrit-nRBJx18e6rtIqcE1cSbgsLnMXe_kdhi1EPI,399
 halib/torchloader.py,sha256=-q9YE-AoHZE1xQX2dgNxdqtucEXYs4sQ22WXdl6EGfI,6500
@@ -32,8 +33,8 @@ halib/sys/filesys.py,sha256=ERpnELLDKJoTIIKf-AajgkY62nID4qmqmX5TkE95APU,2931
 halib/system/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 halib/system/cmd.py,sha256=b2x7JPcNnFjLGheIESVYvqAb-w2UwBM1PAwYxMZ5YjA,228
 halib/system/filesys.py,sha256=ERpnELLDKJoTIIKf-AajgkY62nID4qmqmX5TkE95APU,2931
-halib-0.1.40.dist-info/LICENSE.txt,sha256=qZssdna4aETiR8znYsShUjidu-U4jUT9Q-EWNlZ9yBQ,1100
-halib-0.1.40.dist-info/METADATA,sha256=eFwReuUr51KBxzpHNiwcsVJ9rlsvn1lGf7A9ZZ-i3PE,3344
-halib-0.1.40.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-halib-0.1.40.dist-info/top_level.txt,sha256=7AD6PLaQTreE0Fn44mdZsoHBe_Zdd7GUmjsWPyQ7I-k,6
-halib-0.1.40.dist-info/RECORD,,
+halib-0.1.41.dist-info/LICENSE.txt,sha256=qZssdna4aETiR8znYsShUjidu-U4jUT9Q-EWNlZ9yBQ,1100
+halib-0.1.41.dist-info/METADATA,sha256=c9F-wY1MFnD-88XjO2yCwtljfINAwxJlYrfWvbHzNVw,3483
+halib-0.1.41.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+halib-0.1.41.dist-info/top_level.txt,sha256=7AD6PLaQTreE0Fn44mdZsoHBe_Zdd7GUmjsWPyQ7I-k,6
+halib-0.1.41.dist-info/RECORD,,