mirror/github-octodns
1
0
Fork 0
mirror of https://github.com/github/octodns.git synced 2024-05-11 05:55:00 +00:00
Code Issues Releases Activity

Move the log plan output to PlanLogger

Browse Source
This commit is contained in:
Ross McFarland
2017-12-02 10:34:32 -08:00
parent f50d9b6087
commit b72fba3e35
2 changed files with 45 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
octodns/manager.py
View File

@@ -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')

45
octodns/provider/plan.py
View File

@@ -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())