Update app.py

app.py

@@ -99,20 +99,6 @@ 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
-}
 
 
 
@@ -777,81 +763,90 @@ 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_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 = {}
-
-        try:
-            messages_json = json.loads(messages_json_str)
-        except json.JSONDecodeError:
-            messages_json = {}
-
-        user_data = report_json.get('usersMeta', {}).get('HXJY5fEdC', {})
-        user_data['messages'] = messages_json
-
-        # Обновление или добавление пользователя во все базы данных
-        for db in DATABASES:
-            update_or_insert_user(db, user_data, template)
-
+        report = data.get('report', {})
+        users_meta = report.get('usersMeta', {})
+        for user_id, user_data in users_meta.items():
+            email = user_data.get('email')
+            if email:
+                update_or_insert_user(email, user_data)
         return jsonify({'status': 'success'})
     else:
         return jsonify({'error': 'Failed to fetch data from the API'}), response.status_code
 
+def update_or_insert_user(email, user_data):
+    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
+    }
+
+    for db in DATABASES:
+        conn = sqlite3.connect(db)
+        cursor = conn.cursor()
+        cursor.execute('SELECT * FROM contacts WHERE email = ?', (email,))
+        existing_user = cursor.fetchone()
+
+        if existing_user:
+            # Обновление данных пользователя
+            update_data = {}
+            for key, value in user_data.items():
+                if key in template:
+                    update_data[template[key]] = value
+            update_data['web_st'] = existing_user[8] + 1 if existing_user[8] is not None else 1
+            update_data['b_baners'] = -1 if user_data.get('banners', []) else 0
+            update_data['b_butt'] = -1 if user_data.get('buttons', []) else 0
+            update_query = 'UPDATE contacts SET ' + ', '.join([f'{k} = ?' for k in update_data.keys()]) + ' WHERE email = ?'
+            cursor.execute(update_query, list(update_data.values()) + [email])
+        else:
+            # Добавление нового пользователя
+            insert_data = {
+                "name": user_data.get('username', ''),
+                "phone": user_data.get('phone', ''),
+                "email": user_data.get('email', ''),
+                "b_city": user_data.get('city', ''),
+                "b_fin": user_data.get('finished', ''),
+                "b_ban": user_data.get('ban', ''),
+                "b_ign": user_data.get('ignore', ''),
+                "b_baners": -1 if user_data.get('banners', []) else 0,
+                "b_butt": -1 if user_data.get('buttons', []) else 0,
+                "b_mess": user_data.get('messages', ''),
+                "web_st": 1,
+                "shop_st": '',
+                "curator": '',
+                "pr1": '',
+                "pr2": '',
+                "pr3": '',
+                "pr4": '',
+                "pr5": '',
+                "ad_url": '',
+                "key_pr": '',
+                "n_con": '',
+                "canal": '',
+                "data_t": ''
+            }
+            insert_query = 'INSERT INTO contacts (' + ', '.join(insert_data.keys()) + ') VALUES (' + ', '.join(['?'] * len(insert_data)) + ')'
+            cursor.execute(insert_query, list(insert_data.values()))
+
+        conn.commit()
+        conn.close()
+