Update app.py

app.py

@@ -1570,8 +1570,8 @@ mapp_templates = {
 
 DATABASE_NAME3 = 'data_gc.db'
 
-def add_or_update_contact(contact_data):
-    conn = sqlite3.connect(DATABASE_NAME3)
+def add_or_update_contact(contact_data, db_name):
+    conn = sqlite3.connect(db_name)
     cursor = conn.cursor()
 
     email = contact_data.get('email')
@@ -1584,14 +1584,12 @@ def add_or_update_contact(contact_data):
     msk_now = utc_now.replace(tzinfo=pytz.utc).astimezone(msk_tz)
     contact_data['data_t'] = msk_now.strftime('%Y-%m-%d %H:%M:%S')
 
-    # Список всех возможных полей
     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', 'gc_url', 'key_pr', 'n_con', 'canal', 'data_on', 'data_t', 'utm_source', 'utm_medium', 'utm_campaign', 'utm_term', 'utm_content', 'gcpc'
     ]
 
-    # Устанавливаем значения по умолчанию для отсутствующих полей
     for field in fields:
         if field not in contact_data:
             contact_data[field] = ''
@@ -1610,15 +1608,16 @@ def add_or_update_contact(contact_data):
         cursor.execute(insert_query, tuple(contact_data[field] for field in fields))
 
     conn.commit()
-    replace_null_with_empty_string(conn)
     conn.close()
+
 @app.route('/ver_cur', methods=['GET'])
 def add_data_ver_cur():
     global current_curator_index
 
-    veref_on_off = request.args.get('ver', '0')  # Включает "1" и выключает "0" верификацию номера вместо verifikation_start
-    curator_on_off = request.args.get('cur', '0')  # Включает "1" и выключает "0" назначение куратора
-    
+    veref_on_off = request.args.get('ver', '0')
+    curator_on_off = request.args.get('cur', '0')
+    db_name = request.args.get('db', 'data_gc.db')  # Получаем имя базы данных из запроса
+
     template_key = request.args.get('template_key', 'avp')
     mapping_template_cur = mapp_templates.get(template_key, mt_avp)
 
@@ -1645,20 +1644,20 @@ def add_data_ver_cur():
         user_data['ws_st'] = user_data.get('ws_st', '')
 
     try:
-        add_or_update_contact(user_data)
+        add_or_update_contact(user_data, db_name)
         return jsonify({'status': 'success', 'message': f'User added with curator {user_data.get("curator", "not assigned")}'})
     except Exception as e:
         logging.error(f"Error adding user: {e}")
         return jsonify({'status': 'error', 'message': str(e)}), 500
 
-
 @app.route('/ver_bonus', methods=['GET'])
 def add_data_ver_bonus():
     global current_curator_index
 
-    veref_on_off = request.args.get('ver', '0')  # Включает "1" и выключает "0" верификацию номера вместо verifikation_start
-    curator_on_off = request.args.get('cur', '0')  # Включает "1" и выключает "0" назначение куратора
-    
+    veref_on_off = request.args.get('ver', '0')
+    curator_on_off = request.args.get('cur', '0')
+    db_name = request.args.get('db', 'data_gc.db')  # Получаем имя базы данных из запроса
+
     template_key = request.args.get('template_key', 'avp')
     mapping_template_cur = mapp_templates.get(template_key, mt_avp)
 
@@ -1684,11 +1683,10 @@ def add_data_ver_bonus():
     else:
         user_data['ws_st'] = user_data.get('ws_st', '')
 
-    # Обработка kol для pr1
     email = user_data.get('email')
     new_pr1 = user_data.get('kol')
     if email and new_pr1:
-        conn = sqlite3.connect(DATABASE_NAME3)
+        conn = sqlite3.connect(db_name)
         cursor = conn.cursor()
         cursor.execute("SELECT pr1 FROM contacts WHERE email = ?", (email,))
         result = cursor.fetchone()
@@ -1699,7 +1697,7 @@ def add_data_ver_bonus():
         conn.close()
 
     try:
-        add_or_update_contact(user_data)
+        add_or_update_contact(user_data, db_name)
         return jsonify({'status': 'success', 'message': f'User added with curator {user_data.get("curator", "not assigned")}'})
     except Exception as e:
         logging.error(f"Error adding user: {e}")
@@ -1727,7 +1725,6 @@ def add_data_ver_bonus():
 
 
 
-
 
 
 def clean_phone_number_ss(phone_number):