Modify set_logger

thombashi · Feb 11, 2020 · 50db148 · 50db148
1 parent 8789dfb
commit 50db148
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 10 deletions.
diff --git a/requirements/requirements.txt b/requirements/requirements.txt
@@ -1,4 +1,4 @@
-DataProperty>=0.46.1,<2
+DataProperty>=0.46.2,<2
 enum34;python_version<"3.4"
 futures;python_version<"3.2"
 six>=1.10.0,<2

diff --git a/tabledata/_logger/_logger.py b/tabledata/_logger/_logger.py
@@ -12,28 +12,25 @@
 
 
 MODULE_NAME = "tabledata"
-_is_enable = False
 
 try:
     from loguru import logger
 
     logger.disable(MODULE_NAME)
 except ImportError:
-    logger = NullLogger()
+    logger = NullLogger()  # type: ignore
 
 
-def set_logger(is_enable):
-    global _is_enable
-
-    if is_enable == _is_enable:
-        return
-
+def set_logger(is_enable, propagation_depth=1):
     if is_enable:
         logger.enable(MODULE_NAME)
     else:
         logger.disable(MODULE_NAME)
 
-    dataproperty.set_logger(is_enable)
+    if propagation_depth <= 0:
+        return
+
+    dataproperty.set_logger(is_enable, propagation_depth - 1)
 
 
 def set_log_level(log_level):