Update app.py

app.py

@@ -1187,75 +1187,106 @@ def parse_csv_data(data):
             parsed_data.append(dict(zip(headers, row)))
     return parsed_data
 
-def insert_data(data):
-    conn = sqlite3.connect(DATABASE2)
+def update_or_insert_user(db_name, user_data, mapping_template):
+    conn = sqlite3.connect(db_name)
     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')
 
-        # Проверка наличия ключей перед их использованием и предоставление значений по умолчанию
-        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 ON CONFLICT
-        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)
-
-        update_str = ', '.join([f"{col} = excluded.{col}" for col in columns if col != 'email'])
-
-        query = f'''
-            INSERT INTO contacts ({columns_str})
-            VALUES ({placeholders})
-            ON CONFLICT(email) DO UPDATE SET
-            {update_str}
-        '''
+    # Получение email пользователя из данных
+    email = user_data.get('Email')
+    if not email:
+        logging.error(f"User data missing email: {user_data}")
+        return
 
-        try:
-            cursor.execute(query, values)
-        except Exception as e:
-            print(f"Error inserting row: {row}")
-            print(f"Error message: {str(e)}")
-            conn.rollback()
-            raise
+    logging.debug(f"Processing user with email: {email}")
+    
+    # Проверка существования пользователя в базе данных по email
+    cursor.execute("SELECT web_st FROM contacts WHERE email = ?", (email,))
+    user = cursor.fetchone()
+    logging.debug(f"User found: {user}")
 
-    conn.commit()
-    conn.close()
+    # Вынесение увеличения значения web_st в отдельный блок
+    web_st_value = 1  # Инициализация значения web_st
+    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}")
+
+        # Обновление значения 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()
+
+    # Преобразование данных пользователя на основе шаблона сопоставления
+    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}")
+
+    # Добавление значения web_st в данные для вставки
+    transformed_data['web_st'] = web_st_value
 
+    # Обновление данных пользователя в базе данных
+    if user:
+        update_query = "UPDATE contacts SET "
+        update_values = []
+        for column, value in transformed_data.items():
+            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())
+        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")
 
 @app.route('/upload_csv', methods=['POST'])
 def upload_csv():
@@ -1269,7 +1300,18 @@ def upload_csv():
         csv_input = csv.DictReader(stream)
         data = [row for row in csv_input]
         parsed_data = parse_csv_data(data)
-        insert_data(parsed_data)
+        
+        # Пример шаблона сопоставления
+        mapping_template = {
+            'Email': 'email',
+            'Name': 'name',
+            'Phone': 'phone',
+            # Добавьте остальные поля в шаблон сопоставления
+        }
+
+        for user_data in parsed_data:
+            update_or_insert_user(DATABASE2, user_data, mapping_template)
+
         return jsonify({"message": "Data uploaded and inserted successfully"})
     return jsonify({"error": "Invalid file format"}), 400