Change ipcalc to ipaddr

1 files changed, 71 insertions, 0 deletions

diff --git a/model.py b/model.py
index fbba248..13bcfa8 100644
--- a/model.py
+++ b/model.py
@@ -1,3 +1,5 @@
+from xml.etree import ElementTree
+
 class Network():
     """A simple network definition"""
     
@@ -76,3 +78,72 @@ class Data():
             self.load_data()
 
         return self.persons
+
+class DataXML(model.Data):
+    """Reads network information from a XML file"""
+    
+    def parse_config(self):
+        """Reads and sets up XML config file fields"""
+        
+        self.data_file = self.config['Storage']['xml_file']
+        
+    def load_data(self):
+        """Parse XML for getting network information"""
+
+        root = ElementTree(file=self.data_file).getroot()
+        
+        for elem in root:
+            if elem.tag == 'person':
+                self.add_person(elem)
+            elif elem.tag == 'domain':
+                self.add_domain(elem)
+            elif elem.tag == 'network':
+                self.add_network(elem)
+            else:
+                pass # raise TagNotFoundError
+
+    def add_person(self, elem):
+        """Adds a new person"""
+
+        person = model.Person()
+        for e in elem:
+            if e.tag == 'name':
+                person.name = e.text
+            elif e.tag == 'surname':
+                person.surname = e.text
+            elif e.tag == 'email':
+                person.email = e.text
+            else:
+                person.data[e.tag] = e.text
+
+        self.persons[elem.attrib['id']] = person
+
+    def add_domain(self, elem):
+        """Adds a new domain"""
+
+        domain = model.Domain()
+        for e in elem:
+            if e.tag == 'name':
+                domain.name == e.text
+            else:
+                domain.data[e.tag] = e.text
+
+        self.domains[elem.attrib['id']] = domain                
+
+    def add_network(self, elem):
+        """Adds a new network"""
+        
+        network = model.Network()
+        for e in elem:
+            if e.tag == 'name':
+                network.name = e.text
+            elif e.tag == 'domain':
+                network.domains.append(e.text)            
+            elif e.tag == 'ip_block':
+                network.ip_blocks.append(e.text)
+            elif e.tag == 'admin':
+                network.admins.append(e.text)
+            else:
+                network.data[e.tag] = e.text
+                
+        self.networks[elem.attrib['id']] = network