refactory query to use resource editor class

Change-Id: I1cb37cedde5a79c3683cbe5566d2080f73f49836
Signed-off-by: Doug Hellmann <doug@doughellmann.com>
This commit is contained in:
Doug Hellmann 2017-04-30 21:34:13 +00:00
parent 1719e77a1c
commit c4e3919e9b
1 changed files with 25 additions and 10 deletions

View File

@ -19,6 +19,27 @@ from downpour import resources
LOG = logging.getLogger(__name__)
class ResourceFileEditor:
def __init__(self, filename, save_state):
self._filename = filename
self._save_state = save_state
self._resources = resources.load(filename, missing_ok=True)
self._servers = set(s.name for s in self._resources.servers)
def add_server(self, info):
if info.name in self._servers:
return
LOG.info('found server %s to export', info.name)
self._resources.servers.append({
'name': info.name,
'save_state': self._save_state,
})
def save(self):
resources.save(self._filename, self._resources)
def register_command(subparsers):
do_query = subparsers.add_parser(
'query',
@ -43,17 +64,11 @@ def register_command(subparsers):
def query_data(cloud, config, args):
to_export = resources.load(args.resource_file, missing_ok=True)
servers = set(s.name for s in to_export.servers)
editor = ResourceFileEditor(args.resource_file, save_state=args.save_state)
for pattern in args.server:
LOG.info('searching for server %r', pattern)
LOG.info('searching for servers matching pattern %r', pattern)
for server_info in cloud.search_servers(name_or_id=pattern):
if server_info.name not in servers:
LOG.info('found server %s to export', server_info.name)
to_export.servers.append({
'name': server_info.name,
'save_state': args.save_state,
})
editor.add_server(server_info)
resources.save(args.resource_file, to_export)
editor.save()