Update app.py

app.py

@@ -893,15 +893,47 @@ def send_get_request():
             messages_user = messages_json.get(user_id, [])  # Получение сообщений пользователя
             user_data['messages'] = messages_user  # Добавление сообщений в данные пользователя
 
+            # Проверка наличия email и его уникальности
             email = user_data.get('email')
             if email and email not in processed_emails:
-                update_or_insert_user(DATABASE_NAME, user_data, mapping_template)
-                processed_emails.add(email)
+                update_or_insert_user(DATABASE_NAME, user_data, mapping_template)  # Вставка/обновление пользователя в базе данных
+                processed_emails.add(email)  # Добавление email в множество обработанных
 
         return jsonify({'status': 'User data saved successfully'})
     else:
         return jsonify({'error': 'Failed to fetch data from the API'}), response.status_code
 
+def update_or_insert_user(db_name, user_data, mapping_template):
+    """ Обновляет или вставляет данные пользователя в базу данных. """
+    connection = sqlite3.connect(db_name)
+    cursor = connection.cursor()
+
+    # Преобразование данных пользователя на основе шаблона сопоставления
+    transformed_data = {}
+    for json_key, db_column in mapping_template.items():
+        value = user_data.get(json_key, "")
+        if isinstance(value, list):
+            value = ",".join(value)
+        transformed_data[db_column] = value
+
+    # Вставка или обновление записи пользователя
+    try:
+        cursor.execute('''INSERT INTO contacts (email, name, phone, pr1, pr2, pr3, pr4, pr5, messages)
+                          VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)''',
+                       (transformed_data.get('email'), transformed_data.get('name'), transformed_data.get('phone'),
+                        transformed_data.get('pr1'), transformed_data.get('pr2'), transformed_data.get('pr3'),
+                        transformed_data.get('pr4'), transformed_data.get('pr5'), json.dumps(user_data.get('messages'))))
+        connection.commit()
+    except sqlite3.IntegrityError:
+        cursor.execute('''UPDATE contacts SET name = ?, phone = ?, pr1 = ?, pr2 = ?, pr3 = ?, pr4 = ?, pr5 = ?, messages = ?
+                          WHERE email = ?''',
+                       (transformed_data.get('name'), transformed_data.get('phone'), transformed_data.get('pr1'),
+                        transformed_data.get('pr2'), transformed_data.get('pr3'), transformed_data.get('pr4'),
+                        transformed_data.get('pr5'), json.dumps(user_data.get('messages')), transformed_data.get('email')))
+        connection.commit()
+    finally:
+        connection.close()
+