textfmt 0.2.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

{textfmt-0.2.0.dist-info → textfmt-0.2.1.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: textfmt
-Version: 0.2.0
-Summary: A Python module for terminal text styling without the overhead of rich
+Version: 0.2.1
+Summary: Terminal Formatting for Humans
 Author-email: Rihaan Meher <meherrihaan@gmail.com>
 License: MIT
 Project-URL: Homepage, https://github.com/sharktide/textformat
@@ -55,6 +55,28 @@ data = [
 print(TableFormatter.generate(headers, data))
 ```
 
+File Download Progress Bars
+
+```python
+from textformat.progress import Color, DownloadProgressBar
+from pathlib import Path
+def download_with_progress(url: str, output_path: Path, download_color, complete_color):
+    response = requests.get(url, stream=True)
+    total = int(response.headers.get('content-length', 0))
+
+    if response.status_code != 200:
+        raise Exception(f"Failed to download from {url}. HTTP status: {response.status_code}")
+
+    progress = DownloadProgressBar(total, prefix=output_path.name, 
+                           download_color=download_color, complete_color=complete_color)
+    with open(output_path, "wb") as f:
+        for chunk in response.iter_content(chunk_size=8192):
+            if chunk:
+                f.write(chunk)
+                progress.update(progress.downloaded + len(chunk))
+    print()
+    download_with_progress('https://huggingface.co/sharktide/recyclebot0/resolve/main/tf_model.h5', Path("tf_model.h5"), Colors.CYAN, Colors.GREEN)
+```
 ## License
 
 MIT License – Free to use and modify.

textfmt-0.2.1.dist-info/RECORD

@@ -0,0 +1,10 @@
+textfmt-0.2.1.dist-info/licenses/LICENSE,sha256=Msmy46KiAR2wmzeCe3kSl-wEt7ptbTZOAzdwUUyFq_M,1090
+textformat/__init__.py,sha256=Yz9706rsciMHmMHDvzqS1LxQm04N9NM5rO-xUaFLm8c,306
+textformat/progress.py,sha256=2EXWQxEVjnd9eHmEJUGmH6e3TVqXC5ZNcS-ez6L-W-A,3444
+textformat/table.py,sha256=W51T9__0Qtgv_nxn6s2bit4tduhgiXZ3FwjODGvLf3I,746
+textformat/textformat.py,sha256=hYGgLkra7HIdG5mtVcDvAHOMzbv-0q9TFs82jjdOdew,1681
+textformat/word_art.py,sha256=GsJka0ukV23FTvhuyuGRChhNGOjTWOdKBK3h0e-LnX8,318
+textfmt-0.2.1.dist-info/METADATA,sha256=BserKxfvwaxtgi0GjyNnAseG1XfHSys2VtH9tdd9sMA,2539
+textfmt-0.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+textfmt-0.2.1.dist-info/top_level.txt,sha256=Nf6G1sI6EqA2opfIdIio6ZqKNbpT2EtYf5cd0sKgPRo,11
+textfmt-0.2.1.dist-info/RECORD,,

textformat/__init__.py

@@ -1,5 +1,9 @@
 # textfmt/__init__.py
-from .textformat import TextFormat, bcolors
+from .textformat import TextFormat, bcolors, enable_windows_ansi_support
 from .word_art import WordArt
 from .table import TableFormatter
-__all__ = ["TextFormat", "bcolors", "WordArt", "TableFormatter"]
+import os
+
+enable_windows_ansi_support()
+
+__all__ = ["TextFormat", "bcolors", "WordArt", "TableFormatter", "enable_windows_ansi_support"]

textformat/progress.py

@@ -20,28 +20,14 @@ class Colors:
     LIGHT_MAGENTA = "\033[95m"
     LIGHT_CYAN = "\033[96m"
 
-    # List of all available colors for easy access
     ALL_COLORS = [
         GREEN, BLUE, RED, YELLOW, MAGENTA, CYAN, WHITE,
         LIGHT_GREEN, LIGHT_BLUE, LIGHT_RED, LIGHT_YELLOW, LIGHT_MAGENTA, LIGHT_CYAN
     ]
 
-class ProgressBar:
-    def __init__(self, total, prefix='', length=40, fill='━', empty=' ', print_end="\r", 
+class DownloadProgressBar:
+    def __init__(self, total, prefix='', length=40, fill='━', empty=' ', print_end="\r",
                  download_color=Colors.GREEN, complete_color=Colors.BLUE):
-        """
-        Initializes the progress bar.
-
-        Parameters:
-            total (int): Total number of bytes to download.
-            prefix (str): Prefix for the progress bar.
-            length (int): Length of the progress bar.
-            fill (str): Character to fill the progress bar.
-            empty (str): Character to represent empty space in the progress bar.
-            print_end (str): Ending character for printing.
-            download_color (str): Color code for progress bar during download.
-            complete_color (str): Color code for progress bar when download is complete.
-        """
         self.total = total
         self.prefix = prefix
         self.length = length
@@ -54,33 +40,59 @@ class ProgressBar:
         self.downloaded = 0
 
     def update(self, downloaded):
+        now = time.time()
         self.downloaded = downloaded
-        percent = ("{0:.1f}").format(100 * (self.downloaded / float(self.total)))
-        filled_length = int(self.length * self.downloaded // self.total)
-        bar = self.fill * filled_length + self.empty * (self.length - filled_length)
 
-        # Calculate current file size and total file size
+        if self.total > 0:
+            percent = 100 * (self.downloaded / float(self.total))
+            filled_length = int(self.length * self.downloaded // self.total)
+            total_size = self.format_size(self.total)
+        else:
+            percent = 100
+            filled_length = 40
+            total_size = "?"
+
+        bar = self.fill * filled_length + self.empty * (self.length - filled_length)
         current_size = self.format_size(self.downloaded)
-        total_size = self.format_size(self.total)
 
-        # Set color based on progress
-        if self.downloaded == self.total:
-            color = self.complete_color  # Completed color
-        else:
-            color = self.download_color  # Downloading color
+        # Speed (bytes per second)
+        elapsed = now - self.start_time
+        speed_bps = self.downloaded / elapsed if elapsed > 0 else 0
+        speed_str = self.format_size(speed_bps) + "/s" if speed_bps > 0 else "?"
+
+        # ETA
+        remaining = self.total - self.downloaded if self.total > 0 else 0
+        eta_seconds = remaining / speed_bps if self.total > 0 and speed_bps > 0 else 0
+        eta_str = self.format_time(eta_seconds) if self.total > 0 else "?"
+
+        # Choose color
+        color = self.complete_color if self.downloaded == self.total and self.total > 0 else self.download_color
 
-        # Print the progress bar with color, percentage, and file size
-        sys.stdout.write(f'\r{self.prefix} {color}{bar}{Colors.RESET} {percent}% • {current_size}/{total_size}')
+        # Print
+        sys.stdout.write(
+            f'\r{self.prefix} {color}{bar}{Colors.RESET} {percent:5.1f}% • '
+            f'{current_size}/{total_size} • {speed_str} • {eta_str}'
+        )
         sys.stdout.flush()
 
-        if self.downloaded == self.total:
+        if self.downloaded == self.total and self.total > 0:
             sys.stdout.write(self.print_end)
             sys.stdout.flush()
 
-    def format_size(self, size_in_bytes):
-        """Format the size in a human-readable format (e.g., KB, MB, GB)."""
+    @staticmethod
+    def format_size(size_in_bytes):
+        """Format the size in a human-readable format."""
         for unit in ['B', 'KiB', 'MiB', 'GiB', 'TiB']:
             if size_in_bytes < 1024:
                 return f"{size_in_bytes:.1f} {unit}"
             size_in_bytes /= 1024
+        return f"{size_in_bytes:.1f} PiB"
 
+    @staticmethod
+    def format_time(seconds):
+        """Format seconds into H:MM:SS or M:SS."""
+        if seconds < 0.1:
+            return "0:00"
+        m, s = divmod(int(seconds + 0.5), 60)
+        h, m = divmod(m, 60)
+        return f"{h}:{m:02}:{s:02}" if h else f"{m}:{s:02}"

textformat/table.py

@@ -4,10 +4,15 @@ class TableFormatter:
     @staticmethod
     def generate(headers, data):
         """ Creates a formatted table for structured output """
+        # Calculate max width per column
         col_widths = [max(len(str(item)) for item in col) for col in zip(headers, *data)]
-        border = "+".join("-" * (w + 2) for w in col_widths)
+
+        def build_border():
+            return "+" + "+".join("-" * (w + 2) for w in col_widths) + "+"
 
         def format_row(row):
             return "| " + " | ".join(f"{str(item).ljust(w)}" for item, w in zip(row, col_widths)) + " |"
 
-        return f"{border}\n{format_row(headers)}\n{border}\n" + "\n".join(format_row(row) for row in data) + f"\n{border}"
+        border = build_border()
+        rows = [format_row(headers), border] + [format_row(row) for row in data]
+        return f"{border}\n" + "\n".join(rows) + f"\n{border}"

textformat/textformat.py

@@ -1,3 +1,21 @@
+import os
+
+def enable_windows_ansi_support() -> None:
+    """
+    Enables ANSI escape sequence support in Windows terminals (safe version).
+    """
+    if os.name == 'nt':
+        try:
+            import ctypes
+            kernel32 = ctypes.windll.kernel32
+            handle = kernel32.GetStdHandle(-11)
+            mode = ctypes.c_ulong()
+            if kernel32.GetConsoleMode(handle, ctypes.byref(mode)):
+                mode.value |= 0x0004
+                kernel32.SetConsoleMode(handle, mode)
+        except Exception:
+            pass
+
 class bcolors:
     HEADER = '\033[95m'
     OKBLUE = '\033[94m'
@@ -9,7 +27,6 @@ class bcolors:
     ENDC = '\033[0m'
     BOLD = '\033[1m'
     UNDERLINE = '\033[4m'
-
 class TextFormat:
     """ ANSI Escape Code Formatting for Python CLI Output """
     RESET = "\x1b[0m"

textfmt-0.2.0.dist-info/RECORD

@@ -1,10 +0,0 @@
-textfmt-0.2.0.dist-info/licenses/LICENSE,sha256=Msmy46KiAR2wmzeCe3kSl-wEt7ptbTZOAzdwUUyFq_M,1090
-textformat/__init__.py,sha256=fyib7AnpkuEZ_PAzx6AO5SVKNTB5kyqk5NnJBXx7TYw,200
-textformat/progress.py,sha256=9s0pndMjZ-S5dgcTZuibgAoB-oKeAAWvsjHqxij3Rtw,3253
-textformat/table.py,sha256=GA7MsMWNuug61XxK9Ot3SOg5bl9zh5GO9Hae8-Y2qeE,604
-textformat/textformat.py,sha256=yz6KstoDtrIgoA_b4ZFJAY9EfcylSjeeJ_euhPf8cpw,1121
-textformat/word_art.py,sha256=GsJka0ukV23FTvhuyuGRChhNGOjTWOdKBK3h0e-LnX8,318
-textfmt-0.2.0.dist-info/METADATA,sha256=TABcMirGCQYbNUoj0bp6jViVQKbAUQl0zjyE9UxcxmU,1567
-textfmt-0.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-textfmt-0.2.0.dist-info/top_level.txt,sha256=Nf6G1sI6EqA2opfIdIio6ZqKNbpT2EtYf5cd0sKgPRo,11
-textfmt-0.2.0.dist-info/RECORD,,