proj0.1

dbtable.py

@@ -24,7 +24,6 @@ class DbTable:
         res = sorted(self.columns().keys(), key = lambda x: x)
         if 'id' in res:
             res.remove('id')
-            
         return res
 
     def table_constraints(self):
@@ -40,6 +39,14 @@ class DbTable:
         self.dbconn.conn.commit()
         return
     
+    def delete(self, val):
+        sql = "DELETE FROM " + self.table_name()
+        sql += " WHERE " + "".join(self.column_names_without_id())
+        sql += "=" + "'" + "".join(val) + "';"
+        cur = self.dbconn.conn.cursor()
+        cur.execute(sql)
+        self.dbconn.conn.commit()
+
     def drop(self):
         sql = "DROP TABLE IF EXISTS " + self.table_name()
         cur = self.dbconn.conn.cursor()

main.py

@@ -60,14 +60,6 @@ class Main:
         cth.drop()
         return
 
-#     def show_main_menu(self):
-#         menu = """Добро пожаловать! 
-# Основное меню (выберите цифру в соответствии с необходимым действием): 
-#     1 - просмотр людей;
-#     2 - сброс и инициализация таблиц;
-#     9 - выход."""
-#         print(menu)
-#         return
     def show_main_menu(self):
         menu = """Привутствуем в меню, выберите действие:
         1 - просмотр категорий;
@@ -90,25 +82,87 @@ class Main:
             print("Выбрано неверное число! Повторите ввод!")
             return "0"
         else:
-            return next_step
+            return next_step;
             
-    def show_people(self):
+    def show_cath(self):
-        self.person_id = -1
+        self.cath_id = -1
-        menu = """Просмотр списка блюд!
+        menu = """Просмотр списка категорий!
-№\tНазвание\tВремя приготовления\tКатегория"""
+№\tНазвание\n------------------------"""
         print(menu)
-        lst = DishTable().all()
+        lst = CathTable().all()
         for i in lst:
-            print(str(i[3]) + "\t" + str(i[2]) + "\t\t" + str(i[1]) + "\t\t\t" + str(i[0]))
+            print(str(i[1]) + "\t" + str(i[0]))
-        menu = """Дальнейшие операции: 
+        menu = """------------------------\nДальнейшие операции: 
     0 - возврат в главное меню;
-    3 - добавление нового человека;
+    3 - добавление новой категории;
-    4 - удаление человека;
+    4 - удаление категории;
-    5 - просмотр телефонов человека;
+    5 - просмотр блюд в категории;
     9 - выход."""
         print(menu)
         return
     
+    def after_show_cath(self, next_step):
+        while True:
+            if next_step == "4":
+                CathTable().delete(input('Введите название удаляемой категории: '))
+                return "1"
+            elif next_step == "6" or next_step == "7":
+                print("Пока не реализовано!")
+                next_step = "5"
+            elif next_step == "5":
+                next_step = self.show_dish_in_cath()
+            elif next_step != "0" and next_step != "9" and next_step != "3":
+                print("Выбрано неверное число! Повторите ввод!")
+                return "1"
+            else:
+                return next_step
+
+    def show_add_cath(self):
+        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
+            else:
+                data[0] = input("Название не может быть пустым! Введите название заново (1 - отмена):").strip()
+                if data[0] == "1":
+                    return
+        CathTable().insert_one(data)
+        return
+    
+    def show_dish_in_cath(self):
+        if self.cath_id == -1:
+            while True:
+                name = input("Укажите название категории для вывода блюд (0 - отмена):")
+                while len(name.strip()) == 0:
+                    name = input("Пустая строка. Повторите ввод! Укажите название категории для вывода блюд (0 - отмена):")
+                if name == "0":
+                    return "1"
+                cath = CathTable().find_by_position(int(name))
+                if not cath:
+                    print("Введено неверное название!")
+                else:
+                    self.cath_id = int(cath[1])
+                    self.cath_obj = cath
+                    break
+        print("Выбрана категория: " + self.cath_obj[1])
+        print("Блюда:")
+        lst = DishTable().all_by_cath_id(self.cath_id)
+        for i in lst:
+            print(i[1])
+        menu = """Дальнейшие операции:
+    0 - возврат в главное меню;
+    1 - возврат в просмотр категорий;
+    6 - добавление нового блюда;
+    7 - удаление блюда;
+    9 - выход."""
+        print(menu)
+        return self.read_next_step()
+
     def main_cycle(self):
         current_menu = "0"
         next_step = None
