Update app.py

app.py

@@ -710,6 +710,10 @@ def update_or_insert_user(db_name, user_data, mapping_template):
     conn = sqlite3.connect(db_name)
     cursor = conn.cursor()
 
+    # Получение списка столбцов из таблицы
+    cursor.execute("PRAGMA table_info(contacts)")
+    columns = [column[1] for column in cursor.fetchall()]
+
     # Преобразование данных пользователя на основе шаблона сопоставления
     transformed_data = {}
     for json_key, db_column in mapping_template.items():
@@ -728,10 +732,10 @@ def update_or_insert_user(db_name, user_data, mapping_template):
 
     # Заполнение обязательных полей значениями по умолчанию
     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", 'utm_source', 'utm_medium', 'utm_campaign', 'utm_term', 'utm_content'
+        "id", "name", "phone", "email", "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", "gc_url", "key_pr", "n_con", "canal", "data_on", "data_t",
+        "utm_source", "utm_medium", "utm_campaign", "utm_term", "utm_content", "gcpc"
     ]
     for field in required_fields:
         if field not in transformed_data:
@@ -754,17 +758,18 @@ def update_or_insert_user(db_name, user_data, mapping_template):
         update_query = "UPDATE contacts SET "
         update_values = []
         for column, value in transformed_data.items():
-            update_query += f"{column} = ?, "
-            update_values.append(value)
+            if column in columns:
+                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)
-        insert_query = f"INSERT INTO contacts ({columns}) VALUES ({placeholders})"
-        insert_values = list(transformed_data.values())
+        insert_columns = [column for column in transformed_data.keys() if column in columns]
+        placeholders = ', '.join('?' for _ in insert_columns)
+        insert_query = f"INSERT INTO contacts ({', '.join(insert_columns)}) VALUES ({placeholders})"
+        insert_values = [transformed_data[column] for column in insert_columns]
         logging.debug(f"Insert query: {insert_query} with values: {insert_values}")
         cursor.execute(insert_query, insert_values)