Adiciona Rendimentos Exclusivos na Fonta à Interface

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

diff --git a/src/menu.py b/src/menu.py
index f42281d..4a4dc49 100644 (file)
--- a/src/menu.py
+++ b/src/menu.py
@@ -21,6 +21,8 @@ import contribuinte
 import rendimentoPJ
 import bens
 import sys
+import isentos
+import exclusivos
 
 def List(UI, L, display):
     exit = False
@@ -54,6 +56,20 @@ def Bens(UI, contrib):
     b = bens.Bens(contrib)
     return List(UI, b, "discriminacao")
 
+def Quadro(UI, quadro):
+    return List(UI, quadro, "especificacao")
+
+def Quadros(quadros):
+    def fQuadros(UI, contrib):
+        exit = False
+        while not exit:
+            r = UI.menu(map(lambda x: x.name, quadros.quadros))
+            if r < 0:
+                exit = True
+            else:
+                Quadro(UI, quadros.quadros[r])
+    return fQuadros
+
 def Salvar(UI, contrib):
     contrib.save()
 
@@ -64,6 +80,8 @@ def menu(UI, contrib):
         "Dados Pessoais",
         "Rendimentos PJ",
         "Bens",
+        "Rendimentos Isentos",
+        "Rendimentos Exclusivos na Fonte",
         ]
     f = [
         None,
@@ -71,6 +89,8 @@ def menu(UI, contrib):
         DadosPessoais,
         RendimentosPJ,
         Bens,
+        Quadros(isentos.RendimentosIsentos(contrib)),
+        Quadros(exclusivos.RendimentosExclusivos(contrib)),
         ]
     exit = False
     while not exit: