Update app.py

app.py

@@ -1680,6 +1680,20 @@ def add_data_ver_cur():
 
 DATABASE2 = 'data_gc.db'
 
+def verify_phone_number(phone_number):
+    if verifikation_start == "1":
+        full_url_ver = f"{wa_url}{wa_ak}{ws_url_ver}{wa_api_key}"
+        payload = {"phoneNumber": phone_number}
+        headers = {'Content-Type': 'application/json'}
+        response = requests.post(full_url_ver, headers=headers, json=payload)
+        if response.status_code == 200:
+            response_body = response.json()
+            return response_body.get('existsWhatsapp', 'false')
+        else:
+            return "Error"
+    else:
+        return "false"
+
 def parse_csv_data(data):
     parsed_data = []
     for item in data:
@@ -1689,17 +1703,12 @@ def parse_csv_data(data):
             parsed_data.append(dict(zip(headers, row)))
     return parsed_data
 
-def insert_data(data, db_name):
-    conn = sqlite3.connect(db_name)
+def insert_data(data):
+    global current_curator_index
+    conn = sqlite3.connect(DATABASE2)
     cursor = conn.cursor()
 
     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', '')
         
@@ -1737,6 +1746,9 @@ def insert_data(data, db_name):
         utm_term = row.get('utm_term', '')
         utm_content = row.get('utm_content', '')
 
+        # Взятие даты из столбца CSV-таблицы
+        data_t = row.get('Date', '')
+
         # Проверка существования пользователя в базе данных по email или телефону
         cursor.execute("SELECT 1 FROM contacts WHERE email = ? OR phone = ?", (email, phone))
         user_exists = cursor.fetchone()
@@ -1746,6 +1758,15 @@ def insert_data(data, db_name):
             print(f"User with email {email} or phone {phone} already exists. Skipping insert.")
             continue
 
+        # Добавление куратора, если curator_on_off == "1"
+        if curator_on_off == "1":
+            curator = curators[current_curator_index]
+            current_curator_index = (current_curator_index + 1) % len(curators)
+
+        # Верификация номера телефона, если verifikation_start == "1"
+        if verifikation_start == "1":
+            ws_st = verify_phone_number(phone)
+
         # Создание динамического SQL-запроса для вставки данных
         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', 'utm_source', 'utm_medium', 'utm_campaign', 'utm_term', 'utm_content']
         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, utm_source, utm_medium, utm_campaign, utm_term, utm_content]
@@ -1781,8 +1802,7 @@ def upload_csv():
         csv_input = csv.DictReader(stream)
         data = [row for row in csv_input]
         parsed_data = parse_csv_data(data)
-        for db_name in DATABASES:
-            insert_data(parsed_data, db_name)
+        insert_data(parsed_data)
         return jsonify({"message": "Data uploaded and inserted successfully"})
     return jsonify({"error": "Invalid file format"}), 400