Search by domain

1 files changed, 26 insertions, 1 deletions

diff --git a/model.py b/model.py
index 99f368f..ef5ce28 100644
--- a/model.py
+++ b/model.py
@@ -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):