zebra/pyzebra
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Initialize AnatricConfig with basic elements

Browse Source
This commit is contained in:
usov_i 2020-07-29 14:47:38 +02:00
parent 55d6797c3f
commit 2bd224683d
1 changed files with 14 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
pyzebra/anatric.py
View File

@ -25,19 +25,23 @@ def anatric(config_file):
class AnatricConfig: class AnatricConfig:
def __init__(self, filename=None): def __init__(self, filename=None):
if filename:
self.load_from_file(filename)
def load_from_file(self, filename):
self._tree = ET.parse(filename)
self._root = self._tree.getroot()
self._alg_elems = dict() self._alg_elems = dict()
for alg in ALGORITHMS: for alg in ALGORITHMS:
self._alg_elems[alg] = ET.Element("Algorithm", attrib={"implementation": alg}) self._alg_elems[alg] = ET.Element("Algorithm", attrib={"implementation": alg})
self._alg_elems[alg].text = "\n" self._alg_elems[alg].text = "\n"
self._alg_elems[alg].tail = "\n\n" self._alg_elems[alg].tail = "\n\n"
root_elem = ET.Element("anatric")
root_elem.text = "\n"
root_elem.append(self._alg_elems[ALGORITHMS[0]])
self._tree = ET.ElementTree(element=root_elem)
if filename:
self.load_from_file(filename)
def load_from_file(self, filename):
self._tree.parse(filename)
self._alg_elems[self.algorithm] = self._tree.find("Algorithm") self._alg_elems[self.algorithm] = self._tree.find("Algorithm")
def save_as(self, filename): def save_as(self, filename):
@ -217,8 +221,9 @@ class AnatricConfig:
if value not in ALGORITHMS: if value not in ALGORITHMS:
raise ValueError("Unknown algorithm.") raise ValueError("Unknown algorithm.")
self._root.remove(self._tree.find("Algorithm")) root = self._tree.getroot()
self._root.append(self._alg_elems[value]) root.remove(self._tree.find("Algorithm"))
root.append(self._alg_elems[value])
def _get_alg_attr(self, alg, tag, attr): def _get_alg_attr(self, alg, tag, attr):
param_elem = self._alg_elems[alg].find(tag) param_elem = self._alg_elems[alg].find(tag)