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

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

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: textfmt
-Version: 0.1.1
-Summary: A Python module for ANSI text formatting, word art, and table styling
+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,,

{textfmt-0.1.1.dist-info → textfmt-0.2.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.8.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

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

@@ -0,0 +1,98 @@
+import requests
+import time
+from pathlib import Path
+import sys
+
+# Define color codes for terminal (ANSI escape codes)
+class Colors:
+    RESET = "\033[0m"
+    GREEN = "\033[92m"
+    BLUE = "\033[94m"
+    RED = "\033[91m"
+    YELLOW = "\033[93m"
+    MAGENTA = "\033[95m"
+    CYAN = "\033[96m"
+    WHITE = "\033[97m"
+    LIGHT_GREEN = "\033[92m"
+    LIGHT_BLUE = "\033[94m"
+    LIGHT_RED = "\033[91m"
+    LIGHT_YELLOW = "\033[93m"
+    LIGHT_MAGENTA = "\033[95m"
+    LIGHT_CYAN = "\033[96m"
+
+    ALL_COLORS = [
+        GREEN, BLUE, RED, YELLOW, MAGENTA, CYAN, WHITE,
+        LIGHT_GREEN, LIGHT_BLUE, LIGHT_RED, LIGHT_YELLOW, LIGHT_MAGENTA, LIGHT_CYAN
+    ]
+
+class DownloadProgressBar:
+    def __init__(self, total, prefix='', length=40, fill='━', empty=' ', print_end="\r",
+                 download_color=Colors.GREEN, complete_color=Colors.BLUE):
+        self.total = total
+        self.prefix = prefix
+        self.length = length
+        self.fill = fill
+        self.empty = empty
+        self.print_end = print_end
+        self.download_color = download_color
+        self.complete_color = complete_color
+        self.start_time = time.time()
+        self.downloaded = 0
+
+    def update(self, downloaded):
+        now = time.time()
+        self.downloaded = downloaded
+
+        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)
+
+        # 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
+        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 and self.total > 0:
+            sys.stdout.write(self.print_end)
+            sys.stdout.flush()
+
+    @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.1.1.dist-info/RECORD

@@ -1,9 +0,0 @@
-textfmt-0.1.1.dist-info/licenses/LICENSE,sha256=Msmy46KiAR2wmzeCe3kSl-wEt7ptbTZOAzdwUUyFq_M,1090
-textformat/__init__.py,sha256=fyib7AnpkuEZ_PAzx6AO5SVKNTB5kyqk5NnJBXx7TYw,200
-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.1.1.dist-info/METADATA,sha256=HC4hKXmxtK8eHMssLwSxVQHgrrXg6ouOhIJpcs7iKAg,1566
-textfmt-0.1.1.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-textfmt-0.1.1.dist-info/top_level.txt,sha256=Nf6G1sI6EqA2opfIdIio6ZqKNbpT2EtYf5cd0sKgPRo,11
-textfmt-0.1.1.dist-info/RECORD,,