Tambem le outros dados sobre a declaracao

Inclui dados presentes em iddeclaracoes.xml

diff --git a/contribuinte.py b/contribuinte.py
index 77ed8e9..403910b 100644 (file)
--- a/contribuinte.py
+++ b/contribuinte.py
@@ -78,10 +78,14 @@ class Contribuinte:
                return None
        def _set_attr(self, el, attr, val):
                el.attributes[attr].nodeValue = val
+       def get_declaracao(self, attr):
+               return self._get_attr(self.declaracao, attr)
+       def set_declaracao(self, attr, val):
+               self._set_attr(self.declaracao, attr, val)
        def get_nome(self):
-               return self._get_attr(self.declaracao, "nome")
+               return self.get_declaracao("nome")
        def set_nome(self, nome):
-               self._set_attr(self.declaracao, "nome", nome)
+               self.set_declaracao("nome", nome)
        def get_contribuinte(self, attr):
                if attr == "nome":
                        return self.get_nome()
@@ -91,7 +95,7 @@ class Contribuinte:
                        self.set_nome(val)
                self._set_attr(self.contribuinte, attr, val)
 
-attributes = [
+contribuinte_attributes = [
        "nome",
        "dataNascimento",
        "tituloEleitor",
@@ -119,6 +123,22 @@ attributes = [
        "naturezaOcupacao",
 ]
 
+declaracao_attributes = [
+       "dataUltimoAcesso",
+       "declaracaoRetificadora",
+       "enderecoDiferente",
+       "enderecoMACRede",
+       "exercicio",
+       "nome",
+       "numReciboDecRetif",
+       "numeroReciboDecAnterior",
+       "resultadoDeclaracao",
+       "tipoDeclaracao",
+       "tipoDeclaracaoAES",
+       "transmitida",
+       "versaoBeta"
+]
+
 if __name__ == '__main__':
        import sys
        contribuinte = Contribuinte(sys.argv[1])
@@ -135,5 +155,15 @@ if __name__ == '__main__':
                print "Salvando..."
                contribuinte.save()
        else:
-               for i in attributes:
-                       print i + ": " + contribuinte.get_contribuinte(i)
+               print "\nCONTRIBUINTE:"
+               for i in contribuinte_attributes:
+                       val = contribuinte.get_contribuinte(i)
+                       if val == None:
+                               val = ""
+                       print i + ": " + val
+               print "\nDECLARACAO:"
+               for i in declaracao_attributes:
+                       val = contribuinte.get_declaracao(i)
+                       if val == None:
+                               val = ""
+                       print i + ": " + val