fix: better errors support

This commit is contained in:
egarette@silique.fr 2024-11-28 22:04:39 +01:00
parent 6f08f03070
commit 2958a53b7d

View file

@ -19,6 +19,7 @@ along with this program. If not, see <http://www.gnu.org/licenses/>.
from typing import Optional
from json import dumps
from .i18n import _
from ..output_json import RougailOutputJson
@ -43,39 +44,39 @@ class RougailOutputAnsible(RougailOutputJson):
self.errors = []
self.warnings = []
def run(self) -> None:
self.exporter()
print(dumps(self.json_to_ansible(), ensure_ascii=False, indent=2))
def exporter(self) -> None:
super().exporter()
self.json_to_ansible()
def json_to_ansible(self):
ret = {"_meta": {"hostvars": {}}, "all": {"children": ["ungrouped"]}}
if self.host_namespace not in self.dico:
self.errors.append(_('cannot find hosts namespace "{0}"').format(self.host_namespace))
if 'hostnames' not in self.dico[self.host_namespace]:
elif 'hostnames' not in self.dico[self.host_namespace]:
self.errors.append(_('malformated hosts namespace "{0}", should has "hostnames"').format(self.host_namespace))
ret = {"_meta": {"hostvars": {}}, "all": {"children": ["ungrouped"]}}
# if self.errors:
# ret["_meta"]["hostvars"]["localhost"] = {'_errors': self.errors}
# ret["ungrouped"] = {"hosts": ["localhost"]}
# else:
hostnames = self.dico[self.host_namespace]['hostnames']
ret_hosts = {}
for name, hosts in hostnames.items():
if 'hosts' in hosts:
for idx, host in enumerate(hosts['hosts']):
index = str(idx + 1)
if idx < 9:
index = '0' + index
host_name = hosts['prefix_name'] + index
ret_hosts.setdefault(name, {})[host_name] = host
ret.setdefault(name, {}).setdefault('hosts', []).append(host_name)
else:
ret["all"]["children"].append(name)
ret[name] = hosts
for hosts in ret_hosts.values():
for host, domain_name in hosts.items():
ret['_meta']['hostvars'][host] = {'ansible_host': domain_name}
ret['_meta']['hostvars'][host].update(self.dico)
return ret
if self.errors:
ret["_meta"]["hostvars"]["localhost"] = {'_errors': self.errors}
ret["ungrouped"] = {"hosts": ["localhost"]}
else:
hostnames = self.dico[self.host_namespace]['hostnames']
ret_hosts = {}
for name, hosts in hostnames.items():
if 'hosts' in hosts:
for idx, host in enumerate(hosts['hosts']):
index = str(idx + 1)
if idx < 9:
index = '0' + index
host_name = hosts['prefix_name'] + index
ret_hosts.setdefault(name, {})[host_name] = host
ret.setdefault(name, {}).setdefault('hosts', []).append(host_name)
else:
ret["all"]["children"].append(name)
ret[name] = hosts
for hosts in ret_hosts.values():
for host, domain_name in hosts.items():
ret['_meta']['hostvars'][host] = {'ansible_host': domain_name}
ret['_meta']['hostvars'][host].update(self.dico)
self.dico = ret
RougailOutput = RougailOutputAnsible