submenu to choose customer or user

2025-08-25 16:57:26 +02:00
parent 425565b639
commit 523b1d55d5
2 changed files with 19 additions and 9 deletions
--- a/views.py
+++ b/views.py
@@ -78,11 +78,13 @@ class View:
                   }
        return self.prompt_for_update(options)

-    def prompt_for_contract(self) -> dict:
+    def prompt_for_contract(self, customer_options, commercial_options) -> dict:
        contract = {}
        print("** New contract **")
-        contract['customer'] = input("Customer (id) ? : ")
-        contract['commercial'] = input("Commercial (id) ")
+        print("Customer ? :")
+        contract['customer'] = return_menu(customer_options)
+        print("Commercial ? :")
+        contract['commercial'] = return_menu(commercial_options)
        contract['amount'] = input("Budget ? : ")
        return contract