Update app.py

app.py

@@ -101,22 +101,6 @@ for db in DATABASES:
 
 
 
-template = {
-    "username": "name",
-    "phone": "phone",
-    "email": "email",
-    "city": "b_city",
-    "finished": "b_fin",
-    "ban": "b_ban",
-    "ignore": "b_ign",
-    "banners": "b_baners",  # Если массив заполнен пишем -1, если пустой пишем - 0
-    "buttons": "b_butt",    # Если массив заполнен пишем -1, если пустой пишем - 0
-    "messages": "b_mess",
-    "0": "web_st"  # С первого раза и с каждым обновлением пользователя, увеличиваем на 1
-}
-
-
-
 
 
 
@@ -775,81 +759,94 @@ def send_request():
 
 
 
-def update_or_insert_user(db_name, user_data, mapping_template):
-    conn = sqlite3.connect(db_name)
-    cursor = conn.cursor()
-    
-    email = user_data.get('email')
-    
-    # Проверка существования пользователя
-    cursor.execute("SELECT * FROM contacts WHERE email = ?", (email,))
-    user = cursor.fetchone()
-
-    # Преобразование данных пользователя на основе шаблона
-    transformed_data = {}
-    for json_key, db_column in mapping_template.items():
-        value = user_data.get(json_key, "")
-        if isinstance(value, list):
-            transformed_data[db_column] = "-1" if value else "0"
-        else:
-            transformed_data[db_column] = value
-    
-    if user:
-        # Обновление существующего пользователя
-        update_query = "data_gc.db"
-        update_values = []
-        for column, value in transformed_data.items():
-            update_query += f"{column} = ?, "
-            update_values.append(value)
-        update_query = update_query.rstrip(", ") + " WHERE email = ?"
-        update_values.append(email)
-        cursor.execute(update_query, update_values)
-    else:
-        # Добавление нового пользователя
-        columns = ', '.join(transformed_data.keys())
-        placeholders = ', '.join('?' for _ in transformed_data)
-        insert_query = f"INSERT INTO contacts ({columns}) VALUES ({placeholders})"
-        cursor.execute(insert_query, list(transformed_data.values()))
-
-    conn.commit()
-    conn.close()
-
 @app.route('/send_get_request', methods=['GET'])
 def send_get_request():
     token = request.args.get('token')
     webinarId = request.args.get('webinarId')
     url = f'https://online.bizon365.ru/api/v1/webinars/reports/get?webinarId={webinarId}'
-
     response = requests.get(url, headers={'X-Token': token})
-
     if response.status_code == 200:
         data = response.json()
-
         report = data.get('report', {})
-        messages = data.get('messages', {})
-
         report_json_str = report.get('report', '{}')
+        messages_json_str = report.get('messages', '{}')
+        
         try:
             report_json = json.loads(report_json_str)
         except json.JSONDecodeError:
             report_json = {}
 
-        messages_json_str = report.get('messages', '{}')
         try:
             messages_json = json.loads(messages_json_str)
         except json.JSONDecodeError:
             messages_json = {}
 
-        users_meta = report_json.get('usersMeta', {})
+        result = {
+            'report': report_json,
+            'messages': messages_json
+        }
+        
+        # Маппинг полей
+        template = {
+            "username": "name",
+            "phone": "phone",
+            "email": "email",
+            "city": "b_city", 
+            "finished": "b_fin",
+            "ban": "b_ban",
+            "ignore": "b_ign",
+            "banners": "b_baners", # Если массив заполнен пишем -1, если пустой пишем - 0
+            "buttons": "b_butt",   # Если массив заполнен пишем -1, если пустой пишем - 0
+            "messages": "b_mess",
+            "0": "web_st",  # С первого раза и с каждым обновлением пользователя, увеличиваем на 1
+        }
 
-        # Обновление или добавление каждого пользователя во все базы данных
+        # Получение usersMeta и email
+        users_meta = report_json.get('usersMeta', {})
         for user_id, user_data in users_meta.items():
-            user_data['messages'] = messages_json
+            email = user_data.get('email', '')
 
-            for db in DATABASES:
-                update_or_insert_user(db, user_data, template)
+            if not email:
+                continue
+
+            # Проверка наличия пользователя по email
+            conn = sqlite3.connect('data_gc.db')
+            cursor = conn.cursor()
+            cursor.execute("SELECT * FROM contacts WHERE email = ?", (email,))
+            existing_user = cursor.fetchone()
+
+            # Подготовка данных для вставки/обновления
+            mapped_data = {}
+            for key, value in template.items():
+                if key in user_data:
+                    if key in ["banners", "buttons"]:
+                        mapped_data[value] = -1 if user_data[key] else 0
+                    elif key == "0":
+                        if existing_user:
+                            mapped_data[value] = existing_user[8] + 1  # Увеличиваем на 1 значение поля web_st
+                        else:
+                            mapped_data[value] = 1  # Изначальное значение
+                    else:
+                        mapped_data[value] = user_data[key]
+                elif key == "messages":
+                    mapped_data[value] = json.dumps(messages_json.get(user_id, []))
+
+            # Вставка или обновление пользователя
+            if existing_user:
+                placeholders = ", ".join([f"{field} = ?" for field in mapped_data.keys()])
+                values = list(mapped_data.values())
+                values.append(email)
+                cursor.execute(f"UPDATE contacts SET {placeholders} WHERE email = ?", values)
+            else:
+                fields = ", ".join(mapped_data.keys())
+                placeholders = ", ".join(["?" for _ in mapped_data.values()])
+                values = list(mapped_data.values())
+                cursor.execute(f"INSERT INTO contacts ({fields}) VALUES ({placeholders})", values)
+
+            conn.commit()
+            conn.close()
 
-        return jsonify({'status': 'User data saved successfully'})
+        return jsonify(result)
     else:
         return jsonify({'error': 'Failed to fetch data from the API'}), response.status_code