DmitrMakeev commited on
Commit
5fc7c41
1 Parent(s): f3a6868

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +37 -0
app.py CHANGED
@@ -1818,7 +1818,44 @@ def add_user_bot_route():
1818
  # Конец примера шаблона сопоставления для настройки переменных записываемых в базу данных.
1819
 
1820
 
 
 
1821
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1822
 
1823
 
1824
 
 
1818
  # Конец примера шаблона сопоставления для настройки переменных записываемых в базу данных.
1819
 
1820
 
1821
+ # Список кураторов
1822
+ curators = ["Anna", "Ekaterina", "Ivan"]
1823
 
1824
+ # Переменная для отслеживания текущего куратора
1825
+ current_curator_index = 0
1826
+
1827
+ # Шаблон сопоставления для кураторов
1828
+ mapping_template_cur = {
1829
+ 'name': 'name',
1830
+ 'phone': 'phone',
1831
+ 'email': 'email',
1832
+ 'curator': 'curator'
1833
+ }
1834
+
1835
+ def add_user_cur(user_data, mapping_template_cur):
1836
+ global current_curator_index
1837
+
1838
+ # Преобразование данных пользователя на основе шаблона сопоставления
1839
+ transformed_data = {db_column: user_data.get(json_key, "") for json_key, db_column in mapping_template_cur.items()}
1840
+
1841
+ # Назначение куратора
1842
+ curator_name = transformed_data.get('curator')
1843
+ if curator_name not in curators:
1844
+ logging.error(f"Curator {curator_name} not found in the list of curators")
1845
+ return False
1846
+
1847
+ current_curator_index = curators.index(curator_name)
1848
+ logging.info(f"Current curator set to {curator_name}")
1849
+ return True
1850
+
1851
+ @app.route('/add_user_cur', methods=['POST'])
1852
+ def add_user_cur_route():
1853
+ user_data = request.json
1854
+
1855
+ if add_user_cur(user_data, mapping_template_cur):
1856
+ return jsonify({'status': 'success', 'message': f'Current curator set to {user_data.get("curator")}'})
1857
+ else:
1858
+ return jsonify({'status': 'error', 'message': f'Curator {user_data.get("curator")} not found in the list of curators'}), 400
1859
 
1860
 
1861