Update app.py

app.py

@@ -1702,7 +1702,7 @@ def verify_phone_number(phone_number):
         print("Verification not started")
         return "false"  # Возвращаем значение, что WhatsApp не существует
 
-def add_or_update_contact(contact_data):
+def add_or_update_contact(contact_data, mapping_template_cur):
     conn = sqlite3.connect(DATABASE_NAME3)
     cursor = conn.cursor()
 
@@ -1717,9 +1717,11 @@ 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')
 
-    # Обработка параметра 'cur'
+    # Обработка параметра 'cur' в зависимости от шаблона
     if 'cur' in contact_data:
-        contact_data['curator'] = contact_data.pop('cur')
+        cur_key = next((value for key, value in mapping_template_cur.items() if value == 'cur'), None)
+        if cur_key:
+            contact_data[cur_key] = contact_data.pop('cur')
 
     cursor.execute("SELECT id FROM contacts WHERE email = ?", (email,))
     contact = cursor.fetchone()
@@ -1755,8 +1757,8 @@ def add_data_ver_cur():
     # Получаем данные пользователя из параметров запроса
     user_data = {key: request.args.get(key, "") for key in mapping_template_cur.keys()}
 
-    # Назначаем текущего куратора
-    user_data['curator'] = curators[current_curator_index]
+    # Добавляем параметр 'cur' в user_data
+    user_data['cur'] = request.args.get('cur', "")
 
     # Верификация номера телефона
     phone_verification_response = verify_phone_number(user_data['phone'])
@@ -1765,9 +1767,9 @@ def add_data_ver_cur():
 
     try:
         # Добавляем пользователя в базу данных
-        add_or_update_contact(user_data)
+        add_or_update_contact(user_data, mapping_template_cur)
         current_curator_index = (current_curator_index + 1) % len(curators)
-        return jsonify({'status': 'success', 'message': f'User added with curator {user_data["curator"]}'})
+        return jsonify({'status': 'success', 'message': f'User added with curator {user_data["cur"]}'})
     except Exception as e:
         logging.error(f"Error adding user: {e}")
         return jsonify({'status': 'error', 'message': str(e)}), 500