Update app.py

app.py

@@ -1824,9 +1824,22 @@ curators = ["Anna", "Ekaterina", "Ivan"]
 # Переменная для отслеживания текущего куратора
 current_curator_index = 0
 
-def add_user_cur(curator_name):
+# Шаблон сопоставления для кураторов
+mapping_template_cur = {
+    'name': 'name',
+    'phone': 'phone',
+    'email': 'email',
+    'curator': 'curator'
+}
+
+def add_user_cur(user_data, mapping_template_cur):
     global current_curator_index
 
+    # Преобразование данных пользователя на основе шаблона сопоставления
+    transformed_data = {db_column: user_data.get(json_key, "") for json_key, db_column in mapping_template_cur.items()}
+
+    # Назначение куратора
+    curator_name = transformed_data.get('curator')
     if curator_name is None:
         curator_name = curators[current_curator_index]
     elif curator_name not in curators:
@@ -1839,12 +1852,12 @@ def add_user_cur(curator_name):
 
 @app.route('/add_user_cur', methods=['GET'])
 def add_user_cur_route():
-    curator_name = request.args.get('curator_name')
+    user_data = {key: request.args.get(key, "") for key in mapping_template_cur.keys()}
 
-    if add_user_cur(curator_name):
-        return jsonify({'status': 'success', 'message': f'Current curator set to {curator_name}'})
+    if add_user_cur(user_data, mapping_template_cur):
+        return jsonify({'status': 'success', 'message': f'Current curator set to {curators[current_curator_index]}'})
     else:
-        return jsonify({'status': 'error', 'message': f'Curator {curator_name} not found in the list of curators'}), 400
+        return jsonify({'status': 'error', 'message': f'Curator {user_data.get("curator")} not found in the list of curators'}), 400