diff --git a/app/menu_load.py b/app/menu_load.py index 854b324..685d0ac 100644 --- a/app/menu_load.py +++ b/app/menu_load.py @@ -2,12 +2,33 @@ from app.utils.system import clear_console from app.zadachi.zadacha1 import zadacha1 from app.zadachi.zadacha2 import zadacha2 -class pod_menu: +class menu: def __init__(self): pass # Конструктор пока пустой, при необходимости можно добавить параметры - def zadacha1_menu(): + def main_menu(self): + clear_console() + while True: + print('\nMain Menu:') + print('1. Задача 1') + print('2. Задача 2') + print('0. Exit') + + choice = input('Select an option: ') + + clear_console() + if choice == '1': + self.zadacha1_menu() # Переход к подменю Задачи 1 + elif choice == '2': + self.zadacha2_menu() # Переход к подменю Задачи 2 + elif choice == '0': + print('Exiting program...') + break + else: + print('Invalid option') + + def zadacha1_menu(self): """Меню для задачи 1 с подкатегориями.""" clear_console() while True: @@ -32,7 +53,7 @@ class pod_menu: else: print('Неверный выбор, попробуйте снова.') - def zadacha2_menu(): + def zadacha2_menu(self): """Меню для задачи 2 (расчет коэффициентов методом наименьших квадратов).""" clear_console() z1 = zadacha1() # Используем объект задачи 1 для генерации данных diff --git a/main.py b/main.py index f25480c..613f6fc 100644 --- a/main.py +++ b/main.py @@ -1,29 +1,6 @@ -from app.utils.config import debug -from app.utils.system import clear_console -from app.menu_load import pod_menu - - -def main_menu(): - clear_console() - while True: - print('\nMain Menu:') - print('1. Задача 1') - print('2. Задача 2') - print('0. Exit') - - choice = input('Select an option: ') - - clear_console() - if choice == '1': - pod_menu.zadacha1_menu() # Переход к подменю Задачи 1 - elif choice == '2': - pod_menu.zadacha2_menu() # Переход к подменю Задачи 2 - elif choice == '0': - print('Exiting program...') - break - else: - print('Invalid option') +from app.menu_load import menu if __name__ == "__main__": - main_menu() + mn = menu() + mn.main_menu()