Я предлагаю вам прочитать о Конечном автомате .
Короче говоря, изначально ваша программа должна быть в состоянии main-menu
.после ввода (2) состояние должно измениться на deposit
и запросить дальнейший ввод данных пользователем.В тот момент, когда пользователь вводит (1), вы должны игнорировать блок main-menu
if / else и вместо этого проверять блок deposit
.
, таким образом, решение с псевдокодом, использующее ваш код в качестве основы, будетвыглядеть примерно так:
while (app_running):
if verify_pin(pin):
if app_state == 'main-menu':
# print main menu
# request input
# process input in the main menu if/else case
if app_state == 'deposit':
# print deposit menu
# request input
# process input in the deposit if/else case
...