diff --git a/0.8_indev/__pycache__/dbtable.cpython-311.pyc b/0.8_indev/__pycache__/dbtable.cpython-311.pyc index aa06591..873cecf 100644 Binary files a/0.8_indev/__pycache__/dbtable.cpython-311.pyc and b/0.8_indev/__pycache__/dbtable.cpython-311.pyc differ diff --git a/0.8_indev/main.py b/0.8_indev/main.py index 86a948e..d68caaa 100644 --- a/0.8_indev/main.py +++ b/0.8_indev/main.py @@ -25,11 +25,19 @@ TODO: 1) detele category - done +2) update category - done +3) add category - done FIXME: 1) Откомментировать os.cls 2) Исправить вылет при неверном вводе в самое начало(сделать выход на предыдущую менюшку) 3) При вводе текста пропуск сделать с помощью пустой строки. + +TEST: +1) Удаление категории + ввод неизвестных символов и отмена +2) Обновление категории + ввод неизвестных символов и отмена +3) Добавление категории + ввод неизвестных символов и отмена + ''' @@ -153,16 +161,11 @@ class Main: next_step = "5" elif next_step == "7":#Удаление блюда из категории - if(self.dish_page<(self.max_dish_index//10)): - left = ((self.dish_page-1)*10)+(self.dish_page==1) - right = left+9 - - else: - left = 1+((self.dish_page-1)*10)+(self.dish_page!=1) - right = self.mex_dish_index + left = 0 + right = self.max_dish_index x = add_func.validate_input('Введите номер удаляемого блюда (0 - для отмены): ', left, right) if(x!=-1): - DishTable().delete(self.dish_arr[(int(x)-1)-((self.dish_page-1)*10)][0]) + DishTable().delete(DishTable().get_dish_from_page(x).get("id")) else: return "1" next_step = "5" @@ -171,29 +174,19 @@ class Main: next_step = self.show_dish_in_cath() elif next_step == "8": # Обновление названия категории - if(self.cath_page<(self.max_cath_index//10)): - left = 1+((self.cath_page-1)*10) - right = left+9 - - else: - left = 1+((self.cath_page-1)*10) - right = self.max_cath_index + left = 0 + right = self.max_cath_index x = add_func.validate_input('Введите номер обновляемой категории (0 - для отмены): ', left, right) if(x!=-1): - CathTable().cath_update(self.cath_arr[(int(x)-1)-((self.cath_page-1)*10)]) + CathTable().cath_update(CathTable().get_cath_from_page(x).get("cath_name")) return "1" - elif next_step == "88": - if(self.dish_page<(self.mex_dish_index//10)): - left = 1+((self.dish_page-1)*10) - right = left+9 - - else: - left = 0+((self.dish_page-1)*10) - right = self.mex_dish_index + elif next_step == "88": # Обновление блюда + left = 0 + right = self.max_dish_index x = add_func.validate_input('Введите номер обновляемого блюда (0 - для отмены): ', left, right) if(x!=-1): - DishTable().update_dish(self.dish_arr[(int(x)-1)-((self.cath_page-1)*10)][0]) + DishTable().update_dish(DishTable().get_dish_from_page(x)) else: return "1" @@ -208,86 +201,81 @@ class Main: """ Добавление новой категории в таблицу """ - data = [] - data.append(input("Введите название (1 - отмена): ").strip()) - if data[0] == "1": - return - while((len(data[0].strip()) == 0)or(len(data[0].strip()) > 32)): - if (len(data[0].strip()) > 32): - data[0] = input("Название слишком длинное! Введите название заново (1 - отмена):").strip() - if data[0] == "1": - return + data = input("Введите название (enter - отмена): ").strip() + while((len(data) == 0)or(len(data) > 32)or(CathTable().check_by_name(data))): + if (len(data) > 32): + data = input("Название слишком длинное! Введите название заново (enter - отмена): ").strip() + elif(CathTable().check_by_name(data)): + data = input("Такое название уже существует! Введите название заново (enter - отмена): ").strip() else: - data[0] = input("Название не может быть пустым! Введите название заново (1 - отмена):").strip() - if data[0] == "1": - return + return CathTable().insert_one(data) return + + def show_dish_page(self): + print("Блюда:") + dish_page = 1 + lst = DishTable().get_dish_page(self.cath_id, dish_page) + self.max_len_name = 0 + self.max_dish_index = DishTable().count(self.cath_id) + action = '>' + max_page = math.ceil(self.max_dish_index/10) + lst = DishTable().get_dish_page(self.cath_id, dish_page) + + while True: + self.dish_arr = [] + + for i in lst: + self.dish_arr.append([i[2], str(i[1]), str(i[4])]) + self.max_len_name = max(self.max_len_name, len(i[2])) + add_func.cls() + print("№" + " "*(len(str(self.max_dish_index)) + 1)+ "Название" + " "*(self.max_len_name - 4)\ + +"Время приготовления Краткая инструкция\ + \n-------------------------------------------------------------------------------------------") + + for i in range(len(self.dish_arr)): + txt = add_func.add_zero_before(str((i+1)+(10*(dish_page-1))), \ + len(str(self.max_dish_index))) + " "*(2+len(str(self.max_dish_index))) + txt += self.dish_arr[i][0] + " "*(4+self.max_len_name - len(self.dish_arr[i][0])) + txt += self.dish_arr[i][1] + " "*(5+19-len(self.dish_arr[i][1])) + txt += self.dish_arr[i][2] + print(txt) + action = input('-------------------------------------------------------------------------------------------\n\ + Для переключения между страницами используйте "<" и ">"\nКол-во записей на странице: 10\n\ + Для выхода в меню действий нажмите любую другую клавишу\n=> ') + if(action=='>'): + if(dish_page==max_page): + dish_page = 1 + else: + dish_page += 1 + lst = DishTable().get_dish_page(self.cath_id, dish_page) + elif(action=='<'): + if(dish_page==1): + dish_page = max_page + else: + dish_page -= 1 + lst = DishTable().get_dish_page(self.cath_id, dish_page) + else: + break def show_dish_in_cath(self): """Вывод всех блюд в выбранной пользователем категории """ - if self.cath_id == -1: while True: - if(self.cath_page<(self.max_cath_index//10)): - left = 1+((self.cath_page-1)*10) - right = left+9 - - else: - left = 1+((self.cath_page-1)*10)+(self.cath_page!=1) - right = self.max_cath_index + left = 0 + right = self.max_cath_index x = add_func.validate_input('Выберите номер интересуемой категории (0 - отмена): ', left, right) add_func.cls() if(x!=-1): - self.cath_id = CathTable().find_by_name(self.cath_arr[(int(x)-1)-((self.cath_page-1)*10)]) - self.cath_obj = self.cath_arr[(int(x)-1)-((self.cath_page-1)*10)] + self.cath = CathTable().get_cath_from_page(x) + self.cath_obj = self.cath.get("cath_name") + self.cath_id = self.cath.get("id") else: return "1" - print("Выбрана категория: " + self.cath_obj) - print("Блюда:") - self.dish_page = 1 - lst = DishTable().get_page_dish(self.cath_id, self.dish_page) - self.max_len_name = 0 - self.mex_dish_index = len(DishTable().all_by_cath_id(self.cath_id)) - action = '>' - max_page = math.ceil(self.mex_dish_index/10) - lst = DishTable().get_page_dish(self.cath_id, self.dish_page) + self.show_dish_page() + print(f"Выбрана категория: {self.cath_obj}") - while True: - self.dish_arr = [] - - for i in lst: - self.dish_arr.append([i[2], str(i[1]), str(i[4])]) - self.max_len_name = max(self.max_len_name, len(i[2])) - add_func.cls() - print("№" + " "*(len(str(self.mex_dish_index)) + 1)+ "Название" + " "*(self.max_len_name - 4)\ - +"Время приготовления Краткая инструкция\ - \n-------------------------------------------------------------------------------------------") - - for i in range(len(self.dish_arr)): - txt = add_func.add_zero_before(str((i+1)+(10*(self.dish_page-1))), len(str(self.mex_dish_index))) + " "*(2+len(str(self.mex_dish_index))) - txt += self.dish_arr[i][0] + " "*(4+self.max_len_name - len(self.dish_arr[i][0])) - txt += self.dish_arr[i][1] + " "*(5+19-len(self.dish_arr[i][1])) - txt += self.dish_arr[i][2] - print(txt) - action = input('-------------------------------------------------------------------------------------------\n\ - Для переключения между страницами используйте "<" и ">"\nКол-во записей на странице: 10\n\ - Для выхода в меню действий нажмите любую другую клавишу\n=> ') - if(action=='>'): - if(self.dish_page==max_page): - self.dish_page = 1 - else: - self.dish_page += 1 - lst = DishTable().get_page_dish(self.cath_id, self.dish_page) - elif(action=='<'): - if(self.dish_page==1): - self.dish_page = max_page - else: - self.dish_page -= 1 - lst = DishTable().get_page_dish(self.cath_id, self.dish_page) - else: - break menu = """-------------------------------------------------------------------------------------------\nДальнейшие операции: 0 - возврат в главное меню; diff --git a/0.8_indev/tables/__pycache__/cath_table.cpython-311.pyc b/0.8_indev/tables/__pycache__/cath_table.cpython-311.pyc index f24a087..424a482 100644 Binary files a/0.8_indev/tables/__pycache__/cath_table.cpython-311.pyc and b/0.8_indev/tables/__pycache__/cath_table.cpython-311.pyc differ diff --git a/0.8_indev/tables/__pycache__/dish_table.cpython-311.pyc b/0.8_indev/tables/__pycache__/dish_table.cpython-311.pyc index 7cc03d3..c4ce770 100644 Binary files a/0.8_indev/tables/__pycache__/dish_table.cpython-311.pyc and b/0.8_indev/tables/__pycache__/dish_table.cpython-311.pyc differ diff --git a/0.8_indev/tables/cath_table.py b/0.8_indev/tables/cath_table.py index 7983b69..bc54bbe 100644 --- a/0.8_indev/tables/cath_table.py +++ b/0.8_indev/tables/cath_table.py @@ -37,20 +37,14 @@ class CathTable(DbTable): print(f'''Выбрана категория для изменения: {old}''') cur = self.dbconn.conn.cursor() - data = input("Введите название (1 - отмена): ").strip() - if data == "1": - return + data = input("Введите название (enter - отмена): ").strip() while((len(data.strip()) == 0)or(len(data.strip()) > 32)or(self.check_by_name(data.strip()))): if (len(data.strip()) > 32): - data = input("Название слишком длинное! Введите название заново (1 - отмена):").strip() - if data == "1": - return - elif len(data.strip()) == 0: - data = input("Название не может быть пустым! Введите название заново (1 - отмена):").strip() - if data == "1": - return + data = input("Название слишком длинное! Введите название заново (enter - отмена):").strip() + elif(self.check_by_name(data.strip()) == 1): + data = input("Такое название уже существует. Введите новое (enter - отмена):").strip() else: - data = input("Такое название уже существует. Введите новое (1 - отмена):").strip() + return "1" param_sql = f"UPDATE cath SET cath_name = '{data}' WHERE cath_name = '{old}';" cur.execute(param_sql) @@ -80,7 +74,7 @@ class CathTable(DbTable): final_list = [] cur = self.dbconn.conn.cursor() offset = (page_num - 1) * ROW_PER_PAGE - sql = (f"SELECT * FROM cath LIMIT {ROW_PER_PAGE} OFFSET {offset}") + sql = (f"SELECT * FROM {self.table_name()} LIMIT {ROW_PER_PAGE} OFFSET {offset}") cur.execute(sql) zero_list = cur.fetchall() for i in range(len(zero_list)): diff --git a/0.8_indev/tables/dish_table.py b/0.8_indev/tables/dish_table.py index a683180..7f70c1a 100644 --- a/0.8_indev/tables/dish_table.py +++ b/0.8_indev/tables/dish_table.py @@ -2,6 +2,9 @@ from dbtable import * import add_func +import math + +ROW_PER_PAGE = 10 class DishTable(DbTable): def table_name(self): @@ -17,8 +20,6 @@ class DishTable(DbTable): def table_constraints(self): return ['CONSTRAINT "Name Dish" UNIQUE (dish_name)'] - - def find_by_position(self, num): sql = "SELECT * FROM " + self.table_name() sql += " ORDER BY " @@ -37,22 +38,67 @@ class DishTable(DbTable): cur.execute(sql) return cur.fetchall() - def get_page_dish(self, cath_id, page_num): - row_per_page = 10 - offset = (page_num - 1) * row_per_page - sql = f"""SELECT * FROM dish WHERE cath_id = {str(cath_id)}\ - ORDER BY {", ".join(self.primary_key())} LIMIT {row_per_page} OFFSET {offset}""" - cur = self.dbconn.conn.cursor() - cur.execute(sql) - return cur.fetchall() - - def delete(self, val): - sql = "DELETE FROM dish" - sql += " WHERE dish_name" - sql += "=" + "'" + val + "';" +# def get_page_dish(self, cath_id, page_num): +# offset = (page_num - 1) * ROW_PER_PAGE +# sql = f"""SELECT * FROM dish WHERE cath_id = {str(cath_id)}\ +# ORDER BY {", ".join(self.primary_key())} LIMIT {ROW_PER_PAGE} OFFSET {offset}""" +# cur = self.dbconn.conn.cursor() +# cur.execute(sql) +# return cur.fetchall() + + def delete(self, id): + sql = f"""DELETE FROM {self.table_name()} WHERE id = {id};""" cur = self.dbconn.conn.cursor() cur.execute(sql) self.dbconn.conn.commit() + + def count(self, id): + sql = f"""SELECT COUNT(*) FROM {self.table_name()} WHERE cath_id = {id};""" + cur = self.dbconn.conn.cursor() + cur.execute(sql) + return int(cur.fetchone()[0]) + + def get_dish_from_page(self, row_num): + arr = self.get_dish_page(math.ceil(row_num/ROW_PER_PAGE), self.cath) + row = dict() + for i in arr: + if(int(i[0]) == row_num): + row["id"] = i[1] + row["dish_name"] = str(i[2]) + row["cath_id"] = i[3] + row["cook_time"] = str(i[4]) + row["manual"] = str(i[5]) + return row + + def get_dish_page(self, cath_id, page_num): + """Функция для получения "страницы" из БД по её номеру. + В программе заменяет .all() для вывода информации. + + Args: + page_num (int): номер страницы, определяется пользователем. + cath_id (int): id категории, из которой необходимо получить блюда + + Returns: + lst (list): Список, содержащий в себе все строки для этой страницы. + + 0 - number + 1 - id + 2 - name + 3 - cath_id + 4 - cook_time + 5 - manual + """ + final_list = [] + cur = self.dbconn.conn.cursor() + offset = (page_num - 1) * ROW_PER_PAGE + sql = (f"SELECT * FROM {self.table_name()} WHERE cath_id = {cath_id} LIMIT {ROW_PER_PAGE} OFFSET {offset}") + cur.execute(sql) + zero_list = cur.fetchall() + for i in range(len(zero_list)): + final_list.append([str(i+1+(ROW_PER_PAGE*(page_num-1))), zero_list[i][3],\ + str(zero_list[i][2]), zero_list[i][0], str(zero_list[i][1]), str(zero_list[i][4])]) + return final_list + def check_by_name(self, value): sql = f"SELECT * FROM {self.table_name()} WHERE dish_name='{value}'" @@ -83,99 +129,84 @@ class DishTable(DbTable): def insert_dishone(self, cath_id): - ins_name = input('Введите название добавляемого блюда (1 - для отмены): ') + ins_name = input('Введите название добавляемого блюда (1 - для отмены): ').strip() - while (ins_name.strip() == '')or(len(ins_name.strip()) > 32)\ + while (ins_name == '')or(len(ins_name) > 32)\ or(DishTable().check_by_name(ins_name)): - if(ins_name.strip() == ''): - ins_name = input("\nПустая строка. Повторите ввод! Укажите название удаляемого блюда (0 - отмена): ") - if ins_name == "0": - return "1" - - elif(len(ins_name.strip()) > 32): + if(len(ins_name) > 32): ins_name = input("\nСлишком длинная строка. Повторите ввод!\ - Укажите название удаляемого блюда (0 - отмена): ") - if ins_name == "0": + Укажите название добавляемого блюда (enter - отмена): ").strip() + if ins_name == "": return "1" else: print('\nТакое блюдо уже существует') - ins_name = input("Повторите ввод! Укажите название блюда (0 - отмена): ") - if ins_name == "0": + ins_name = input("Повторите ввод! Укажите название блюда (enter - отмена): ").strip() + if ins_name == "": return "1" ins_time = input(f'\nПроцесс добавления блюда: {ins_name}\ - \nВведите время приготовления в формате 10 minutes/1 hour 5 minutes (1 - для отмены): ') - while(ins_time==0)or(add_func.validate_time_format(ins_time)==False): - if ins_time == "0": + \nВведите время приготовления в формате 10 minutes/1 hour 5 minutes (enter - для отмены): ').strip() + if(ins_time==''): + return "1" + while(add_func.validate_time_format(ins_time)==False): + ins_time = input('Время введено в неверном формате. Повторите попытку.\n\ +Введите время приготовления в формате 10 minutes/1 hour 5 minutes (enter - для отмены): ').strip() + if(ins_time==''): return "1" - else: - ins_time = input('Время введено в неверном формате. Повторите попытку.\n\ -Введите время приготовления в формате 10 minutes/1 hour 5 minutes (1 - для отмены): ') - + ins_manual = input(f'\nПроцесс добавления блюда: {ins_name}\ - \nВведите краткую инструкцию приготовления блюда (1 - для отмены): ') - if ins_manual == "0": + \nВведите краткую инструкцию приготовления блюда (enter - для отмены): ').strip() + if ins_manual == "": return "1" - insert = [cath_id, ins_time, ins_name, ins_manual] - self.insert_one(insert) - def update_dish(self, dish): + def update_dish(self, obj: dict): print('После изменения выберите другой пункт для изменения или введите 0 для выхода') - n = dish x = -10000 - obj_id = self.id_by_name(dish)[1] + dish_id = obj.get("id") + dish_name = obj.get("dish_name") + tmp = dish_name while True: - x = add_func.validate_input(f'\nВыбранное блюдо: {n}\nЧто вы хотите изменить в рецепте?\n\ + x = add_func.validate_input(f'\nВыбранное блюдо: {tmp}\nЧто вы хотите изменить в рецепте?\n\ 1 - Название\n2 - Время приготовления\n3 - Инструкция\n0 - для отмены\n=> ', 0, 3) if(x==1): - new_name = input(f"Текущее название {n}.\nВведите новое название (0 - для отмены): ") + new_name = input(f"Текущее название {tmp}.\nВведите новое название (enter - для отмены): ") while (new_name.strip() == '')or(len(new_name.strip()) > 32)\ - or(DishTable().check_by_name(new_name))or(x=='0'): - - if(new_name.strip() == ''): - new_name = input("Пустая строка. Повторите ввод! Укажите название блюда (0 - отмена): ") - if new_name == "0": - break - - elif(DishTable().check_by_name(new_name)): + or(DishTable().check_by_name(new_name)): + if(DishTable().check_by_name(new_name)): print('Такое блюдо уже существует') - new_name = input("Повторите ввод! Укажите название блюда (0 - отмена): ") - if new_name == "0": - break + new_name = input("Повторите ввод! Укажите название блюда (enter - отмена): ") elif(len(new_name.strip()) > 32): new_name = input("Слишком длинная строка. Повторите ввод!\ - Укажите название блюда (0 - отмена): ") - if new_name == "0": - break + Укажите название блюда (enter - отмена): ") else: break - # dish = new_name - n = new_name + tmp = new_name new_name = "'" + new_name + "'" - self.update('dish_name', new_name, obj_id) + self.update('dish_name', new_name, dish_id) elif(x==2): - new_time = input(f'Введите время приготовления в формате 10 minutes/1 hour 5 minutes (0 - для отмены): ') - while(new_time==0)or(add_func.validate_time_format(new_time)==False): - if new_time == "0": + new_time = input(f'Введите время приготовления в формате 10 minutes/1 hour 5 minutes (enter - для отмены): ') + while(new_time=='')or(add_func.validate_time_format(new_time)==False): + if new_time == '': break else: - new_time = input('Введите время приготовления в формате 10 minutes/1 hour 5 minutes (0 - для отмены): ') + new_time = input('Введите время приготовления в формате 10 minutes/1 hour 5 minutes\ + (enter - для отмены): ') new_time = "'" + new_time + "'" - self.update('cook_time', new_time, obj_id) + self.update('cook_time', new_time, dish_id) elif(x==3): - new_manual = input(f'Введите краткую инструкцию приготовления блюда (0 - для отмены): ') - if new_manual == "0": + new_manual = input(f'Введите краткую инструкцию приготовления блюда (enter - для отмены): ') + if new_manual == "": break new_manual = "'" + new_manual + "'" - self.update('manual', new_manual, obj_id) + self.update('manual', new_manual, dish_id) elif(x==-1): break @@ -274,4 +305,5 @@ class DishTable(DbTable): self.insert_one([1, "1 hour 20 minutes", "Паста с песто из авокадо", "Паста с соусом из авокадо, базилика и сыра."]) self.insert_one([1, "30 minutes", "Яичница с авокадо", "Яичница с добавлением авокадо и других ингредиентов."]) self.insert_one([1, "55 minutes", "Веганский пад-тай", "Пад-тай с использованием растительных ингредиентов."]) - return \ No newline at end of file + return + \ No newline at end of file