DmitrMakeev commited on
Commit
e4435ed
·
verified ·
1 Parent(s): d5377f0

Update webhook_handler.py

Browse files
Files changed (1) hide show
  1. webhook_handler.py +11 -2
webhook_handler.py CHANGED
@@ -40,11 +40,20 @@ def handle_webhook(data, key_callback_vk, db_name):
40
  group_id = data['group_id']
41
  print(f"User {user_id} unsubscribed from the mailing list of group {group_id}")
42
 
43
- # Удаляем информацию о подписке из базы данных
44
  conn = sqlite3.connect(db_name)
45
  cursor = conn.cursor()
46
 
47
- cursor.execute("UPDATE contacts SET canal = NULL WHERE vk_id = ?", (user_id,))
 
 
 
 
 
 
 
 
 
48
 
49
  conn.commit()
50
  conn.close()
 
40
  group_id = data['group_id']
41
  print(f"User {user_id} unsubscribed from the mailing list of group {group_id}")
42
 
43
+ # Записываем информацию об отписке в базу данных
44
  conn = sqlite3.connect(db_name)
45
  cursor = conn.cursor()
46
 
47
+ # Проверяем, существует ли запись для данного пользователя
48
+ cursor.execute("SELECT * FROM contacts WHERE vk_id = ?", (user_id,))
49
+ result = cursor.fetchone()
50
+
51
+ if result:
52
+ # Обновляем запись, если пользователь уже существует
53
+ cursor.execute("UPDATE contacts SET canal = 0 WHERE vk_id = ?", (user_id,))
54
+ else:
55
+ # Создаем новую запись, если пользователь не существует
56
+ cursor.execute("INSERT INTO contacts (vk_id, canal) VALUES (?, 0)", (user_id,))
57
 
58
  conn.commit()
59
  conn.close()