mirror of
https://github.com/ada-dmitry/lab6.git
synced 2026-09-24 09:20:15 +00:00
add nekit project
This commit is contained in:
ada-dmitry
@@ -0,0 +1,304 @@
|
||||
import sys
|
||||
|
||||
sys.path.append('tables')
|
||||
|
||||
from project_config import *
|
||||
from dbconnection import *
|
||||
from cathegory_table import *
|
||||
from dish_table import *
|
||||
|
||||
|
||||
class Main:
|
||||
config = ProjectConfig()
|
||||
connection = DbConnection(config)
|
||||
|
||||
def __init__(self):
|
||||
DbTable.dbconn = self.connection
|
||||
return
|
||||
|
||||
def db_init(self):
|
||||
ct = CathegoryTable()
|
||||
dt = DishTable()
|
||||
ct.create()
|
||||
dt.create()
|
||||
return
|
||||
|
||||
def db_insert_somethings(self):
|
||||
ct = CathegoryTable()
|
||||
dt = DishTable()
|
||||
ct.insert_one(["Салат"])
|
||||
ct.insert_one(["Второе"])
|
||||
ct.insert_one(["Суп"])
|
||||
dt.insert_one(["Борщ", "120", "Сварить супчик", 3])
|
||||
dt.insert_one(["Домашний салат", "15", "Настругать салатик", 1])
|
||||
dt.insert_one(["Жаркое", "60", "Нажарить на славу", 2])
|
||||
dt.insert_one(["Цезарь", "40", "Настругать салатик с курицей", 1])
|
||||
|
||||
def db_drop(self):
|
||||
ct = CathegoryTable()
|
||||
dt = DishTable()
|
||||
ct.drop()
|
||||
dt.drop()
|
||||
return
|
||||
|
||||
def show_main_menu(self):
|
||||
menu = """Добро пожаловать!
|
||||
Основное меню (выберите цифру в соответствии с необходимым действием):
|
||||
1 - просмотр всех категорий;
|
||||
2 - сброс и инициализация таблиц;
|
||||
3 - просмотр всех блюд;
|
||||
9 - выход."""
|
||||
print(menu)
|
||||
return
|
||||
|
||||
def read_next_step(self):
|
||||
return input("=> ").strip()
|
||||
|
||||
def after_main_menu(self, next_step):
|
||||
if next_step == "2":
|
||||
self.db_drop()
|
||||
self.db_init()
|
||||
self.db_insert_somethings()
|
||||
print("Таблицы созданы заново!")
|
||||
return "0"
|
||||
if not next_step.isdigit():
|
||||
print("Неверный тип данных! Введите число!")
|
||||
return "0"
|
||||
elif next_step != "1" and next_step != "9" and next_step != "3":
|
||||
print("Выбрано неверное число! Повторите ввод!")
|
||||
return "0"
|
||||
else:
|
||||
return next_step
|
||||
|
||||
def show_cathegory(self):
|
||||
menu = """Просмотр категорий!
|
||||
№\tНазвание"""
|
||||
print(menu)
|
||||
allCathegory = CathegoryTable().all()
|
||||
self.max_cath_index = len(allCathegory)
|
||||
for i in allCathegory:
|
||||
print(str(i[0]) + "\t" + str(i[1]))
|
||||
menu = """Дальнейшие операции:
|
||||
0 - возврат в главное меню;
|
||||
4 - добавление новой категории;
|
||||
5 - удаление категории;
|
||||
6 - изменить название категории;
|
||||
9 - выход."""
|
||||
print(menu)
|
||||
return
|
||||
|
||||
def show_dish(self):
|
||||
menu = """Просмотр блюд!
|
||||
Формат: № ; Название ; Время приготовления ; Мануал ; Номер категории"""
|
||||
print(menu)
|
||||
allDish = DishTable().all()
|
||||
for i in allDish:
|
||||
print(str(i[0]) + " ; " + str(i[1]) + " ; " + str(i[2]) + " ; " + str(i[3]) + " ; " + str(i[4]))
|
||||
menu = """Дальнейшие операции:
|
||||
0 - возврат в главное меню;
|
||||
1 - просмотр блюд по номеру категории;
|
||||
2 - удаление блюда;
|
||||
3 - добавление блюда;
|
||||
9 - выход."""
|
||||
print(menu)
|
||||
return
|
||||
|
||||
def after_show_dish(self, next_step):
|
||||
while True:
|
||||
if next_step == "1":
|
||||
self.all_dish_by_cathegory()
|
||||
return "0"
|
||||
elif next_step == "2":
|
||||
self.del_dish()
|
||||
return "3"
|
||||
elif next_step == "3":
|
||||
self.insert_dish()
|
||||
return "3"
|
||||
elif next_step != "0" and next_step != "9":
|
||||
print("Выбрано неверное число! Повторите ввод!")
|
||||
next_step = self.read_next_step()
|
||||
else:
|
||||
return next_step
|
||||
|
||||
def all_dish_by_cathegory(self):
|
||||
cathegory_number = input("Введите номер категории(enter - отмена): ").strip()
|
||||
if cathegory_number == "":
|
||||
return
|
||||
try:
|
||||
cathegory_number = int(cathegory_number)
|
||||
except ValueError:
|
||||
print("Неверный тип. Ожидается число")
|
||||
return
|
||||
txt = """"Просмотр блюд!
|
||||
Формат: № ; Название ; Время приготовления ; Мануал ; Номер категории"""
|
||||
print(txt)
|
||||
allDish = DishTable().get_all_by_cathegory_number(cathegory_number)
|
||||
for i in allDish:
|
||||
print(str(i[0]) + " ; " + str(i[1]) + " ; " + str(i[2]) + " ; " + str(i[3]) + " ; " + str(i[4]))
|
||||
|
||||
|
||||
def del_dish(self):
|
||||
dish_id = input("Введите № блюда для удаления: ").strip()
|
||||
try:
|
||||
dish_id = int(dish_id)
|
||||
if dish_id > 0:
|
||||
DishTable().delete_one(dish_id)
|
||||
else:
|
||||
print("Неверное значение. Ожидается целое положительное число")
|
||||
return
|
||||
except ValueError:
|
||||
print("Неверный тип. Ожидается число")
|
||||
return
|
||||
dish_table = DishTable()
|
||||
if not dish_table.exist_by_number(dish_id):
|
||||
print("Указанный номер блюда не найден. Попробуйте заново")
|
||||
return
|
||||
|
||||
dish_table.delete_one(dish_id)
|
||||
print("Блюдо успешно удалено")
|
||||
|
||||
|
||||
def insert_dish(self):
|
||||
data = []
|
||||
|
||||
data.append(input("Введите имя блюда(enter - отмена): ").strip())
|
||||
while len(data[0].strip()) > 64:
|
||||
data[0] = input("Неверная длина! Имя должно быть не пустым и не больше 64 символов. Введите имя заново (1 - отмена):").strip()
|
||||
if data[0] == "":
|
||||
print("Отмена действия")
|
||||
return
|
||||
|
||||
data.append(input("Введите время приготовления в минутах(0 - отмена): ").strip())
|
||||
flag = True
|
||||
while flag:
|
||||
try:
|
||||
data[1] = int(data[1])
|
||||
except ValueError:
|
||||
data[1] = input("Неверный тип. Ожидается число. Введите число заново (0 - отмена):").strip()
|
||||
continue
|
||||
flag = False
|
||||
if data[1] == 0:
|
||||
return
|
||||
if data[1] < 0:
|
||||
print("Время не может быть отрицательным")
|
||||
return
|
||||
|
||||
data.append(input("Введите мануал(enter - отмена): ").strip())
|
||||
#while len(data[2].strip()) == 0:
|
||||
# data[2] = input("Неверная длина! Мануал должно быть не пустым. Введите мануал заново (enter - отмена):").strip()
|
||||
if data[2] == "":
|
||||
print("Отмена действия")
|
||||
return
|
||||
|
||||
data.append(input("Введите номер категории(enter - отмена): ").strip())
|
||||
if data[3] == "":
|
||||
return
|
||||
flag = True
|
||||
while flag:
|
||||
try:
|
||||
data[3] = int(data[3])
|
||||
except ValueError:
|
||||
data[3] = input("Неверный тип. Ожидается число. Введите число заново (enter - отмена):").strip()
|
||||
if data[3] == "":
|
||||
return
|
||||
continue
|
||||
flag = False
|
||||
|
||||
if not (CathegoryTable().exist_by_number(data[3])):
|
||||
print("Указанный номер категории не найдем. Попробуйте заново")
|
||||
return
|
||||
DishTable().insert_one(data)
|
||||
|
||||
def after_show_cathegory(self, next_step):
|
||||
while True:
|
||||
if next_step == "4":
|
||||
self.insert_cathegory()
|
||||
return "1"
|
||||
elif next_step == "5":
|
||||
self.del_cathegory()
|
||||
return "1"
|
||||
elif next_step == "6" :
|
||||
self.update_cathegory()
|
||||
return "1"
|
||||
elif next_step != "0" and next_step != "9":
|
||||
print("Выбрано неверное число! Повторите ввод!")
|
||||
next_step = self.read_next_step()
|
||||
else:
|
||||
return next_step
|
||||
|
||||
def insert_cathegory(self):
|
||||
data = []
|
||||
data.append(input("Введите имя категории(1 - отмена): ").strip())
|
||||
while len(data[0].strip()) == 0 or len(data[0].strip())>20:
|
||||
data[0] = input("Неверная длина! Имя должно быть не пустым и не больше 20 символов. Введите имя заново (1 - отмена):").strip()
|
||||
if data[0] == "1":
|
||||
return
|
||||
|
||||
CathegoryTable().insert_one(data)
|
||||
|
||||
def del_cathegory(self):
|
||||
while True:
|
||||
cathegory_id = input("Введите № категории для удаления: ").strip()
|
||||
try:
|
||||
cathegory_id = int(cathegory_id)
|
||||
except ValueError:
|
||||
print("Неверный тип. Ожидается число")
|
||||
# return
|
||||
if (cathegory_id <= 0)or(cathegory_id > self.max_cath_index):
|
||||
print(f'Число должно лежать в диапазоне от 1 до {self.max_cath_index}.')
|
||||
elif not (CathegoryTable().exist_by_number(cathegory_id)):
|
||||
print("Указанный номер категории не найдем. Попробуйте заново")
|
||||
# return
|
||||
else:
|
||||
break
|
||||
DishTable().delete_all_with_cathegory_id(cathegory_id)
|
||||
CathegoryTable().delete_one(cathegory_id)
|
||||
return
|
||||
|
||||
def update_cathegory(self):
|
||||
data = []
|
||||
cathegory_id = input("Введите № категории для изменения: ").strip()
|
||||
try:
|
||||
cathegory_id = int(cathegory_id)
|
||||
except ValueError:
|
||||
print("Неверный тип. Ожидается число")
|
||||
return
|
||||
if not (CathegoryTable().exist_by_number(cathegory_id)):
|
||||
print("Указанный номер категории не найдем. Попробуйте заново")
|
||||
return
|
||||
|
||||
data.append(input("Введите имя категории(1 - отмена): ").strip())
|
||||
while len(data[0].strip()) == 0 or len(data[0].strip()) > 20:
|
||||
data[0] = input("Неверная длина! Имя должно быть не пустым и не больше 20 символов. Введите имя заново (1 - отмена):").strip()
|
||||
if data[0] == "1":
|
||||
return
|
||||
CathegoryTable().update(cathegory_id, data)
|
||||
|
||||
def main_cycle(self):
|
||||
current_menu = "0"
|
||||
next_step = None
|
||||
while (current_menu != "9"):
|
||||
if current_menu == "0":
|
||||
self.show_main_menu()
|
||||
next_step = self.read_next_step()
|
||||
current_menu = self.after_main_menu(next_step)
|
||||
elif current_menu == "1":
|
||||
self.show_cathegory()
|
||||
next_step = self.read_next_step()
|
||||
current_menu = self.after_show_cathegory(next_step)
|
||||
elif current_menu == "3":
|
||||
self.show_dish()
|
||||
next_step = self.read_next_step()
|
||||
current_menu = self.after_show_dish(next_step)
|
||||
print("До свидания!")
|
||||
return
|
||||
|
||||
def test(self):
|
||||
DbTable.dbconn.test()
|
||||
|
||||
|
||||
m = Main()
|
||||
# Откоментируйте эту строку и закоментируйте следующую для теста
|
||||
# соединения с БД
|
||||
# m.test()
|
||||
m.main_cycle()
|
||||
Reference in New Issue
Block a user