From 64fa281bc1185911c8e8a1c6f9e6c74afae2a374 Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Sun, 28 May 2023 18:52:01 -0700
Subject: [PATCH] parameter for flushing logs

---
 .../python_overrides/logger/__init__.py       | 46 ++++++++++++++++---
 1 file changed, 39 insertions(+), 7 deletions(-)

diff --git a/container_includes/python_overrides/logger/__init__.py b/container_includes/python_overrides/logger/__init__.py
index dacb959a29..60796f5d8c 100644
--- a/container_includes/python_overrides/logger/__init__.py
+++ b/container_includes/python_overrides/logger/__init__.py
@@ -45,7 +45,7 @@ class FakeYouLogger:
     def __init__(self):
         self.level = _get_log_level()
         self.handlers = []  # We don't care about this complexity
-        self.flush = True
+        self.flush = _get_flush_setting()
         pass
 
     # noinspection PyPep8Naming
@@ -57,16 +57,20 @@ class FakeYouLogger:
         pass
 
     def warning(self, *args):
-        print(args, flush=self.flush)
+        if self.level >= WARN:
+            print('warning', args, flush=self.flush)
 
     def warn(self, *args):
-        print(args, flush=self.flush)
+        if self.level >= WARN:
+            print('warn', args, flush=self.flush)
 
     def info(self, *args):
-        print(args, flush=self.flush)
+        if self.level >= INFO:
+            print('info', args, flush=self.flush)
 
     def debug(self, *args):
-        print(args, flush=self.flush)
+        if self.level >= DEBUG:
+            print('debug', args, flush=self.flush)
 
 
 #---------------------------------------------------------------------------
@@ -108,16 +112,44 @@ _nameToLevel = {
     'NOTSET': NOTSET,
 }
 
+_DEFAULT_LOG_LEVEL = WARN
+
 
 def _get_log_level():
     # FakeYou-specific
     level = os.getenv("PYTHON_LOG_LEVEL")
     if not level or type(level) is not str:
-        return WARN
+        return _DEFAULT_LOG_LEVEL
     level = level.strip().upper()
     if level in _nameToLevel:
         return _nameToLevel[level]
-    return WARN
+    return _DEFAULT_LOG_LEVEL
+
+
+_strToBool = {
+    'false': False,
+    'f': False,
+    'no': False,
+    '0': False,
+
+    'true': True,
+    't': True,
+    'yes': True,
+    '1': True,
+}
+
+_DEFAULT_FLUSH_SETTING = True  # NB: This might slow the code down.
+
+
+def _get_flush_setting():
+    # FakeYou-specific
+    setting = os.getenv("PYTHON_FLUSH_LOGS")
+    if not setting or type(setting) is not str:
+        return _DEFAULT_FLUSH_SETTING
+    setting = setting.strip().lower()
+    if setting in _strToBool:
+        return _strToBool[setting]
+    return _DEFAULT_FLUSH_SETTING
 
 
 LOGGER = FakeYouLogger()