diff options
Diffstat (limited to 'model.py')
| -rw-r--r-- | model.py | 27 |
1 files changed, 26 insertions, 1 deletions
@@ -10,6 +10,29 @@ class Network(): self.ip_blocks = [] self.data = {} + def as_dict(self, data): + # Beautify + result = {} + domains = data.get_domains() + persons = data.get_persons() + + result['name'] = self.name + + # Domains + result['domains'] = [] + for d in self.domains: + result['domains'].append(domains[d]) + + # Admins + result['admins'] = [] + for a in self.admins: + result['admins'].append(persons[a]) + + # Networks + result['networks'] = self.ip_blocks + + return result + class Domain(): """A simple domain definition""" @@ -121,6 +144,7 @@ class DataXML(Data): else: person.data[e.tag] = e.text + print("[+] Read person: {0} - {1} - {2}".format(person.name, person.surname, person.email)) self.persons[elem.attrib['id']] = person def add_domain(self, elem): @@ -129,10 +153,11 @@ class DataXML(Data): domain = Domain() for e in elem: if e.tag == 'name': - domain.name == e.text + domain.name = e.text else: domain.data[e.tag] = e.text + print("[+] Read domain: {}".format(domain.name)) self.domains[elem.attrib['id']] = domain def add_network(self, elem): |