Update app.py

app.py

@@ -782,7 +782,6 @@ 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()
 
@@ -793,9 +792,9 @@ def update_or_insert_user(db_name, user_data, mapping_template):
         return
 
     logging.debug(f"Processing user with email: {email}")
-
+    
     # Проверка существования пользователя в базе данных по email
-    cursor.execute("SELECT web_st, b_mess FROM contacts WHERE email = ?", (email,))
+    cursor.execute("SELECT web_st FROM contacts WHERE email = ?", (email,))
     user = cursor.fetchone()
     logging.debug(f"User found: {user}")
 
@@ -810,16 +809,16 @@ def update_or_insert_user(db_name, user_data, mapping_template):
         # Обновление значения 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}")
-
-        # Извлечение старых сообщений
-        old_messages = json.loads(user[1]) if user[1] else []
-        logging.debug(f"Old messages: {old_messages}")
-
     else:
-        old_messages = []
+        conn.close()
         logging.debug(f"User {email} not found, proceeding with insert")
 
+    # Открываем соединение снова для остальных операций
+    conn = sqlite3.connect(db_name)
+    cursor = conn.cursor()
+
     # Преобразование данных пользователя на основе шаблона сопоставления
     transformed_data = {}
     for json_key, db_column in mapping_template.items():
@@ -853,40 +852,29 @@ def update_or_insert_user(db_name, user_data, mapping_template):
     # Добавление значения web_st в данные для вставки
     transformed_data['web_st'] = web_st_value
 
-    # Баннеры: Проверяется наличие ключа banners в user_data и устанавливается b_baners в 1 или 0.
-    if 'banners' in user_data and isinstance(user_data['banners'], list) and user_data['banners']:
-        transformed_data['b_baners'] = '1'
-    else:
-        transformed_data['b_baners'] = '0'
-    logging.debug(f"Transformed data after processing banners: {transformed_data}")
-
-    # Объединение старых и новых сообщений
-    new_messages = user_data.get('messages', [])
-    all_messages = old_messages + new_messages
-    transformed_data['b_mess'] = json.dumps(all_messages)
-    logging.debug(f"Combined messages: {all_messages}")
-
-    # Обновление или вставка данных в таблицу contacts
+    # Обновление данных пользователя в базе данных
     if user:
-        # Обновление существующего пользователя
-        update_fields = ', '.join([f"{col} = ?" for col in transformed_data.keys()])
-        update_values = list(transformed_data.values()) + [email]
-        cursor.execute(f"UPDATE contacts SET {update_fields} WHERE email = ?", update_values)
-        logging.debug(f"Update query: UPDATE contacts SET {update_fields} WHERE email = {email}")
+        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 = ', '.join('?' for _ in transformed_data)
+        insert_query = f"INSERT INTO contacts ({columns}) VALUES ({placeholders})"
         insert_values = list(transformed_data.values())
-        cursor.execute(f"INSERT INTO contacts ({columns}) VALUES ({placeholders})", insert_values)
-        logging.debug(f"Insert query: INSERT INTO contacts ({columns}) VALUES ({placeholders})")
-        logging.info(f"New user {email} has been added to the database.")
+        logging.debug(f"Insert query: {insert_query} with values: {insert_values}")
+        cursor.execute(insert_query, insert_values)
 
     # Подтверждение изменений и закрытие соединения
     conn.commit()
     conn.close()
-    logging.info(f"User {email} data has been successfully saved/updated.")
-
+    logging.debug(f"User with email {email} processed successfully")
 
 @app.route('/send_get_request', methods=['GET'])
 def send_get_request():
@@ -902,14 +890,26 @@ def send_get_request():
         report = data.get('report', {})
         messages = data.get('messages', {})
 
-        # Получение пользователей
-        users_meta = report.get('usersMeta', {})
+        report_json_str = report.get('report', '{}')
+        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', {})
 
+        # Обновление или добавление каждого пользователя в базу данных data_gc.db
         processed_emails = set()
         for user_id, user_data in users_meta.items():
-            user_data['messages'] = messages.get(user_id, [])
+            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)