@@ -118,13 +172,13 @@ class Main:
                 next_step = self.read_next_step()
                 current_menu = self.after_main_menu(next_step)
             elif current_menu == "1":
-                self.show_people()
+                self.show_cath()
                 next_step = self.read_next_step()
-                current_menu = self.after_show_people(next_step)
+                current_menu = self.after_show_cath(next_step)
             elif current_menu == "2":
                 self.show_main_menu()
             elif current_menu == "3":
-                self.show_add_person()
+                self.show_add_cath()
                 current_menu = "1"
         print("До свидания!")    
         return
@@ -133,81 +187,7 @@ class Main:
             DbTable.dbconn.test()
 
 m = Main()
-
 # Откоментируйте эту строку и закоментируйте следующую для теста
 # соединения с БД
 # m.test()
 m.main_cycle() 
-
-
-    # def after_show_people(self, next_step):
-    #     while True:
-    #         if next_step == "4":
-    #             print("Пока не реализовано!")
-    #             return "1"
-    #         elif next_step == "6" or next_step == "7":
-    #             print("Пока не реализовано!")
-    #             next_step = "5"
-    #         elif next_step == "5":
-    #             next_step = self.show_phones_by_people()
-    #         elif next_step != "0" and next_step != "9" and next_step != "3":
-    #             print("Выбрано неверное число! Повторите ввод!")
-    #             return "1"
-    #         else:
-    #             return next_step
-
-    # def show_add_person(self):
-    #     # Не реализована проверка на максимальную длину строк. Нужно доделать самостоятельно!
-    #     data = []
-    #     data.append(input("Введите имя (1 - отмена): ").strip())
-    #     if data[0] == "1":
-    #         return
-    #     while len(data[0].strip()) == 0:
-    #         data[0] = input("Имя не может быть пустым! Введите имя заново (1 - отмена):").strip()
-    #         if data[0] == "1":
-    #             return
-    #     data.append(input("Введите фамилию (1 - отмена): ").strip())
-    #     if data[1] == "1":
-    #         return
-    #     while len(data[1].strip()) == 0:
-    #         data[1] = input("Фамилия не может быть пустой! Введите фамилию заново (1 - отмена):").strip()
-    #         if data[1] == "1":
-    #             return
-    #     data.append(input("Введите отчество (1 - отмена):").strip())
-    #     if data[2] == "1":
-    #         return
-    #     PeopleTable().insert_one(data)
-    #     return
-
-    
-    # def show_phones_by_people(self):
-    #     if self.person_id == -1:
-    #         while True:
-    #             num = input("Укажите номер строки, в которой записана интересующая Вас персона (0 - отмена):")
-    #             while len(num.strip()) == 0:
-    #                 num = input("Пустая строка. Повторите ввод! Укажите номер строки, в которой записана интересующая Вас персона (0 - отмена):")
-    #             if num == "0":
-    #                 return "1"
-    #             person = PeopleTable().find_by_position(int(num))
-    #             if not person:
-    #                 print("Введено число, неудовлетворяющее количеству людей!")
-    #             else:
-    #                 self.person_id = int(person[1])
-    #                 self.person_obj = person
-    #                 break
-    #     print("Выбран человек: " + self.person_obj[2] + " " + self.person_obj[0] + " " + self.person_obj[3])
-    #     print("Телефоны:")
-    #     lst = PhonesTable().all_by_person_id(self.person_id)
-    #     for i in lst:
-    #         print(i[1])
-    #     menu = """Дальнейшие операции:
-    # 0 - возврат в главное меню;
-    # 1 - возврат в просмотр людей;
-    # 6 - добавление нового телефона;
-    # 7 - удаление телефона;
-    # 9 - выход."""
-    #     print(menu)
-    #     return self.read_next_step()
-
-
-    

tables/cath_table.py

@@ -15,6 +15,7 @@ class CathTable(DbTable):
     
     
     def find_by_position(self, num):
+        
         sql = "SELECT * FROM " + self.table_name()
         sql += " ORDER BY "
         sql += ", ".join(self.primary_key())