stouputils 1.2.18__tar.gz → 1.2.20__tar.gz

{stouputils-1.2.18 → stouputils-1.2.20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stouputils
-Version: 1.2.18
+Version: 1.2.20
 Summary: Stouputils is a collection of utility modules designed to simplify and enhance the development process. It includes a range of tools for tasks such as execution of doctests, display utilities, decorators, as well as context managers, and many more.
 Project-URL: Homepage, https://github.com/Stoupy51/stouputils
 Project-URL: Issues, https://github.com/Stoupy51/stouputils/issues

{stouputils-1.2.18 → stouputils-1.2.20}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "stouputils"
-version = "1.2.18"
+version = "1.2.20"
 description = "Stouputils is a collection of utility modules designed to simplify and enhance the development process. It includes a range of tools for tasks such as execution of doctests, display utilities, decorators, as well as context managers, and many more."
 readme = "README.md"
 requires-python = ">=3.10"

{stouputils-1.2.18 → stouputils-1.2.20}/stouputils/ctx.py

@@ -12,7 +12,7 @@ This module provides context managers for temporarily silencing output.
 import os
 import sys
 from typing import IO, TextIO, Callable, Any
-from .print import remove_colors
+from .print import TeeMultiOutput
 from .io import super_open
 
 
@@ -55,54 +55,6 @@ class Muffle:
 			sys.stderr = self.original_stderr
 
 
-# TeeMultiOutput class to duplicate output to multiple file-like objects
-class TeeMultiOutput(object):
-	""" File-like object that duplicates output to multiple file objects.
-
-	Args:
-		*files        (IO[Any]):  One or more file-like objects that have write and flush methods
-		strip_colors  (bool):     Whether to strip ANSI color codes from output sent to non-stdout/stderr files
-
-	Examples:
-		>>> f = open("logfile.txt", "w")
-		>>> original_stdout = sys.stdout
-		>>> sys.stdout = TeeMultiOutput(sys.stdout, f)
-		>>> print("Hello World")  # Output goes to both console and file
-		>>> sys.stdout = original_stdout
-		>>> f.close()
-	"""
-	def __init__(self, *files: IO[Any], strip_colors: bool = True) -> None:
-		self.files: tuple[IO[Any], ...] = files
-		""" File-like objects to write to """
-		self.strip_colors: bool = strip_colors
-		""" Whether to strip ANSI color codes from output sent to non-stdout/stderr files """
-
-	def write(self, obj: str) -> None:
-		""" Write the object to all files.
-
-		Args:
-			obj (str): String to write
-		"""
-		for f in self.files:
-			# Strip colors for files that are not stdout or stderr
-			if self.strip_colors and f not in (sys.stdout, sys.stderr):
-				f.write(remove_colors(obj))
-			else:
-				f.write(obj)
-
-	def flush(self) -> None:
-		""" Flush all files. """
-		for f in self.files:
-			f.flush()
-
-	# Add other methods that might be expected from a file-like object
-	def isatty(self) -> bool:
-		""" Return whether the first file is connected to a tty-like device. """
-		return hasattr(self.files[0], 'isatty') and self.files[0].isatty()
-
-	def fileno(self) -> int:
-		""" Return the file descriptor of the first file. """
-		return self.files[0].fileno()
 
 # Context manager to log to a file
 class LogToFile:

{stouputils-1.2.18 → stouputils-1.2.20}/stouputils/parallel.py

@@ -123,7 +123,7 @@ def multiprocessing(func: Callable[[T], R], args: list[T], use_starmap: bool = F
 	# Do multiprocessing only if there is more than 1 argument and more than 1 CPU
 	if max_workers > 1 and len(args) > 1:
 		if verbose > 0:
-			return list(process_map(func, args, max_workers=max_workers, chunksize=chunksize, desc=desc, bar_format=BAR_FORMAT)) # type: ignore
+			return list(process_map(func, args, max_workers=max_workers, chunksize=chunksize, desc=desc, bar_format=BAR_FORMAT, ascii=False)) # type: ignore
 		else:
 			with Pool(max_workers) as pool:
 				return list(pool.map(func, args, chunksize=chunksize))	# type: ignore
@@ -131,7 +131,7 @@ def multiprocessing(func: Callable[[T], R], args: list[T], use_starmap: bool = F
 	# Single process execution
 	else:
 		if verbose > 0:
-			return [func(arg) for arg in tqdm(args, total=len(args), desc=desc, bar_format=BAR_FORMAT)]
+			return [func(arg) for arg in tqdm(args, total=len(args), desc=desc, bar_format=BAR_FORMAT, ascii=False)]
 		else:
 			return [func(arg) for arg in args]
 
@@ -176,7 +176,7 @@ def multithreading(func: Callable[[T], R], args: list[T], use_starmap: bool = Fa
 	if max_workers > 1 and len(args) > 1:
 		if verbose > 0:
 			with ThreadPoolExecutor(max_workers) as executor:
-				return list(tqdm(executor.map(func, args), total=len(args), desc=desc, bar_format=BAR_FORMAT))
+				return list(tqdm(executor.map(func, args), total=len(args), desc=desc, bar_format=BAR_FORMAT, ascii=False))
 		else:
 			with ThreadPoolExecutor(max_workers) as executor:
 				return list(executor.map(func, args))
@@ -184,7 +184,7 @@ def multithreading(func: Callable[[T], R], args: list[T], use_starmap: bool = Fa
 	# Single process execution
 	else:
 		if verbose > 0:
-			return [func(arg) for arg in tqdm(args, total=len(args), desc=desc, bar_format=BAR_FORMAT)]
+			return [func(arg) for arg in tqdm(args, total=len(args), desc=desc, bar_format=BAR_FORMAT, ascii=False)]
 		else:
 			return [func(arg) for arg in args]
 

{stouputils-1.2.18 → stouputils-1.2.20}/stouputils/print.py

@@ -10,7 +10,7 @@ If a message is printed multiple times, it will be displayed as "(xN) message" w
 # Imports
 import sys
 import time
-from typing import Callable, TextIO, Any
+from typing import Callable, IO, TextIO, Any
 
 # Colors constants
 RESET: str   = "\033[0m"
@@ -224,7 +224,73 @@ def breakpoint(*values: Any, print_function: Callable[..., None] = warning, **pr
 		sys.exit(1)
 
 
+# TeeMultiOutput class to duplicate output to multiple file-like objects
+class TeeMultiOutput(object):
+	""" File-like object that duplicates output to multiple file objects.
 
+	Args:
+		*files        (IO[Any]):  One or more file-like objects that have write and flush methods
+		strip_colors  (bool):     Whether to strip ANSI color codes from output sent to non-stdout/stderr files
+		force_tty     (bool):     Whether to force reporting as a TTY device (helps with tqdm)
+		ascii_only    (bool):     Whether to replace non-ASCII characters with their ASCII equivalents for non-stdout/stderr files
+
+	Examples:
+		>>> f = open("logfile.txt", "w")
+		>>> original_stdout = sys.stdout
+		>>> sys.stdout = TeeMultiOutput(sys.stdout, f)
+		>>> print("Hello World")  # Output goes to both console and file
+		>>> sys.stdout = original_stdout
+		>>> f.close()
+	"""
+	def __init__(self, *files: IO[Any], strip_colors: bool = True, ascii_only: bool = True) -> None:
+		self.files: tuple[IO[Any], ...] = files
+		""" File-like objects to write to """
+		self.strip_colors: bool = strip_colors
+		""" Whether to strip ANSI color codes from output sent to non-stdout/stderr files """
+		self.ascii_only: bool = ascii_only
+		""" Whether to replace non-ASCII characters with their ASCII equivalents for non-stdout/stderr files """
+
+	def write(self, obj: str) -> None:
+		""" Write the object to all files while stripping colors if needed.
+
+		Args:
+			obj (str): String to write
+		"""
+		for i, f in enumerate(self.files):
+			try:
+				content = obj
+				if i != 0:
+					# First file (i = 0) (often stdout/stderr) gets the original content
+					# Other files (i != 0) get processed content
+
+					# Strip colors if needed
+					if self.strip_colors:
+						content = remove_colors(content)
+
+					# Replace Unicode block characters with ASCII equivalents
+					# Replace other problematic Unicode characters as needed
+					if self.ascii_only:
+						content = content.replace('█', '#')
+						content = ''.join(c if ord(c) < 128 else '?' for c in content)
+
+				# Write content to file
+				f.write(content)
+
+			except Exception:
+				pass
+
+	def flush(self) -> None:
+		""" Flush all files. """
+		for f in self.files:
+			if hasattr(f, 'flush'):
+				try:
+					f.flush()
+				except Exception:
+					pass
+
+	def fileno(self) -> int:
+		""" Return the file descriptor of the first file. """
+		return self.files[0].fileno() if hasattr(self.files[0], "fileno") else 0