Update app.py

app.py

@@ -737,9 +737,11 @@ def send_to_google_forms(user_data, gog_url):
         
 
 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}")
@@ -747,6 +749,7 @@ 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,))
     user = cursor.fetchone()
     logging.debug(f"User found: {user}")
@@ -756,6 +759,7 @@ def update_or_insert_user(db_name, user_data, mapping_template, gog_url):
     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}")
 
+    # Трансформация данных
     transformed_data = {}
     for json_key, db_column in mapping_template.items():
         value = user_data.get(json_key, "")
@@ -769,6 +773,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",
@@ -780,6 +785,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('+'):
@@ -790,13 +796,12 @@ def update_or_insert_user(db_name, user_data, mapping_template, gog_url):
     # Столбцы, которые не нужно перезаписывать
     no_overwrite_columns = ['ws_st', 'curator']  # Добавляйте сюда столбцы
 
-    # Проверяем наличие 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:
@@ -809,11 +814,12 @@ def update_or_insert_user(db_name, user_data, mapping_template, gog_url):
         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():
-            if column not in no_overwrite_columns:
+            if column not in no_overwrite_columns:  # Исключаем столбцы, которые не нужно перезаписывать
                 update_query += f"{column} = ?, "
                 update_values.append(value)
         update_query = update_query.rstrip(", ") + " WHERE email = ?"
@@ -829,13 +835,17 @@ def update_or_insert_user(db_name, user_data, mapping_template, gog_url):
         cursor.execute(insert_query, insert_values)
 
     conn.commit()
+
+    # Извлечение обновленных данных из базы данных
+    cursor.execute("SELECT * FROM contacts WHERE email = ?", (email,))
+    updated_user = cursor.fetchone()
+    updated_data = dict(zip([column[0] for column in cursor.description], updated_user))
+
     conn.close()
     logging.debug(f"User with email {email} processed successfully")
 
     # Отправка данных в Google Forms
-    send_to_google_forms(transformed_data, gog_url)
-
-
+    send_to_google_forms(updated_data, gog_url)