Update app.py

app.py

@@ -782,81 +782,99 @@ def send_request():
 
 DATABASE_NAME = 'data_gc.db'
 def update_or_insert_user(db_name, user_data, mapping_template):
-    conn = sqlite3.connect(db_name)  # Подключение к базе данных
-    cursor = conn.cursor()  # Создание курсора для выполнения SQL-запросов
+    conn = sqlite3.connect(db_name)
+    cursor = conn.cursor()
 
-    email = user_data.get('email')  # Получение email пользователя из данных
-    if not email:  # Если email отсутствует
-        return  # Прекращение выполнения функции, так как нет email
+    # Получение email пользователя из данных
+    email = user_data.get('email')
+    if not email:
+        logging.error(f"User data missing email: {user_data}")
+        return
 
-    cursor.execute("SELECT web_st FROM contacts WHERE email = ?", (email,))  # Выполнение SQL-запроса для проверки существования пользователя
-    user = cursor.fetchone()  # Получение первого найденного результата
+    logging.debug(f"Processing user with email: {email}")
+    
+    # Проверка существования пользователя в базе данных по email
+    cursor.execute("SELECT web_st FROM contacts WHERE email = ?", (email,))
+    user = cursor.fetchone()
+    logging.debug(f"User found: {user}")
 
+    # Вынесение увеличения значения web_st в отдельный блок
     web_st_value = 1  # Инициализация значения web_st
-    if user:  # Если пользователь найден
-        current_web_st = user[0] if user[0] is not None and user[0] != "" else 0  # Получение текущего значения web_st или установка в 0, если значение пустое
-        web_st_value = int(current_web_st) + 1  # Увеличение значения web_st на 1
-
-        cursor.execute("UPDATE contacts SET web_st = ? WHERE email = ?", (web_st_value, email))  # Выполнение SQL-запроса для обновления значения web_st
-        conn.commit()  # Подтверждение изменений в базе данных
-        conn.close()  # Закрытие соединения с базой данных
-
+    if user:
+        # Проверка текущего значения web_st и его инкрементация
+        current_web_st = user[0] if user[0] is not None and user[0] != "" else 0
+        web_st_value = int(current_web_st) + 1
+        logging.debug(f"Calculated web_st_value: {web_st_value}")
+
+        # Обновление значения web_st
+        cursor.execute("UPDATE contacts SET web_st = ? WHERE email = ?", (web_st_value, email))
+        conn.commit()
+        conn.close()
+        logging.debug(f"User {email} web_st updated to {web_st_value}")
     else:
-        conn.close()  # Закрытие соединения, если пользователь не найден
+        conn.close()
+        logging.debug(f"User {email} not found, proceeding with insert")
 
-    conn = sqlite3.connect(db_name)  # Повторное подключение к базе данных
-    cursor = conn.cursor()  # Создание нового курсора
+    # Открываем соединение снова для остальных операций
+    conn = sqlite3.connect(db_name)
+    cursor = conn.cursor()
 
-    transformed_data = {}  # Создание словаря для хранения преобразованных данных
-    for json_key, db_column in mapping_template.items():  # Проход по всем элементам шаблона сопоставления
-        value = user_data.get(json_key, "")  # Получение значения из данных пользователя по ключу json_key
-        if isinstance(value, list):  # Если значение является списком
-            transformed_data[db_column] = "1" if value else "0"  # Преобразование списка в "1" или "0"
+    # Преобразование данных пользователя на основе шаблона сопоставления
+    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] = str(value)  # Преобразование значения в строку
-    logging.debug(f"Transformed data: {transformed_data}")  # Логирование преобразованных данных
+            transformed_data[db_column] = str(value)
+    logging.debug(f"Transformed data: {transformed_data}")
 
