xlwings-utils 0.0.7.post2__py3-none-any.whl → 0.0.7.post3__py3-none-any.whl

xlwings_utils/xlwings_utils.py

@@ -13,7 +13,7 @@ from pathlib import Path
 import os
 import sys
 
-_org_stdout = sys.stdout
+_buffer=[]
 
 dbx = None
 Pythonista = sys.platform == "ios"
@@ -420,60 +420,48 @@ def captured_stdout_as_value(clear=True):
     return [[line] for line in captured_stdout_as_str(clear).splitlines()]
 
 
-class handle_capture_stdout:
-    def __init__(self):
-        pass
 
-    def isatty(self):
-        return False
 
-    def write(self, data):
-        _buffer.append(data)
-        if _do_print:
-            _org_stdout.write(data)
-
-    def flush(self):
-        if _do_print:
-            _org_stdout.flush()
-        _buffer.append("\n")
+class capture_stdout:
+    """
+    specifies how to capture stdout
 
+    Parameters
+    ----------
+    include_print : bool
+        if True (default), the output is also printed out as normal
 
-def reset():
-    global _buffer
-    sys.stdout=_org_stdout
-    _buffer=[]
+        if False, no output is printed
 
-reset()
+    Note
+    ----
+    Use this function as a context manager, like ::
 
-def capture_stdout(do_print=True, do_capture=True, clear=True):
-    global _do_print
+        with capture_stdout():
+            ...
     """
-    start capture stdout
 
-    Parameters
-    ----------
-    do_print : bool
-        if True (default), the output is (also) printed out as normal
+    def __init__(self, include_print: bool = True, clear=True):
+        self.stdout = sys.stdout
+        self.include_print = include_print
+        if clear:
+            _buffer.clear()
 
-        if False, no output is printed
+    def __enter__(self):
+        sys.stdout = self
 
-    do_capture : bool
-        if True (default), the output is (also) captured to a buffer
+    def __exit__(self, exc_type, exc_value, tb):
+        sys.stdout = self.stdout
 
-        if False, no output is captured
+    def write(self, data):
+        _buffer.append(data)
+        if self.include_print:
+            self.stdout.write(data)
 
-    clear : bool
-        if True (default), the capture buffer is emptied
-    """
-    if not (do_print or do_capture):
-        raise ValueError("at least one of do_print and do_capture should be True")
-
-    if do_capture:
-        if sys.stdout is _org_stdout:
-            sys.stdout = handle_capture_stdout()
-        _do_print=do_print
-    else:
-        sys.stdout=_org_stdout
+    def flush(self):
+        if self.include_print:
+            self.stdout.flush()
+        _buffer.append("\n")
 
 if __name__ == "__main__":
     ...

{xlwings_utils-0.0.7.post2.dist-info → xlwings_utils-0.0.7.post3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: xlwings_utils
-Version: 0.0.7.post2
+Version: 0.0.7.post3
 Summary: xlwings_utils
 Author-email: Ruud van der Ham <rt.van.der.ham@gmail.com>
 Project-URL: Homepage, https://github.com/salabim/xlwings_utils

xlwings_utils-0.0.7.post3.dist-info/RECORD

@@ -0,0 +1,6 @@
+xlwings_utils/__init__.py,sha256=FdaRztevSu5akGL7KBUBRzqwLMRTdvVUuS2Kfp2f1Uc,68
+xlwings_utils/xlwings_utils.py,sha256=8eVnPAY7ZzRt_beOq8uyBG2mY7Hu7uiyriYRLOK9DoU,12839
+xlwings_utils-0.0.7.post3.dist-info/METADATA,sha256=GDhsqh9Vu0ZCfe9_FGERX07Xxvrb874BWel3x55-1V4,5695
+xlwings_utils-0.0.7.post3.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
+xlwings_utils-0.0.7.post3.dist-info/top_level.txt,sha256=kf5SEv0gZiRObPhUoYcc1O_iX_wwTOPeUIYvzyYeAM4,14
+xlwings_utils-0.0.7.post3.dist-info/RECORD,,

xlwings_utils-0.0.7.post2.dist-info/RECORD

@@ -1,6 +0,0 @@
-xlwings_utils/__init__.py,sha256=FdaRztevSu5akGL7KBUBRzqwLMRTdvVUuS2Kfp2f1Uc,68
-xlwings_utils/xlwings_utils.py,sha256=kbfgeEj3IXA6-5ZTqtp2IJcJbKhJ-bqZZa7UgerqLVA,13140
-xlwings_utils-0.0.7.post2.dist-info/METADATA,sha256=rLzJ1x21CLpZLgZvnIlKq3kJ_4ieRRBxfv5ieAlDsao,5695
-xlwings_utils-0.0.7.post2.dist-info/WHEEL,sha256=DnLRTWE75wApRYVsjgc6wsVswC54sMSJhAEd4xhDpBk,91
-xlwings_utils-0.0.7.post2.dist-info/top_level.txt,sha256=kf5SEv0gZiRObPhUoYcc1O_iX_wwTOPeUIYvzyYeAM4,14
-xlwings_utils-0.0.7.post2.dist-info/RECORD,,