Adiciona lista de bens à interface.

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

diff --git a/src/menu.py b/src/menu.py
index acbdcdf..f42281d 100644 (file)
--- a/src/menu.py
+++ b/src/menu.py
@@ -19,6 +19,7 @@
 import baseui
 import contribuinte
 import rendimentoPJ
+import bens
 import sys
 
 def List(UI, L, display):
@@ -49,12 +50,28 @@ def DadosPessoais(UI, contrib):
     UI.form(contrib.form())
     return True
 
+def Bens(UI, contrib):
+    b = bens.Bens(contrib)
+    return List(UI, b, "discriminacao")
+
 def Salvar(UI, contrib):
     contrib.save()
 
 def menu(UI, contrib):
-    m = [ "Sair", "Salvar", "Dados Pessoais", "Rendimentos PJ" ]
-    f = [ None, Salvar, DadosPessoais, RendimentosPJ ]
+    m = [ 
+        "Sair",
+        "Salvar",
+        "Dados Pessoais",
+        "Rendimentos PJ",
+        "Bens",
+        ]
+    f = [
+        None,
+        Salvar,
+        DadosPessoais,
+        RendimentosPJ,
+        Bens,
+        ]
     exit = False
     while not exit:
         r = UI.menu(m)