Refac: get_contribuinte -> get_campo_contribuinte

[cascardo/irpf-gui.git] / src / contribuinte.py

diff --git a/src/contribuinte.py b/src/contribuinte.py
index 587762c..ad10a21 100644 (file)
--- a/src/contribuinte.py
+++ b/src/contribuinte.py
@@ -97,7 +97,7 @@ class Contribuinte:
     def set_nome(self, nome):
         self.set_declaracao("nome", nome)
 
-    def get_contribuinte(self, attr):
+    def get_campo_contribuinte(self, attr):
         if attr == "nome":
             return self.get_nome()
         return self._get_attr(self.contribuinte, attr)
@@ -155,21 +155,24 @@ if __name__ == '__main__':
     import sys
     contribuinte = Contribuinte(sys.argv[1])
     print "Carregando CPF " + contribuinte._normalize_cpf(sys.argv[1])
-    if contribuinte._validate_cpf(sys.argv[1]):
-        print "CPF valido"
-    else:
-        print "CPF invalido"
-        sys.exit(1)
+
     if len(sys.argv) == 4:
-        print "Valor anterior: " + contribuinte.get_contribuinte(sys.argv[2])
+        print "Valor anterior: " + contribuinte.get_campo_contribuinte(sys.argv[2])
         contribuinte.set_contribuinte(sys.argv[2], sys.argv[3])
-        print "Valor atual: " + contribuinte.get_contribuinte(sys.argv[2])
+        print "Valor atual: " + contribuinte.get_campo_contribuinte(sys.argv[2])
         print "Salvando..."
         contribuinte.save()
+    elif len(sys.argv) == 3:
+        campo = sys.argv[2]
+        valor = contribuinte.get_campo_contribuinte(campo)
+        if valor:
+            print ("Valor de " + campo + ": " + valor)
+        else:
+            print ("Campo " + campo + " retornou vazio")
     else:
         print "\nCONTRIBUINTE:"
         for i in contribuinte_attributes:
-            val = contribuinte.get_contribuinte(i)
+            val = contribuinte.get_campo_contribuinte(i)
             if val == None:
                 val = ""
             print i + ": " + val