Update webhook_handler.py

webhook_handler.py

@@ -2,7 +2,6 @@
 
 import sqlite3
 import json
-
 def handle_webhook(data, key_callback_vk, db_name):
     # Проверяем тип события
     if data['type'] == 'confirmation':
@@ -25,15 +24,16 @@ def handle_webhook(data, key_callback_vk, db_name):
 
         if result:
             # Обновляем запись, если пользователь уже существует
-            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))
+            canal_data = result[0] if result[0] else ""
+            canal_entries = canal_data.split(',')
+            canal_dict = {entry.split(':')[0]: entry.split(':')[1] for entry in canal_entries if entry}
+            canal_dict[str(group_id)] = '1'
+            canal_data = ','.join([f"{key}:{value}" for key, value in canal_dict.items()])
+            cursor.execute("UPDATE contacts SET canal = ? WHERE vk_id = ?", (canal_data, user_id))
         else:
             # Создаем новую запись, если пользователь не существует
-            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))
+            canal_data = f"{group_id}:1"
+            cursor.execute("INSERT INTO contacts (vk_id, canal) VALUES (?, ?)", (user_id, canal_data))
 
         conn.commit()
         conn.close()
@@ -56,15 +56,16 @@ def handle_webhook(data, key_callback_vk, db_name):
 
         if result:
             # Обновляем запись, если пользователь уже существует
-            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))
+            canal_data = result[0] if result[0] else ""
+            canal_entries = canal_data.split(',')
+            canal_dict = {entry.split(':')[0]: entry.split(':')[1] for entry in canal_entries if entry}
+            canal_dict[str(group_id)] = '0'
+            canal_data = ','.join([f"{key}:{value}" for key, value in canal_dict.items()])
+            cursor.execute("UPDATE contacts SET canal = ? WHERE vk_id = ?", (canal_data, user_id))
         else:
             # Создаем новую запись, если пользователь не существует
-            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))
+            canal_data = f"{group_id}:0"
+            cursor.execute("INSERT INTO contacts (vk_id, canal) VALUES (?, ?)", (user_id, canal_data))
 
         conn.commit()
         conn.close()