Update app.py

app.py

@@ -1818,12 +1818,8 @@ def add_user_bot_route():
 # Конец примера шаблона сопоставления для настройки переменных записываемых в базу данных.
 
 
-# Список кураторов
 curators = ["Anna", "Ekaterina", "Ivan"]
 
-# Переменная для отслеживания текущего куратора
-current_curator_index = 0
-
 # Шаблон сопоставления для кураторов
 mapping_template_cur = {
     'name': 'name',
@@ -1832,32 +1828,29 @@ mapping_template_cur = {
     'curator': 'curator'
 }
 
-def add_user_cur(user_data, mapping_template_cur):
-    global current_curator_index
+def add_user_cur(curator_index):
+    if curator_index is None:
+        logging.error("Curator index is required")
+        return False
 
-    # Преобразование данных пользователя на основе шаблона сопоставления
-    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:
-        logging.error(f"Curator {curator_name} not found in the list of curators")
+    if not (0 <= curator_index < len(curators)):
+        logging.error(f"Invalid curator index: {curator_index}")
         return False
 
-    current_curator_index = curators.index(curator_name)
-    logging.info(f"Current curator set to {curator_name}")
+    logging.info(f"Current curator set to {curators[curator_index]}")
     return True
 
 @app.route('/add_user_cur', methods=['GET'])
 def add_user_cur_route():
-    user_data = {key: request.args.get(key, "") for key in mapping_template_cur.keys()}
+    curator_index = request.args.get('curator', type=int)
+
+    if curator_index is None:
+        return jsonify({'status': 'error', 'message': 'Curator index is required'}), 400
 
-    if add_user_cur(user_data, mapping_template_cur):
-        return jsonify({'status': 'success', 'message': f'Current curator set to {curators[current_curator_index]}'})
+    if add_user_cur(curator_index):
+        return jsonify({'status': 'success', 'message': f'Current curator set to {curators[curator_index]}'})
     else:
-        return jsonify({'status': 'error', 'message': f'Curator {user_data.get("curator")} not found in the list of curators'}), 400
+        return jsonify({'status': 'error', 'message': f'Invalid curator index: {curator_index}'}), 400