diff options
Diffstat (limited to 'data.py')
-rw-r--r-- | data.py | 56 |
1 files changed, 0 insertions, 56 deletions
diff --git a/data.py b/data.py deleted file mode 100644 index 4b48e19..0000000 --- a/data.py +++ /dev/null @@ -1,56 +0,0 @@ -from xml.etree.ElementTree import ElementTree - -class Network(): - """A simple network definition""" - - def __init__(self, name): - self.name = name - self.domain = None - self.ip_blocks = [] - self.data = {} - -class Data(): - """Abstract class for storing network information""" - - def __init__(self, config): - self.networks = [] - self.config = config - - def parse_config(self): - """Abstract method""" - - pass - - def load_data(self): - """Abstract method""" - - pass - - def get_networks(self): - """Return all networks. Common method for all kind of storages.""" - - if self.networks == None: - self.load_data() - - return self.networks - -class DataXML(Data): - """Reads network information from a XML file""" - - def parse_config(): - """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""" # Ugly implementation. Beautify. - root = ElementTree(file=self.data_file).getroot() - for elem in root: - network = WhoisNetwork(elem.attrib['name']) - for e in elem: - if e.tag == 'ip_block': - network.ip_blocks.append(e.text) - else: - network.data[e.tag] = e.text - - self.networks.append(network) |