DmitrMakeev
commited on
Commit
•
5fc7c41
1
Parent(s):
f3a6868
Update app.py
Browse files
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 |
|