Update app.py

app.py

@@ -1657,7 +1657,7 @@ def verify_phone_number(phone_number):
         return response_body.get('existsWhatsapp', 'false')
     else:
         return "Error"
-
+        
 def parse_csv_data(data):
     parsed_data = []
     for item in data:
@@ -1748,11 +1748,6 @@ def upload_csv():
         return jsonify({"message": "Data uploaded and inserted successfully"})
     return jsonify({"error": "Invalid file format"}), 400
 
-
-
-
-
-
 @app.route('/upl_csv', methods=['GET'])
 def se_upl_csv():
     api_sys_control = request.args.get('api_sys')
@@ -1766,7 +1761,114 @@ def se_upl_csv():
 
 
 
+DATABASE2 = 'data_gc.db'
+
+def verify_phone_number_b(phone_number):
+    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"
+
+def parse_csv_data_b(data):
+    parsed_data = []
+    for row in data:
+        parsed_row = {}
+        for key, value in row.items():
+            headers = key.split(',')
+            values = value.split(',')
+            parsed_row.update(dict(zip(headers, values)))
+        parsed_data.append(parsed_row)
+    return parsed_data
+
+def clean_phone_number_b(phone_number):
+    return re.sub(r'\D', '', phone_number)
+
+def insert_data_b(data, verify_phone, add_curator):
+    global current_curator_index
+    conn = sqlite3.connect(DATABASE2)
+    cursor = conn.cursor()
+
+    for row in data:
+        name = row.get('Имя', '')
+        phone = row.get('WhatsApp', '').lstrip('+')
+        email = row.get('Email', '')
+        data_t = row.get('Дата', '').strip('"')
+
+        phone = clean_phone_number_b(phone)
+
+        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
+
+        if add_curator == "1":
+            curator = curators[current_curator_index]
+            current_curator_index = (current_curator_index + 1) % len(curators)
+        else:
+            curator = row.get('Куратор', '')
+
+        if verify_phone == "1":
+            ws_st = verify_phone_number_b(phone)
+        else:
+            ws_st = row.get('ws_st', '')
+
+        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', 'gc_url', 'key_pr', 'n_con', 'canal', 'data_t', 'utm_source', 'utm_medium', 'utm_campaign', 'utm_term', 'utm_content']
+        values = [name, phone, email, row.get('ВКонтакте', ''), row.get('Телеграм', ''), ws_st, row.get('ws_stop', ''), row.get('web_st', 0), row.get('fin_prog', 0), row.get('Город', ''), row.get('b_fin', ''), row.get('b_ban', ''), row.get('b_ign', ''), row.get('b_baners', ''), row.get('b_butt', ''), row.get('Реплики', ''), row.get('Статус покупки', ''), curator, row.get('pr1', ''), row.get('pr2', ''), row.get('pr3', ''), row.get('Канал трафика', ''), row.get('pr5', ''), row.get('GetCurse', ''), row.get('Ключ PR', ''), row.get('n_con', ''), row.get('Канал', ''), data_t, row.get('utm_source', ''), row.get('utm_medium', ''), row.get('utm_campaign', ''), row.get('utm_term', ''), row.get('utm_content', '')]
+
+        placeholders = ', '.join(['?' for _ in columns])
+        columns_str = ', '.join(columns)
+
+        query = f'''
+            INSERT INTO contacts ({columns_str})
+            VALUES ({placeholders})
+        '''
+
+        try:
+            cursor.execute(query, values)
+            user_data = dict(zip(columns, values))
+            send_to_google_forms(user_data, gog_url)
+        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_b', methods=['POST'])
+def upload_csv_b():
+    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_b(data)
+        verify_phone = request.form.get('verify_phone', '0')
+        add_curator = request.form.get('add_curator', '0')
+        print(f"Verify Phone: {verify_phone}")
+        print(f"Add Curator: {add_curator}")
+        insert_data_b(parsed_data, verify_phone, add_curator)
+        return jsonify({"message": "Data uploaded and inserted successfully"})
+    return jsonify({"error": "Invalid file format"}), 400
 
+@app.route('/upl_csv_b', methods=['GET'])
+def se_upl_csv_b():
+    api_sys_control = request.args.get('api_sys')
+    if api_sys_control != api_key_sys:
+        return "EUR 22", 200
+    return render_template('upl_csv_b.html')