summaryrefslogtreecommitdiff
path: root/cloudinit/config
diff options
context:
space:
mode:
Diffstat (limited to 'cloudinit/config')
-rw-r--r--cloudinit/config/cc_package_update_upgrade_install.py15
1 files changed, 2 insertions, 13 deletions
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!")