Update webhook_handler.py

webhook_handler.py

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