Update app.py

app.py

@@ -1187,80 +1187,69 @@ def parse_csv_data(data):
             parsed_data.append(dict(zip(headers, row)))
     return parsed_data
 
-
-def update_or_insert_user(db_name, user_data, mapping_template):
-    conn = sqlite3.connect(db_name)
+def insert_data(data):
+    conn = sqlite3.connect(DATABASE2)
     cursor = conn.cursor()
+    # Инвертируем порядок данных
+    data = reversed(data)
+    for row in data:
+        # Получение текущего времени в московском часовом поясе
+        utc_now = datetime.utcnow()
+        msk_tz = pytz.timezone('Europe/Moscow')
+        msk_now = utc_now.replace(tzinfo=pytz.utc).astimezone(msk_tz)
+        data_t = msk_now.strftime('%Y-%m-%d %H:%M:%S')
 
-    # Получение 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 1 FROM contacts WHERE email = ?", (email,))
-    user = cursor.fetchone()
-    logging.debug(f"User found: {user}")
-
-    # Если пользователь существует, закрываем соединение и возвращаемся
-    if user:
-        conn.close()
-        logging.debug(f"User with email {email} already exists. Skipping insert.")
-        return
-
-    # Преобразование данных пользователя на основе шаблона сопоставления
-    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)  # Сохраняем сообщения в строку
-            else:
-                logging.error(f"Expected list of strings for key {json_key}, but got: {value}")
-                transformed_data[db_column] = ""
-        else:
-            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",
-        "key_pr", "n_con", "canal", "data_t"
-    ]
-    for field in required_fields:
-        if field not in transformed_data:
-            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('+'):
-            phone = phone[1:]
-        transformed_data['phone'] = phone
-    logging.debug(f"Transformed data after phone processing: {transformed_data}")
+        # Проверка наличия ключей перед их использованием и предоставление значений по умолчанию
+        name = row.get('Name', '')
+        phone = row.get('Phone', '')
+        email = row.get('Email', '')
+        vk_id = row.get('vk_id', '')
+        chat_id = row.get('chat_id', '')
+        ws_st = row.get('ws_st', '')
+        ws_stop = row.get('ws_stop', '')
+        web_st = row.get('web_st', 0)
+        fin_prog = row.get('fin_prog', 0)
+        b_city = row.get('b_city', '')
+        b_fin = row.get('b_fin', '')
+        b_ban = row.get('b_ban', '')
+        b_ign = row.get('b_ign', '')
+        b_baners = row.get('b_baners', '')
+        b_butt = row.get('b_butt', '')
+        b_mess = row.get('b_mess', '')
+        shop_st = row.get('shop_st', '')
+        curator = row.get('curator', '')
+        pr1 = row.get('pr1', '')
+        pr2 = row.get('pr2', '')
+        pr3 = row.get('pr3', '')
+        pr4 = row.get('pr4', '')
+        pr5 = row.get('pr5', '')
+        ad_url = row.get('ad_url', '')
+        key_pr = row.get('key_pr', '')
+        n_con = row.get('n_con', '')
+        canal = row.get('canal', '')
+
+        # Создание динамического SQL-запроса с использованием INSERT OR REPLACE
+        columns = ['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', 'ad_url', 'key_pr', 'n_con', 'canal', 'data_t']
+        values = [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, ad_url, key_pr, n_con, canal, data_t]
+
+        placeholders = ', '.join(['?' for _ in columns])
+        columns_str = ', '.join(columns)
+
+        query = f'''
+            INSERT OR REPLACE INTO contacts ({columns_str})
+            VALUES ({placeholders})
+        '''
 
-    # Добавление нового пользователя в базу данных
-    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())
-    logging.debug(f"Insert query: {insert_query} with values: {insert_values}")
-    cursor.execute(insert_query, insert_values)
+        try:
+            cursor.execute(query, values)
+        except Exception as e:
+            print(f"Error inserting row: {row}")
+            print(f"Error message: {str(e)}")
+            conn.rollback()
+            raise
 
-    # Подтверждение изменений и закрытие соединения
     conn.commit()
     conn.close()
-    logging.debug(f"User with email {email} inserted successfully")
-
-
 
 @app.route('/upload_csv', methods=['POST'])
 def upload_csv():
@@ -1274,18 +1263,7 @@ def upload_csv():
         csv_input = csv.DictReader(stream)
         data = [row for row in csv_input]
         parsed_data = parse_csv_data(data)
-        
-        # Пример шаблона сопоставления
-        mapping_template = {
-            'Email': 'email',
-            'Name': 'name',
-            'Phone': 'phone',
-            # Добавьте остальные поля в шаблон сопоставления
-        }
-
-        for user_data in parsed_data:
-            update_or_insert_user(DATABASE2, user_data, mapping_template)
-
+        insert_data(parsed_data)
         return jsonify({"message": "Data uploaded and inserted successfully"})
     return jsonify({"error": "Invalid file format"}), 400
 
@@ -1310,7 +1288,6 @@ def upload_csv():
 
 
 
-
 
 
 @app.route('/total_users', methods=['GET'])