Update webhook_handler.py

webhook_handler.py

@@ -2,7 +2,9 @@
 
 # webhook_handler.py
 
-def handle_webhook(data, key_callback_vk):
+import sqlite3
+
+def handle_webhook(data, key_callback_vk, db_name):
     # Проверяем тип события
     if data['type'] == 'confirmation':
         # Возвращаем подтверждение
@@ -13,6 +15,25 @@ def handle_webhook(data, key_callback_vk):
         user_id = data['object']['user_id']
         group_id = data['group_id']
         print(f"User {user_id} subscribed to the mailing list of group {group_id}")
+
+        # Записываем информацию о подписке в базу данных
+        conn = sqlite3.connect(db_name)
+        cursor = conn.cursor()
+
+        # Проверяем, существует ли запись для данного пользователя
+        cursor.execute("SELECT * FROM contacts WHERE vk_id = ?", (user_id,))
+        result = cursor.fetchone()
+
+        if result:
+            # Обновляем запись, если пользователь уже существует
+            cursor.execute("UPDATE contacts SET canal = 1 WHERE vk_id = ?", (user_id,))
+        else:
+            # Создаем новую запись, если пользователь не существует
+            cursor.execute("INSERT INTO contacts (vk_id, canal) VALUES (?, 1)", (user_id,))
+
+        conn.commit()
+        conn.close()
+
         return 'ok'
 
     # Обработка события отписки от рассылки
@@ -20,6 +41,16 @@ def handle_webhook(data, key_callback_vk):
         user_id = data['object']['user_id']
         group_id = data['group_id']
         print(f"User {user_id} unsubscribed from the mailing list of group {group_id}")
+
+        # Удаляем информацию о подписке из базы данных
+        conn = sqlite3.connect(db_name)
+        cursor = conn.cursor()
+
+        cursor.execute("UPDATE contacts SET canal = NULL WHERE vk_id = ?", (user_id,))
+
+        conn.commit()
+        conn.close()
+
         return 'ok'
 
     # Если тип события неизвестен, возвращаем ошибку