Update app.py

app.py

@@ -1671,59 +1671,61 @@ def clean_phone_number(phone_number):
 
 def insert_data(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('"')
+    for db_name in DATABASES:
+        conn = sqlite3.connect(db_name)
+        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(phone)
+            # Очистка номера телефона
+            phone = clean_phone_number(phone)
 
-        cursor.execute("SELECT 1 FROM contacts WHERE email = ? OR phone = ?", (email, phone))
-        user_exists = cursor.fetchone()
+            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 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(phone)
-        else:
-            ws_st = row.get('ws_st', '')
+            if add_curator == "1":
+                curator = curators[current_curator_index]
+                current_curator_index = (current_curator_index + 1) % len(curators)
+            else:
+                curator = row.get('Куратор', '')
 
-        # Обновленные столбцы и значения
-        columns = ['name', 'phone', 'email', 'vk_id', 'chat_id', 'ws_st', 'city', 'curator', 'purchase_status', 'traffic_channel', 'date', 'pr_key', 'channel', 'date']
-        values = [name, phone, email, row.get('ВКонтакте', ''), row.get('Телеграм', ''), ws_st, row.get('Город', ''), curator, row.get('Статус покупки', ''), row.get('Канал трафика', ''), data_t, row.get('Ключ PR', ''), row.get('Канал', ''), row.get('Дата', '')]
+            if verify_phone == "1":
+                ws_st = verify_phone_number(phone)
+            else:
+                ws_st = row.get('ws_st', '')
 
-        placeholders = ', '.join(['?' for _ in columns])
-        columns_str = ', '.join(columns)
+            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('b_mess', ''), row.get('shop_st', ''), curator, row.get('pr1', ''), row.get('pr2', ''), row.get('pr3', ''), row.get('pr4', ''), row.get('pr5', ''), row.get('gc_url', ''), row.get('key_pr', ''), row.get('n_con', ''), row.get('canal', ''), data_t, row.get('utm_source', ''), row.get('utm_medium', ''), row.get('utm_campaign', ''), row.get('utm_term', ''), row.get('utm_content', '')]
 
-        query = f'''
-            INSERT INTO contacts ({columns_str})
-            VALUES ({placeholders})
-        '''
+            placeholders = ', '.join(['?' for _ in columns])
+            columns_str = ', '.join(columns)
 
-        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
+            query = f'''
+                INSERT INTO contacts ({columns_str})
+                VALUES ({placeholders})
+            '''
 
-    conn.commit()
-    conn.close()
+            try:
+                cursor.execute(query, values)
+                # Отправка данных в Google Forms
+                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', methods=['POST'])
 def upload_csv():
@@ -1752,6 +1754,16 @@ def se_upl_csv():
         return "EUR 22", 200
     return render_template('upl_csv.html')
 
+@app.route('/download_csv', methods=['GET'])
+def download_csv():
+    data = io.StringIO()
+    writer = csv.writer(data)
+    writer.writerow(['Имя', 'WhatsApp', 'Email', 'Город'])  # Заголовки для CSV
+    # Заполнение CSV данными, здесь можно добавить логику получения данных
+    writer.writerow(['Олег', '79033456555', 'risaga@mail.ru', 'Москва'])
+
+    data.seek(0)
+    return send_file(data, mimetype='text/csv', attachment_filename='download.csv', as_attachment=True)