Substitui ContribuinteForm por AttrForm
[cascardo/irpf-gui.git] / src / contribuinte.py
index 80c17c0..28c2ccb 100644 (file)
@@ -1,6 +1,6 @@
 # coding=utf-8
 #
-#   Copyright 2013 Thadeu Lima de Souza Cascardo <cascardo@cascardo.info>
+#   Copyright 2013-2014 Thadeu Lima de Souza Cascardo <cascardo@cascardo.info>
 #
 #   This program is free software: you can redistribute it and/or modify
 #   it under the terms of the GNU General Public License as published by
 #   You should have received a copy of the GNU General Public License
 #   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 # -*- mode: python; encoding: utf-8; -*-
+
 import xml.dom.minidom
 import dirs
 import os
+import form
+import ocupacoes
+import declaracoes
 
 class Contribuinte:
     def __init__(self, cpf):
@@ -28,28 +32,20 @@ class Contribuinte:
             raise RuntimeError("Invalid CPF: " + self.cpf)
 
         if not os.path.exists(irpf_dir.get_resource_dir()):
-            raise RuntimeError("O caminho para o resource não existe: " + \
+            raise RuntimeError("O caminho para o resource não existe: " + \
                     irpf_dir.get_resource_dir())
 
         if not os.path.exists(irpf_dir.get_userdata_dir()):
-            raise RuntimeError("O caminho para os dados não existe: " + \
+            raise RuntimeError("O caminho para os dados não existe: " + \
                     irpf_dir.get_userdata_dir())
 
         self.cpf_file = irpf_dir.get_userdata_file("%s/%s.xml" % (self.cpf, self.cpf))
-        self.iddecl_file = irpf_dir.get_userdata_file("iddeclaracoes.xml")
-        self.declaracao = self._find_id()
+        ncpf = self._normalize_cpf(self.cpf)
+        self.declaracoes = declaracoes.Declaracoes()
+        self.declaracao = self.declaracoes.find("cpf", ncpf)
         self.dados = xml.dom.minidom.parse(self.cpf_file)
         self.contribuinte = self.dados.getElementsByTagName("contribuinte")[0]
 
-    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:
-            if "cpf" in i.attributes.keys():
-                if i.attributes["cpf"].nodeValue == cpf:
-                    return i
-        return None
-
     # CPF normalizado se parece com 000.000.000-00
     def _normalize_cpf(self, cpf):
         ncpf = ""
@@ -68,31 +64,48 @@ class Contribuinte:
 
     # CPF minimizado se parece com 01234567890
     def _minimize_cpf(self, cpf):
-        ncpf = bytearray(self._normalize_cpf(cpf))
-        del ncpf[11]
-        del ncpf[7]
-        del ncpf[3]
-        return str(ncpf)
+        return self._minimize_valor(cpf)
+
+    def _minimize_valor(self, valor):
+        nvalor = ''.join(e for e in valor if e.isalnum())
+        return str(nvalor)
 
     def _validate_cpf(self, cpf):
-        ncpf = self._minimize_cpf(cpf)
-        if len(ncpf) != 11:
+        if len(cpf) != 11:
             return False
-        v = (11 - sum(map(lambda x: x[0]*x[1], zip(range(10, 1, -1), map(lambda x: ord(x) - ord('0'), ncpf[0:9]))))) % 11
-        if v >= 10:
-            v = 0
-        if v != ord(ncpf[9]) - ord('0'):
-            return False
-        v = (11 - sum(map(lambda x: x[0]*x[1], zip(range(11, 1, -1), map(lambda x: ord(x) - ord('0'), ncpf[0:10]))))) % 11
-        if v >= 10:
-            v = 0
-        if v != ord(ncpf[10]) - ord('0'):
-            return False
-        return True
+        return self._validate_generico(cpf)
+
+    def _validate_generico(self, valor):
+        def calcula_digito_verificador(numero):
+            n = len(numero) + 1
+
+            soma = 0
+            for i in range(n):
+                if i > len(numero) - 1:
+                    break
+                soma = soma + int(numero[i]) * ( n - i)
+
+            dv =  soma % 11
+
+            if dv < 2:
+                dv = 0
+            else:
+                dv = 11 - dv
+
+            return numero + str(dv)
+
+        mcpf = self._minimize_valor(valor)
+        cpf_sem_dv = mcpf[:-2]
+
+        primeiro_dv = str(calcula_digito_verificador(cpf_sem_dv))
+        segundo_dv = calcula_digito_verificador(primeiro_dv)
+
+        return segundo_dv == mcpf
 
     def save(self):
         self.dados.writexml(open(self.cpf_file, "w"))
-        self.declaracoes.writexml(open(self.iddecl_file, "w"))
+        self.declaracoes.save()
+
     def _get_attr(self, el, attr):
         if attr in el.attributes.keys():
             return el.attributes[attr].nodeValue
@@ -102,10 +115,10 @@ class Contribuinte:
         el.attributes[attr].nodeValue = val
 
     def get_declaracao(self, attr):
-        return self._get_attr(self.declaracao, attr)
+        return self.declaracao.get_attr(attr)
 
     def set_declaracao(self, attr, val):
-        self._set_attr(self.declaracao, attr, val)
+        self.declaracao.set_attr(attr, val)
 
     def get_nome(self):
         return self.get_declaracao("nome")
@@ -124,49 +137,47 @@ class Contribuinte:
         else:
             self._set_attr(self.contribuinte, attr, val)
 
-contribuinte_attributes = [
-        "nome",
-        "dataNascimento",
-        "tituloEleitor",
-        "doencaDeficiencia",
-        "exterior",
-        "pais",
-        "cep",
-        "uf",
-        "cidade",
-        "municipio",
-        "tipoLogradouro",
-        "logradouro",
-        "numero",
-        "complemento",
-        "bairro",
-        "bairroExt",
-        "cepExt",
-        "logradouroExt",
-        "numeroExt",
-        "complementoExt",
-        "ocupacaoPrincipal",
-        "codigoExterior",
-        "ddd",
-        "telefone",
-        "naturezaOcupacao",
-        ]
-
-declaracao_attributes = [
-        "dataUltimoAcesso",
-        "declaracaoRetificadora",
-        "enderecoDiferente",
-        "enderecoMACRede",
-        "exercicio",
-        "nome",
-        "numReciboDecRetif",
-        "numeroReciboDecAnterior",
-        "resultadoDeclaracao",
-        "tipoDeclaracao",
-        "tipoDeclaracaoAES",
-        "transmitida",
-        "versaoBeta"
-        ]
+    def get_attr(self, attr):
+        return self.get_campo_contribuinte(attr)
+
+    def set_attr(self, attr, val):
+        self.set_campo_contribuinte(attr, val)
+
+    def form(self):
+        f = []
+        f.append(form.AttrForm("Nome", "nome", self))
+        f.append(ocupacoes.OcupacaoForm(self))
+        for i in self.attributes:
+            f.append(form.AttrForm(i, i, self))
+        return f
+
+    attributes = [
+            "nome",
+            "dataNascimento",
+            "tituloEleitor",
+            "doencaDeficiencia",
+            "exterior",
+            "pais",
+            "cep",
+            "uf",
+            "cidade",
+            "municipio",
+            "tipoLogradouro",
+            "logradouro",
+            "numero",
+            "complemento",
+            "bairro",
+            "bairroExt",
+            "cepExt",
+            "logradouroExt",
+            "numeroExt",
+            "complementoExt",
+            "ocupacaoPrincipal",
+            "codigoExterior",
+            "ddd",
+            "telefone",
+            "naturezaOcupacao",
+            ]
 
 if __name__ == '__main__':
     import sys
@@ -188,13 +199,13 @@ if __name__ == '__main__':
             print ("Campo " + campo + " retornou vazio")
     else:
         print "\nCONTRIBUINTE:"
-        for i in contribuinte_attributes:
+        for i in Contribuinte.attributes:
             val = contribuinte.get_campo_contribuinte(i)
             if val == None:
                 val = ""
             print i + ": " + val
         print "\nDECLARACAO:"
-        for i in declaracao_attributes:
+        for i in declaracoes.Declaracoes.attributes:
             val = contribuinte.get_declaracao(i)
             if val == None:
                 val = ""