Obtém elementos pelo nome da tag.

Ao invés de assumir que os nós são apenas elementos, e não podem ser
texto, busca por elementos com a tag. O teste do contribuinte passa a
funcionar após esse patch.

diff --git a/src/contribuinte.py b/src/contribuinte.py
index a9e055b..f477d0a 100644 (file)
--- a/src/contribuinte.py
+++ b/src/contribuinte.py
@@ -44,7 +44,7 @@ class Contribuinte:
     def _find_id(self):
         cpf = self._normalize_cpf(self.cpf)
         self.declaracoes = xml.dom.minidom.parse(self.iddecl_file)
-        for i in self.declaracoes.childNodes[0].childNodes:
+        for i in self.declaracoes.getElementsByTagName("item"):
             if "cpf" in i.attributes.keys():
                 if i.attributes["cpf"].nodeValue == cpf:
                     return i

diff --git a/src/municipios.py b/src/municipios.py
index 733b8fe..c9cd47a 100644 (file)
--- a/src/municipios.py
+++ b/src/municipios.py
@@ -23,7 +23,7 @@ class Municipios:
         self.l = []
 
     def _list(self):
-        for i in self.xml.childNodes[0].childNodes:
+        for i in self.xml.getElementsByTagName("ITEM"):
             if "COL3" in i.attributes.keys():
                 self.l.append((i.attributes["COL1"].nodeValue, \
                         i.attributes["COL2"].nodeValue, \

diff --git a/src/ocupacoes.py b/src/ocupacoes.py
index 87b9f32..755d070 100644 (file)
--- a/src/ocupacoes.py
+++ b/src/ocupacoes.py
@@ -28,7 +28,7 @@ class Ocupacoes:
         self._group()
 
     def _list(self):
-        for i in self.xml.childNodes[0].childNodes:
+        for i in self.xml.getElementsByTagName("ITEM"):
             if "COL4" in i.attributes.keys():
                 self.l.append((i.attributes["COL1"].nodeValue, \
                         i.attributes["COL2"].nodeValue, \