Update app.py

app.py

@@ -99,7 +99,20 @@ for db in DATABASES:
     init_db(db)
 
 
-
+# Шаблон маппинга
+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
+}
 
 
 
@@ -756,49 +769,86 @@ 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 = "UPDATE contacts SET "
+        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', {})
-        
-        # Извлечение полей из отчета
 
-        report_json_str = report.get('report', '{}')
-        
-        # Обработка строкового JSON в report
+        report_json_str = data.get('report', {}).get('report', '{}')
+        messages_json_str = data.get('messages', '{}')
+
         try:
             report_json = json.loads(report_json_str)
         except json.JSONDecodeError:
-            report_json = {}        
-        # Извлечение данных отчета
-        messages = data.get('messages', {})
-        
-        # Извлечение полей из отчета
+            report_json = {}
 
-        messages_json_str = report.get('messages', '{}')
-        
-        # Обработка строкового JSON в report
         try:
             messages_json = json.loads(messages_json_str)
         except json.JSONDecodeError:
-            messages_json = {}       
-        # Возврат результата
-        result = {
+            messages_json = {}
 
-            'report': report_json,
-            'messages': messages_json
+        user_data = report_json.get('usersMeta', {}).get('HXJY5fEdC', {})
+        user_data['messages'] = messages_json
 
-        }
-        
-        return jsonify(result)
+        # Обновление или добавление пользователя во все базы данных
+        for db in DATABASES:
+            update_or_insert_user(db, user_data, template)
+
+        return jsonify({'status': 'success'})
     else:
         return jsonify({'error': 'Failed to fetch data from the API'}), response.status_code
 
@@ -821,20 +871,6 @@ def send_get_request():
 
 
 
-
-
-
-
-
-
-
-
-
-
-
-
-
-