From 85a412c172044ae89d381f69ddb309ce8b3cea6e Mon Sep 17 00:00:00 2001
From: Joshua Harlow <harlowja@yahoo-inc.com>
Date: Tue, 16 Oct 2012 21:14:51 -0700
Subject: Move the recursive flushing to the log module.

---
 cloudinit/config/cc_package_update_upgrade_install.py | 15 ++-------------
 cloudinit/log.py                                      | 12 ++++++++++++
 2 files changed, 14 insertions(+), 13 deletions(-)

(limited to 'cloudinit')

diff --git a/cloudinit/config/cc_package_update_upgrade_install.py b/cloudinit/config/cc_package_update_upgrade_install.py
index e319e147..73b0e30d 100644
--- a/cloudinit/config/cc_package_update_upgrade_install.py
+++ b/cloudinit/config/cc_package_update_upgrade_install.py
@@ -16,10 +16,10 @@
 #    You should have received a copy of the GNU General Public License
 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-from logging import StreamHandler
 import os
 import time
 
+from cloudinit import log as logging
 from cloudinit import util
 
 REBOOT_FILE = "/var/run/reboot-required"
@@ -33,17 +33,6 @@ def _multi_cfg_bool_get(cfg, *keys):
     return False
 
 
-def _flush_loggers(root):
-    for h in root.handlers:
-        if isinstance(h, (StreamHandler)):
-            try:
-                h.flush()
-            except IOError:
-                pass
-    if root.parent:
-        _flush_loggers(root.parent)
-
-
 def _fire_reboot(log, wait_attempts=6, initial_sleep=1, backoff=2):
     util.subp(REBOOT_CMD)
     start = time.time()
@@ -98,7 +87,7 @@ def handle(_name, cfg, cloud, log, _args):
         try:
             log.warn("Rebooting after upgrade or install per %s", REBOOT_FILE)
             # Flush the above warning + anything else out...
-            _flush_loggers(log)
+            logging.flushLoggers(log)
             _fire_reboot(log)
         except Exception as e:
             util.logexc(log, "Requested reboot did not happen!")
diff --git a/cloudinit/log.py b/cloudinit/log.py
index 2333e5ee..da6c2851 100644
--- a/cloudinit/log.py
+++ b/cloudinit/log.py
@@ -53,6 +53,18 @@ def setupBasicLogging():
     root.setLevel(DEBUG)
 
 
+def flushLoggers(root):
+    if not root:
+        return
+    for h in root.handlers:
+        if isinstance(h, (logging.StreamHandler)):
+            try:
+                h.flush()
+            except IOError:
+                pass
+    flushLoggers(root.parent)
+
+
 def setupLogging(cfg=None):
     # See if the config provides any logging conf...
     if not cfg:
-- 
cgit v1.2.3