Improve Print to support nested verbosity regions

sergioburdisso · Feb 22, 2020 · 78176ab · 78176ab
1 parent 9c7e4fd
commit 78176ab
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/pyss3/util.py b/pyss3/util.py
@@ -1721,7 +1721,7 @@ class Print:
     style = Style
 
     __verbosity__ = VERBOSITY.NORMAL
-    __verbosity_old__ = None
+    __verbosity_region_stack__ = []
 
     @staticmethod
     def error(msg='', raises=None, offset=0, decorator=True):
@@ -1926,8 +1926,8 @@ def verbosity_region_begin(level):
                       (see ``set_verbosity`` documentation for valid values)
         :type level: int
         """
+        Print.__verbosity_region_stack__.append(Print.__verbosity__)
         if not Print.is_quiet():
-            Print.__verbosity_old__ = Print.__verbosity__
             Print.__verbosity__ = level
 
     @staticmethod
@@ -1948,9 +1948,7 @@ def verbosity_region_end():
         >>> Print.verbosity_region_end()
         >>> # the verbosity level is restored to what it was before entering the region
         """
-        if Print.__verbosity_old__ is not None:
-            Print.__verbosity__ = Print.__verbosity_old__
-            Print.__verbosity_old__ = None
+        Print.__verbosity__ = Print.__verbosity_region_stack__.pop()
 
 
 def round_fix(v, precision=4):