Update webhook_handler.py

webhook_handler.py

@@ -1,8 +1,9 @@
 # webhook_handler.py
 import sqlite3
 import json
+import requests
 
-def handle_webhook(data, key_callback_vk, db_name):
+def handle_webhook(data, key_callback_vk, db_name, vk_st_alone):
     # Преобразуем строку key_callback_vk в JSON
     key_callback_dict = json.loads(key_callback_vk)
 
@@ -24,20 +25,20 @@ def handle_webhook(data, key_callback_vk, db_name):
         cursor = conn.cursor()
 
         # Проверяем, существует ли запись для данного пользователя
-        cursor.execute("SELECT canal FROM contacts WHERE vk_id = ?", (user_id,))
+        cursor.execute("SELECT orders FROM contacts WHERE vk_id = ?", (user_id,))
         result = cursor.fetchone()
 
         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))
+            orders_data = json.loads(result[0]) if result[0] else {}
+            orders_data[str(group_id)] = '1'
+            orders_json = json.dumps(orders_data)
+            cursor.execute("UPDATE contacts SET orders = ? WHERE vk_id = ?", (orders_json, 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))
+            orders_data = {str(group_id): '1'}
+            orders_json = json.dumps(orders_data)
+            cursor.execute("INSERT INTO contacts (vk_id, orders) VALUES (?, ?)", (user_id, orders_json))
 
         conn.commit()
         conn.close()
@@ -55,25 +56,68 @@ def handle_webhook(data, key_callback_vk, db_name):
         cursor = conn.cursor()
 
         # Проверяем, существует ли запись для данного пользователя
-        cursor.execute("SELECT canal FROM contacts WHERE vk_id = ?", (user_id,))
+        cursor.execute("SELECT orders FROM contacts WHERE vk_id = ?", (user_id,))
         result = cursor.fetchone()
 
         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))
+            orders_data = json.loads(result[0]) if result[0] else {}
+            orders_data[str(group_id)] = '0'
+            orders_json = json.dumps(orders_data)
+            cursor.execute("UPDATE contacts SET orders = ? WHERE vk_id = ?", (orders_json, 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))
+            orders_data = {str(group_id): '0'}
+            orders_json = json.dumps(orders_data)
+            cursor.execute("INSERT INTO contacts (vk_id, orders) VALUES (?, ?)", (user_id, orders_json))
 
         conn.commit()
         conn.close()
 
         return 'ok'
 
+    # Обработка события вступления в закрытую группу
+    if data['type'] == 'group_join':
+        user_id = data['object']['user_id']
+        group_id = data['group_id']
+        print(f"User {user_id} requested to join group {group_id}")
+
+        # Читаем информацию из базы данных
+        conn = sqlite3.connect(db_name)
+        cursor = conn.cursor()
+
+        # Проверяем, существует ли запись для данного пользователя
+        cursor.execute("SELECT orders FROM contacts WHERE vk_id = ?", (user_id,))
+        result = cursor.fetchone()
+
+        if result:
+            # Получаем данные о группе
+            orders_data = json.loads(result[0]) if result[0] else {}
+            group_status = orders_data.get(str(group_id), 'not')
+
+            if group_status == 'yes':
+                # Одобряем запрос на вступление
+                approve_join_request(group_id, user_id, vk_st_alone)
+                return 'ok'
+            else:
+                # Отклоняем запрос на вступление
+                return 'not'
+        else:
+            # Если запись не найдена, отклоняем запрос
+            return 'not'
+
+        conn.close()
+
     # Если тип события неизвестен, возвращаем ошибку
-    return 'unknown event', 400
+    return 'unknown event', 400
+
+def approve_join_request(group_id, user_id, vk_st_alone):
+    url = f'https://api.vk.com/method/groups.approveRequest'
+    params = {
+        'group_id': group_id,
+        'user_id': user_id,
+        'access_token': vk_st_alone,
+        'v': '5.131'
+    }
+    response = requests.get(url, params=params).json()
+    return response