From b72fba3e35d5877f3ae5d9ae5837bb66a8f8d6f4 Mon Sep 17 00:00:00 2001
From: Ross McFarland <ross@github.com>
Date: Sat, 2 Dec 2017 10:34:32 -0800
Subject: [PATCH] Move the log plan output to PlanLogger

---
 octodns/manager.py       | 35 +------------------------------
 octodns/provider/plan.py | 45 +++++++++++++++++++++++++++++++++++++++-
 2 files changed, 45 insertions(+), 35 deletions(-)

diff --git a/octodns/manager.py b/octodns/manager.py
index e08754c..c4617aa 100644
--- a/octodns/manager.py
+++ b/octodns/manager.py
@@ -5,7 +5,6 @@
 from __future__ import absolute_import, division, print_function, \
     unicode_literals
 
-from StringIO import StringIO
 from concurrent.futures import ThreadPoolExecutor
 from importlib import import_module
 from os import environ
@@ -260,39 +259,7 @@ class Manager(object):
         # plan pairs.
         plans = [p for f in futures for p in f.result()]
 
-        hr = '*************************************************************' \
-            '*******************\n'
-        buf = StringIO()
-        buf.write('\n')
-        if plans:
-            current_zone = None
-            for target, plan in plans:
-                if plan.desired.name != current_zone:
-                    current_zone = plan.desired.name
-                    buf.write(hr)
-                    buf.write('* ')
-                    buf.write(current_zone)
-                    buf.write('\n')
-                    buf.write(hr)
-
-                buf.write('* ')
-                buf.write(target.id)
-                buf.write(' (')
-                buf.write(target)
-                buf.write(')\n*   ')
-                for change in plan.changes:
-                    buf.write(change.__repr__(leader='* '))
-                    buf.write('\n*   ')
-
-                buf.write('Summary: ')
-                buf.write(plan)
-                buf.write('\n')
-        else:
-            buf.write(hr)
-            buf.write('No changes were planned\n')
-        buf.write(hr)
-        buf.write('\n')
-        self.log.info(buf.getvalue())
+        PlanLogger(self.log).output(plans)
 
         if not force:
             self.log.debug('sync:   checking safety')
diff --git a/octodns/provider/plan.py b/octodns/provider/plan.py
index cde859f..ab84a10 100644
--- a/octodns/provider/plan.py
+++ b/octodns/provider/plan.py
@@ -5,7 +5,8 @@
 from __future__ import absolute_import, division, print_function, \
     unicode_literals
 
-from logging import getLogger
+from StringIO import StringIO
+from logging import INFO, getLogger
 
 
 class UnsafePlan(Exception):
@@ -77,3 +78,45 @@ class Plan(object):
             .format(self.change_counts['Create'], self.change_counts['Update'],
                     self.change_counts['Delete'],
                     len(self.existing.records))
+
+
+class PlanLogger(object):
+
+    def __init__(self, log, level=INFO):
+        self.log = log
+        self.level = level
+
+    def output(self, plans):
+        hr = '*************************************************************' \
+            '*******************\n'
+        buf = StringIO()
+        buf.write('\n')
+        if plans:
+            current_zone = None
+            for target, plan in plans:
+                if plan.desired.name != current_zone:
+                    current_zone = plan.desired.name
+                    buf.write(hr)
+                    buf.write('* ')
+                    buf.write(current_zone)
+                    buf.write('\n')
+                    buf.write(hr)
+
+                buf.write('* ')
+                buf.write(target.id)
+                buf.write(' (')
+                buf.write(target)
+                buf.write(')\n*   ')
+                for change in plan.changes:
+                    buf.write(change.__repr__(leader='* '))
+                    buf.write('\n*   ')
+
+                buf.write('Summary: ')
+                buf.write(plan)
+                buf.write('\n')
+        else:
+            buf.write(hr)
+            buf.write('No changes were planned\n')
+        buf.write(hr)
+        buf.write('\n')
+        self.log.log(self.level, buf.getvalue())