-    required_fields = [  # Список обязательных полей
+    # Заполнение обязательных полей значениями по умолчанию
+    required_fields = [
         "vk_id", "chat_id", "ws_st", "ws_stop", "web_st", "fin_prog",
         "b_city", "b_fin", "b_ban", "b_ign", "b_baners", "b_butt", "b_mess",
         "shop_st", "curator", "pr1", "pr2", "pr3", "pr4", "pr5", "ad_url",
         "key_pr", "n_con", "canal", "data_t"
     ]
-    for field in required_fields:  # Проход по всем обязательным полям
-        if field not in transformed_data:  # Если поле отсутствует в преобразованных данных
-            transformed_data[field] = ""  # Установка значения по умолчанию для отсутствующего поля
-    logging.debug(f"Transformed data after adding required fields: {transformed_data}")  # Логирование данных после добавления обязательных полей
-
-    if 'phone' in user_data:  # Если в данных пользователя присутствует телефон
-        phone = user_data['phone']  # Получение номера телефона
-        if phone.startswith('+'):  # Если телефон начинается с "+"
-            phone = phone[1:]  # Удаление "+" из начала номера телефона
-        transformed_data['phone'] = phone  # Добавление номера телефона в преобразованные данные
-    logging.debug(f"Transformed data after phone processing: {transformed_data}")  # Логирование данных после обработки номера телефона
-
-    transformed_data['web_st'] = web_st_value  # Добавление значения web_st в преобразованные данные
-
-    if user:  # Если пользователь найден
-        update_query = "UPDATE contacts SET "  # Начало формирования SQL-запроса для обновления данных
-        update_values = []  # Список для хранения значений для обновления
-        for column, value in transformed_data.items():  # Проход по всем преобразованным данным
-            update_query += f"{column} = ?, "  # Добавление столбца и placeholder в запрос
-            update_values.append(value)  # Добавление значения в список значений
-        update_query = update_query.rstrip(", ") + " WHERE email = ?"  # Завершение запроса и добавление условия
-        update_values.append(email)  # Добавление email в список значений
-        logging.debug(f"Update query: {update_query} with values: {update_values}")  # Логирование запроса и значений
-        cursor.execute(update_query, update_values)  # Выполнение SQL-запроса на обновление
+    for field in required_fields:
+        if field not in transformed_data:
+            transformed_data[field] = ""
+    logging.debug(f"Transformed data after adding required fields: {transformed_data}")
+
+    # Обработка номера телефона, если он есть
+    if 'phone' in user_data:
+        phone = user_data['phone']
+        if phone.startswith('+'):
+            phone = phone[1:]
+        transformed_data['phone'] = phone
+    logging.debug(f"Transformed data after phone processing: {transformed_data}")
+
+    # Добавление значения web_st в данные для вставки
+    transformed_data['web_st'] = web_st_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)
+        logging.debug(f"Update query: {update_query} with values: {update_values}")
+        cursor.execute(update_query, update_values)
     else:
-        columns = ', '.join(transformed_data.keys())  # Формирование списка столбцов для вставки
-        placeholders = ', '.join('?' for _ in transformed_data)  # Формирование списка placeholders
-        insert_query = f"INSERT INTO contacts ({columns}) VALUES ({placeholders})"  # Формирование SQL-запроса для вставки данных
-        insert_values = list(transformed_data.values())  # Получение значений для вставки
-        logging.debug(f"Insert query: {insert_query} with values: {insert_values}")  # Логирование запроса и значений
-        cursor.execute(insert_query, insert_values)  # Выполнение SQL-запроса на вставку
-
-    conn.commit()  # Подтверждение изменений в базе данных
-    conn.close()  # Закрытие соединения с базой данных
-    logging.debug(f"User with email {email} processed successfully")  # Логирование успешной обработки пользователя
+        columns = ', '.join(transformed_data.keys())
+        placeholders = ', '.join('?' for _ in transformed_data)
+        insert_query = f"INSERT INTO contacts ({columns}) VALUES ({placeholders})"
+        insert_values = list(transformed_data.values())
+        logging.debug(f"Insert query: {insert_query} with values: {insert_values}")
+        cursor.execute(insert_query, insert_values)
+
+    # Подтверждение изменений и закрытие соединения
+    conn.commit()
+    conn.close()
+    logging.debug(f"User with email {email} processed successfully")
 
 @app.route('/send_get_request', methods=['GET'])
 def send_get_request():
@@ -864,37 +882,34 @@ def send_get_request():
     webinarId = request.args.get('webinarId')
     url = f'https://online.bizon365.ru/api/v1/webinars/reports/get?webinarId={webinarId}'
 
-    # Выполнение GET-запроса к API для получения отчета и сообщений
     response = requests.get(url, headers={'X-Token': token})
 
-    # Проверка, успешно ли выполнен запрос
     if response.status_code == 200:
-        data = response.json()  # Получение данных в формате JSON
+        data = response.json()
 
-        # Получение разделов 'report' и 'messages' из данных
         report = data.get('report', {})
-        messages_str = data.get('messages', '{}')  # Получение строки JSON для сообщений
+        messages = data.get('messages', {})
 
-        # Преобразование строки JSON в объект Python
+        report_json_str = report.get('report', '{}')
         try:
-            messages_json = json.loads(messages_str)  # Преобразование строки JSON в объект
+            report_json = json.loads(report_json_str)
         except json.JSONDecodeError:
-            messages_json = {}  # Установка пустого объекта при ошибке преобразования
+            report_json = {}
 
-        # Получение метаинформации пользователей из отчета
-        users_meta = report.get('usersMeta', {})
+        messages_json_str = report.get('messages', '{}')
+        try:
+            messages_json = json.loads(messages_json_str)
+        except json.JSONDecodeError:
+            messages_json = {}
 
-        # Создание множества для хранения обработанных email-ов
-        processed_emails = set()
+        users_meta = report_json.get('usersMeta', {})
 
-        # Проход по всем пользователям
+        # Обновление или добавление каждого пользователя в базу данных data_gc.db
+        processed_emails = set()
         for user_id, user_data in users_meta.items():
-            # Извлечение сообщений конкретного пользователя
-            messages_user = messages_json.get(user_id, [])  # Получение сообщений пользователя
-            user_data['messages'] = messages_user  # Добавление сообщений в данные пользователя
-
+            user_data['messages'] = messages_json
             email = user_data.get('email')
-            if email and email not in processed_emails:
+            if email not in processed_emails:
                 update_or_insert_user(DATABASE_NAME, user_data, mapping_template)
                 processed_emails.add(email)