Update app.py

app.py

@@ -1194,106 +1194,9 @@ def se_upl_csv():
 
 
 
-DATABASE2 = 'data_gc.db'
-
-def parse_csv_data(data):
-    parsed_data = []
-    for item in data:
-        for key, value in item.items():
-            headers = key.split(';')
-            row = value.split(';')
-            parsed_data.append(dict(zip(headers, row)))
-    return parsed_data
-
-def insert_data(data):
-    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', '')
-        
-        # Удаление ведущего плюсика из номера телефона, если он есть
-        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', '')
-
-        # Проверка существования пользователя в базе данных по email или телефону
-        cursor.execute("SELECT 1 FROM contacts WHERE email = ? OR phone = ?", (email, phone))
-        user_exists = cursor.fetchone()
-
-        # Если пользователь существует, пропускаем вставку
-        if user_exists:
-            print(f"User with email {email} or phone {phone} already exists. Skipping insert.")
-            continue
-
-        # Создание динамического 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']
-        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 INTO contacts ({columns_str})
-            VALUES ({placeholders})
-        '''
 
-        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()
-@app.route('/upload_csv', methods=['POST'])
-def upload_csv():
-    if 'file' not in request.files:
-        return jsonify({"error": "No file part"}), 400
-    file = request.files['file']
-    if file.filename == '':
-        return jsonify({"error": "No selected file"}), 400
-    if file and file.filename.endswith('.csv'):
-        stream = io.StringIO(file.stream.read().decode("UTF8"), newline=None)
-        csv_input = csv.DictReader(stream)
-        data = [row for row in csv_input]
-        parsed_data = parse_csv_data(data)
-        insert_data(parsed_data)
-        return jsonify({"message": "Data uploaded and inserted successfully"})
-    return jsonify({"error": "Invalid file format"}), 400
 
 
 
@@ -1769,6 +1672,133 @@ def add_data_ver_cur():
 
 
 
+
+
+
+
+
+
+
+
+
+
+
+
+
+DATABASE2 = 'data_gc.db'
+def verify_phone_number2(phone_number):
+    current_curator_index = 0
+    verifikation_start = "1"  # Глобальная переменная для управления верификацией
+    curator_on_off = "1"  # Глобальная переменная для управления назначением куратора
+    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 insert_data2(data, template_key):
+    conn = sqlite3.connect(DATABASE2)
+    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:
+            print(f"Missing required fields in row: {row}. 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)
+
+        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'
+        ]
+
+        placeholders = ', '.join(['?' for _ in fields])
+        columns_str = ', '.join(fields)
+
+        query = f'''
+            INSERT INTO contacts ({columns_str})
+            VALUES ({placeholders})
+        '''
+
+        try:
+            cursor.execute(query, [user_data.get(field, '') for field in fields])
+        except Exception as e:
+            print(f"Error inserting row: {row}")
+            print(f"Error message: {str(e)}")
+            conn.rollback()
+            raise
+
+    conn.commit()
+    conn.close()
+
+@app.route('/upload_csv', methods=['POST'])
+def upload_csv():
+    global verifikation_start, curator_on_off
+
+    verifikation_start = request.form.get('verifikation_start', '0')
+    curator_on_off = request.form.get('curator_on_off', '0')
+    template_key = request.form.get('template_key', 'avp')
+
+    if 'file' not in request.files:
+        return jsonify({"error": "No file part"}), 400
+    file = request.files['file']
+    if file.filename == '':
+        return jsonify({"error": "No selected file"}), 400
+    if file and file.filename.endswith('.csv'):
+        stream = io.StringIO(file.stream.read().decode("UTF8"), newline=None)
+        csv_input = csv.DictReader(stream)
+        data = [row for row in csv_input]
+        insert_data2(data, template_key)
+        return jsonify({"message": "Data uploaded and inserted successfully"})
+    return jsonify({"error": "Invalid file format"}), 400
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+