0.3(добавление функции добавки блюда)

tables/cath_table.py

@@ -15,23 +15,29 @@ class CathTable(DbTable):
     
     def delete(self, val):
         sql = "DELETE FROM " + self.table_name()
-        sql += " WHERE " + "".join(self.column_names_without_id())
+        sql += " WHERE cath_name"
         sql += "=" + "'" + "".join(val) + "';"
         cur = self.dbconn.conn.cursor()
         cur.execute(sql)
         self.dbconn.conn.commit()
     
     def find_by_name(self, name):
+        flag = 0
         cur = self.dbconn.conn.cursor()
-        
         sql_sel = "SELECT id FROM " + self.table_name()
         sql_sel += " WHERE cath_name = " + "'" + name + "'" + ";"
-        cur.execute(sql_sel) 
+        cur.execute(sql_sel)            
+        ret = cur.fetchone()
+        # print(ret)
+        if(ret != None):
+            flag = 1
+            return [flag, list(ret)[0]]
+        else:
+            return [flag, list(ret)[0]]   
         # sql = "SELECT * FROM " + self.table_name()
         # sql += " ORDER BY "
         # sql += ", ".join(self.primary_key())
         # sql += " LIMIT 1 OFFSET %(offset)s"
         # cur.execute(sql, {"offset": num - 1})
-        # print(cur)
-        return cur.fetchone()       
+        # print(cur)    
     

tables/dish_table.py

@@ -1,6 +1,7 @@
 # Таблица с блюдами и действия с ними
 
 from dbtable import *
+import add_func
 
 class DishTable(DbTable):
     def table_name(self):
@@ -27,3 +28,75 @@ class DishTable(DbTable):
         cur.execute(sql, {"offset": num - 1})
         return cur.fetchone()       
     
+    def all_by_cath_id(self, cath_id):
+        sql = "SELECT * FROM " + self.table_name()
+        sql += " WHERE cath_id = " + str(cath_id)
+        sql += " ORDER BY "
+        sql += ", ".join(self.primary_key())
+        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 + "';"
+        cur = self.dbconn.conn.cursor()
+        cur.execute(sql)
+        self.dbconn.conn.commit()
+        
+    def check_by_name(self, value):
+        sql = f"SELECT * FROM {self.table_name()} WHERE dish_name='{value}'" 
+        cur = self.dbconn.conn.cursor()
+        cur.execute(sql)
+        result = cur.fetchone()
+        cur.close()
+        if result:
+            return True
+        else:
+            return False
+    
+    def insert_dishone(self, cath_id):
+        
+        ins_name = input('Введите название добавляемого блюда (1 - для отмены): ')
+        
+        while (ins_name.strip() == '')or(len(ins_name.strip()) > 32)\
+            or(add_func.is_cyr_or_dig(ins_name.strip())==0)or(DishTable().check_by_name(ins_name)):
+                
+            if(ins_name.strip() == ''):
+                ins_name = input("Пустая строка. Повторите ввод! Укажите название удаляемого блюда (0 - отмена): ")
+                if ins_name == "0":
+                    return "1"
+                
+            elif(len(ins_name.strip()) > 32):
+                ins_name = input("Слишком длинная строка. Повторите ввод!\
+                    Укажите название удаляемого блюда (0 - отмена): ")
+                if ins_name == "0":
+                    return "1"
+                
+            elif(add_func.is_cyr_or_dig(ins_name.strip())==0):
+                ins_name = input("Название должно состоять только из символов кириллицы. Повторите ввод!\
+                    Укажите название удаляемого блюда (0 - отмена): ")
+                if ins_name == "0":
+                    return "1"
+                
+            else:
+                print('Такое блюдо уже существует')
+                ins_name = input("Повторите ввод! Укажите название блюда (0 - отмена): ")
+                if ins_name == "0":
+                    return "1"
+        
+        ins_time = input(f'Процесс добавления блюда: {ins_name}\
+            \nВведите время приготовления в формате 10 minutes/1 hour 5 minutes (1 - для отмены): ')
+        if ins_time == "0":
+            return "1"
+        ins_manual = input(f'Процесс добавления блюда: {ins_name}\
+            \nВведите краткую инструкцию приготовления блюда (1 - для отмены): ')
+        if ins_time == "0":
+            return "1"
+        
+        insert = [cath_id, ins_time, ins_name, ins_manual]
+        DishTable().insert_one(insert)
+        
+        
+