Update app.py

app.py

@@ -738,6 +738,16 @@ def send_to_google_forms(user_data, gog_url):
 
 
 def update_or_insert_user(db_name, user_data, mapping_template, gog_url):
+    """
+    Обновляет или вставляет данные пользователя в базу данных и отправляет их в Google Forms.
+    
+    Параметры:
+    - db_name: Имя файла базы данных SQLite.
+    - user_data: Словарь с данными пользователя.
+    - mapping_template: Шаблон сопоставления ключей данных пользователя и колонок в базе данных.
+    - gog_url: URL Google Forms для отправки данных.
+    """
+    
     conn = sqlite3.connect(db_name)
     cursor = conn.cursor()
 
@@ -748,15 +758,15 @@ def update_or_insert_user(db_name, user_data, mapping_template, gog_url):
 
     logging.debug(f"Processing user with email: {email}")
 
-    cursor.execute("SELECT web_st, ws_st, b_mess FROM contacts WHERE email = ?", (email,))
+    # Получаем текущие данные пользователя из базы данных
+    cursor.execute("SELECT * FROM contacts WHERE email = ?", (email,))
     user = cursor.fetchone()
     logging.debug(f"User found: {user}")
 
-    current_web_st = user[0] if user else None
-    current_ws_st = user[1] if user else None
-    current_messages = user[2] if user else ""
-    logging.debug(f"Current web_st: {current_web_st}, current_ws_st: {current_ws_st}, current_messages: {current_messages}")
+    current_data = {description[0]: value for description, value in zip(cursor.description, user)} if user else {}
+    logging.debug(f"Current data: {current_data}")
 
+    # Преобразуем данные пользователя
     transformed_data = {}
     for json_key, db_column in mapping_template.items():
         value = user_data.get(json_key, "")
@@ -770,6 +780,7 @@ def update_or_insert_user(db_name, user_data, mapping_template, gog_url):
             transformed_data[db_column] = str(value)
     logging.debug(f"Transformed data: {transformed_data}")
 
+    # Заполняем все необходимые поля
     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",
@@ -781,6 +792,7 @@ def update_or_insert_user(db_name, user_data, mapping_template, gog_url):
             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('+'):
@@ -788,35 +800,42 @@ def update_or_insert_user(db_name, user_data, mapping_template, gog_url):
         transformed_data['phone'] = phone
     logging.debug(f"Transformed data after phone processing: {transformed_data}")
 
-    # Проверяем наличие ws_st в базе данных и не перезаписываем его, если оно уже существует
-    if current_ws_st is not None and current_ws_st != "":
-        transformed_data['ws_st'] = current_ws_st
+    # Обрабатываем статус WhatsApp
+    ws_st_value = user_data.get('ws_st', "")
+    if ws_st_value == "TRUE":
+        transformed_data['ws_st'] = 1
+    elif ws_st_value == "FALSE":
+        transformed_data['ws_st'] = 0
     else:
-        transformed_data['ws_st'] = user_data.get('ws_st', "")
+        transformed_data['ws_st'] = current_data.get('ws_st', "")
 
-    # Увеличиваем значение web_st на 1, если пользователь уже существует в базе данных
-    if current_web_st is not None and current_web_st != "":
-        transformed_data['web_st'] = int(current_web_st) + 1
+    # Увеличиваем значение web_st на 1, если пользователь уже существует
+    if current_data.get('web_st') is not None and current_data.get('web_st') != "":
+        transformed_data['web_st'] = int(current_data['web_st']) + 1
     else:
         transformed_data['web_st'] = 1
 
+    # Обрабатываем сообщения
     new_messages = transformed_data.get('b_mess', "")
-    if current_messages:
-        transformed_data['b_mess'] = current_messages + "; " + new_messages
+    if current_data.get('b_mess'):
+        transformed_data['b_mess'] = current_data['b_mess'] + "; " + new_messages
     else:
         transformed_data['b_mess'] = new_messages
     logging.debug(f"Transformed data after message processing: {transformed_data}")
 
+    # Вставка или обновление данных в базе данных
     if user:
         update_query = "UPDATE contacts SET "
         update_values = []
         for column, value in transformed_data.items():
-            update_query += f"{column} = ?, "
-            update_values.append(value)
+            if value != current_data.get(column):
+                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)
+        if update_values:
+            cursor.execute(update_query, update_values)
     else:
         columns = ', '.join(transformed_data.keys())
         placeholders = ', '.join('?' for _ in transformed_data)