Update app.py

app.py

@@ -1680,21 +1680,6 @@ def add_data_ver_cur():
 
 DATABASE2 = 'data_gc.db'
 
-# Функция для верификации номера телефона через внешний API
-def verify_phone_number2(phone_number):
-    if verifikation_start == "1":
-        full_url_ver = f"{wa_url}{wa_ak}{ws_url_ver}{wa_api_key}"
-        payload = json.dumps({"phoneNumber": phone_number})
-        headers = {'Content-Type': 'application/json'}
-        response = requests.post(full_url_ver, headers=headers, data=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:
@@ -1704,53 +1689,69 @@ def parse_csv_data(data):
             parsed_data.append(dict(zip(headers, row)))
     return parsed_data
 
-def insert_data2(data, template_key):
-    global current_curator_index
-    conn = sqlite3.connect(DATABASE2)
+def insert_data(data, db_name):
+    conn = sqlite3.connect(db_name)
     cursor = conn.cursor()
 
-    mapping_template_cur = mapp_templates.get(template_key, mt_avp)
-
     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')
 
-        user_data = {mapping_template_cur[key]: row.get(key, "") for key in mapping_template_cur}
-
-        name = user_data.get('name', '')
-        phone = user_data.get('phone', '').lstrip('+')
-        email = user_data.get('email', '')
-
-        if not name or not phone or not email:
-            logging.error(f"Missing required fields in row: {row}. Skipping insert.")
-            continue
-
+        # Проверка наличия ключей перед их использованием и предоставление значений по умолчанию
+        name = row.get('Name', '')
+        
+        # Удаление ведущего плюсика из номера телефона, если он есть
+        phone = row.get('Phone', '').lstrip('+')
+        
+        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', '')
+        utm_source = row.get('utm_source', '')
+        utm_medium = row.get('utm_medium', '')
+        utm_campaign = row.get('utm_campaign', '')
+        utm_term = row.get('utm_term', '')
+        utm_content = row.get('utm_content', '')
+
+        # Проверка существования пользователя в базе данных по email или телефону
         cursor.execute("SELECT 1 FROM contacts WHERE email = ? OR phone = ?", (email, phone))
         user_exists = cursor.fetchone()
 
+        # Если пользователь существует, пропускаем вставку
         if user_exists:
-            logging.debug(f"User with email {email} or phone {phone} already exists. Skipping insert.")
+            print(f"User with email {email} or phone {phone} already exists. Skipping insert.")
             continue
 
-        if curator_on_off == "1":
-            user_data['curator'] = curators[current_curator_index]
-            current_curator_index = (current_curator_index + 1) % len(curators)
-
-        if verifikation_start == "1":
-            user_data['ws_st'] = verify_phone_number2(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]
 
-        user_data['data_t'] = data_t
-
-        fields = [
-            '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'
-        ]
-
-        placeholders = ', '.join(['?' for _ in fields])
-        columns_str = ', '.join(fields)
+        placeholders = ', '.join(['?' for _ in columns])
+        columns_str = ', '.join(columns)
 
         query = f'''
             INSERT INTO contacts ({columns_str})
@@ -1758,27 +1759,15 @@ def insert_data2(data, template_key):
         '''
 
         try:
-            cursor.execute(query, [user_data.get(field, '') for field in fields])
-            logging.debug(f"Row inserted successfully: {user_data}")
+            cursor.execute(query, values)
         except Exception as e:
-            logging.error(f"Error inserting row: {row}")
-            logging.error(f"Error message: {str(e)}")
+            print(f"Error inserting row: {row}")
+            print(f"Error message: {str(e)}")
             conn.rollback()
             raise
 
-    try:
-        conn.commit()
-        logging.debug("Transaction committed successfully.")
-    except Exception as e:
-        logging.error(f"Error committing transaction: {str(e)}")
-        conn.rollback()
-    finally:
-        conn.close()
-
-def verify_api_key():
-    api_key = request.args.get('api_sys')
-    if api_key != "fasSd345D":
-        return jsonify({"error": "Invalid API key"}), 403
+    conn.commit()
+    conn.close()
 
 @app.route('/upload_csv', methods=['POST'])
 def upload_csv():
@@ -1792,7 +1781,8 @@ def upload_csv():
         csv_input = csv.DictReader(stream)
         data = [row for row in csv_input]
         parsed_data = parse_csv_data(data)
-        insert_data2(parsed_data, 'avp')
+        for db_name in DATABASES:
+            insert_data(parsed_data, db_name)
         return jsonify({"message": "Data uploaded and inserted successfully"})
     return jsonify({"error": "Invalid file format"}), 400