stouputils 1.2.37__tar.gz → 1.2.38__tar.gz

{stouputils-1.2.37 → stouputils-1.2.38}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stouputils
-Version: 1.2.37
+Version: 1.2.38
 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.37 → stouputils-1.2.38}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "stouputils"
-version = "1.2.37"
+version = "1.2.38"
 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.37 → stouputils-1.2.38}/stouputils/ctx.py

@@ -9,6 +9,7 @@ This module provides context managers for temporarily silencing output.
 """
 
 # Imports
+from __future__ import annotations
 import os
 import sys
 import time
@@ -37,7 +38,7 @@ class Muffle:
 		self.original_stderr: TextIO = sys.stderr
 		""" Attribute remembering original stderr """
 
-	def __enter__(self) -> None:
+	def __enter__(self) -> Muffle:
 		""" Enter context manager which redirects stdout and stderr to devnull """
 		# Redirect stdout to devnull
 		sys.stdout = open(os.devnull, "w", encoding="utf-8")
@@ -46,6 +47,9 @@ class Muffle:
 		if self.mute_stderr:
 			sys.stderr = open(os.devnull, "w", encoding="utf-8")
 
+		# Return self
+		return self
+
 	def __exit__(self, exc_type: type[BaseException]|None, exc_val: BaseException|None, exc_tb: Any|None) -> None:
 		""" Exit context manager which restores original stdout and stderr """
 		# Restore original stdout
@@ -110,15 +114,17 @@ class LogToFile:
 		self.original_stderr: TextIO = sys.stderr
 		""" Original stderr before redirection """
 
-	def __enter__(self) -> None:
+	def __enter__(self) -> LogToFile:
 		""" Enter context manager which opens the log file and redirects stdout/stderr """
 		# Redirect stdout and stderr if requested
 		if self.tee_stdout:
 			sys.stdout = TeeMultiOutput(self.original_stdout, self.file, ignore_lineup=self.ignore_lineup)
-
 		if self.tee_stderr:
 			sys.stderr = TeeMultiOutput(self.original_stderr, self.file, ignore_lineup=self.ignore_lineup)
 
+		# Return self
+		return self
+
 	def __exit__(self, exc_type: type[BaseException]|None, exc_val: BaseException|None, exc_tb: Any|None) -> None:
 		""" Exit context manager which closes the log file and restores stdout/stderr """
 		# Restore original stdout and stderr
@@ -204,9 +210,10 @@ class MeasureTime:
 		self.start_ns: int = 0
 		""" Start time in nanoseconds """
 
-	def __enter__(self) -> None:
+	def __enter__(self) -> MeasureTime:
 		""" Enter context manager, record start time """
 		self.start_ns = self.ns()
+		return self
 
 	def __exit__(self, exc_type: type[BaseException]|None, exc_val: BaseException|None, exc_tb: Any|None) -> None:
 		""" Exit context manager, calculate duration and print """