Update app.py

app.py

@@ -723,57 +723,31 @@ gog_url = "https://docs.google.com/forms/d/e/1FAIpQLSc-JbmXvgpgGq6KrkXsYSsfMACVM
 
 DATABASE_NAME = 'data_gc.db'
 
-def update_or_insert_user(db_name, user_data, mapping_template):
+def update_or_insert_user(db_name, user_data, mapping_template, gog_url):
     conn = sqlite3.connect(db_name)
     cursor = conn.cursor()
 
-    # Получение email пользователя из данных
     email = user_data.get('email')
     if not email:
         logging.error(f"User data missing email: {user_data}")
         return
 
     logging.debug(f"Processing user with email: {email}")
-    
-    # Проверка существования пользователя в базе данных по email
-    cursor.execute("SELECT web_st, ws_st FROM contacts WHERE email = ?", (email,))
+
+    cursor.execute("SELECT web_st, ws_st, b_mess FROM contacts WHERE email = ?", (email,))
     user = cursor.fetchone()
     logging.debug(f"User found: {user}")
 
-    # Вынесение увеличения значения web_st в отдельный блок
-    web_st_value = 1  # Инициализация значения web_st
-    current_ws_st = None
-    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}")
-
-        # Получение текущего значения ws_st
-        current_ws_st = user[1] if user[1] is not None and user[1] != "" else None
-
-        # Обновление значения 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()
-        logging.debug(f"User {email} not found, proceeding with insert")
-
-    # Открываем соединение снова для остальных операций
-    conn = sqlite3.connect(db_name)
-    cursor = conn.cursor()
+    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 ""
 
-    # Преобразование данных пользователя на основе шаблона сопоставления
     transformed_data = {}
     for json_key, db_column in mapping_template.items():
         value = user_data.get(json_key, "")
-        
         if isinstance(value, list):
-            # Проверяем тип элементов списка
             if all(isinstance(item, str) for item in value):
-                transformed_data[db_column] = "; ".join(value)  # Сохраняем сообщения в строку
+                transformed_data[db_column] = "; ".join(value)
             else:
                 logging.error(f"Expected list of strings for key {json_key}, but got: {value}")
                 transformed_data[db_column] = ""
@@ -781,11 +755,10 @@ def update_or_insert_user(db_name, user_data, mapping_template):
             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",
-        "shop_st", "curator", "pr1", "pr2", "pr3", "pr4", "pr5", "ad_url",
+        "shop_st", "curator", "pr1", "pr2", "pr3", "pr4", "pr5", "gc_url",
         "key_pr", "n_con", "canal", "data_t", 'utm_source', 'utm_medium', 'utm_campaign', 'utm_term', 'utm_content'
     ]
     for field in required_fields:
@@ -793,7 +766,6 @@ def update_or_insert_user(db_name, user_data, mapping_template):
             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('+'):
@@ -801,16 +773,25 @@ def update_or_insert_user(db_name, user_data, mapping_template):
         transformed_data['phone'] = phone
     logging.debug(f"Transformed data after phone processing: {transformed_data}")
 
-    # Добавление значения web_st в данные для вставки
-    transformed_data['web_st'] = web_st_value
-
-    # Проверка наличия ws_st в базе данных и не перезаписываем его, если оно уже существует
-    if current_ws_st is not None:
+    # Проверяем наличие ws_st в базе данных и не перезаписываем его, если оно уже существует
+    if current_ws_st is not None and current_ws_st != "":
         transformed_data['ws_st'] = current_ws_st
     else:
         transformed_data['ws_st'] = user_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
+    else:
+        transformed_data['web_st'] = 1
+
+    new_messages = transformed_data.get('b_mess', "")
+    if current_messages:
+        transformed_data['b_mess'] = current_messages + "; " + 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 = []
@@ -829,11 +810,13 @@ def update_or_insert_user(db_name, user_data, mapping_template):
         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")
 
+    # Отправка данных в Google Forms
+    send_to_google_forms(transformed_data, gog_url)
+
 
 @app.route('/send_request', methods=['POST'])
 def send_request():