Update webhook_handler.py

webhook_handler.py

@@ -1,6 +1,6 @@
 # webhook_handler.py
 
-import sqlite3
+
 
 def handle_webhook(data, key_callback_vk, db_name):
     # Проверяем тип события
@@ -19,15 +19,20 @@ def handle_webhook(data, key_callback_vk, db_name):
         cursor = conn.cursor()
 
         # Проверяем, существует ли запись для данного пользователя
-        cursor.execute("SELECT * FROM contacts WHERE vk_id = ?", (user_id,))
+        cursor.execute("SELECT canal FROM contacts WHERE vk_id = ?", (user_id,))
         result = cursor.fetchone()
 
         if result:
             # Обновляем запись, если пользователь уже существует
-            cursor.execute("UPDATE contacts SET canal = 1 WHERE vk_id = ?", (user_id,))
+            canal_data = json.loads(result[0]) if result[0] else {}
+            canal_data[str(group_id)] = 1
+            canal_json = json.dumps(canal_data)
+            cursor.execute("UPDATE contacts SET canal = ? WHERE vk_id = ?", (canal_json, user_id))
         else:
             # Создаем новую запись, если пользователь не существует
-            cursor.execute("INSERT INTO contacts (vk_id, canal) VALUES (?, 1)", (user_id,))
+            canal_data = {str(group_id): 1}
+            canal_json = json.dumps(canal_data)
+            cursor.execute("INSERT INTO contacts (vk_id, canal) VALUES (?, ?)", (user_id, canal_json))
 
         conn.commit()
         conn.close()
@@ -45,15 +50,20 @@ def handle_webhook(data, key_callback_vk, db_name):
         cursor = conn.cursor()
 
         # Проверяем, существует ли запись для данного пользователя
-        cursor.execute("SELECT * FROM contacts WHERE vk_id = ?", (user_id,))
+        cursor.execute("SELECT canal FROM contacts WHERE vk_id = ?", (user_id,))
         result = cursor.fetchone()
 
         if result:
             # Обновляем запись, если пользователь уже существует
-            cursor.execute("UPDATE contacts SET canal = 0 WHERE vk_id = ?", (user_id,))
+            canal_data = json.loads(result[0]) if result[0] else {}
+            canal_data[str(group_id)] = 0
+            canal_json = json.dumps(canal_data)
+            cursor.execute("UPDATE contacts SET canal = ? WHERE vk_id = ?", (canal_json, user_id))
         else:
             # Создаем новую запись, если пользователь не существует
-            cursor.execute("INSERT INTO contacts (vk_id, canal) VALUES (?, 0)", (user_id,))
+            canal_data = {str(group_id): 0}
+            canal_json = json.dumps(canal_data)
+            cursor.execute("INSERT INTO contacts (vk_id, canal) VALUES (?, ?)", (user_id, canal_json))
 
         conn.commit()
         conn.close()