DmitrMakeev
commited on
Commit
•
8f688ea
1
Parent(s):
2e7a805
Update app.py
Browse files
app.py
CHANGED
@@ -1819,7 +1819,10 @@ def add_user_bot_route():
|
|
1819 |
|
1820 |
|
1821 |
# Список кураторов
|
1822 |
-
curators = ["Anna", "Ekaterina", "Ivan"]
|
|
|
|
|
|
|
1823 |
|
1824 |
# Шаблон сопоставления для кураторов
|
1825 |
mapping_template_cur = {
|
@@ -1829,20 +1832,6 @@ mapping_template_cur = {
|
|
1829 |
'curator': 'curator'
|
1830 |
}
|
1831 |
|
1832 |
-
def add_user_cur(curator_index):
|
1833 |
-
# Проверяем, что индекс куратора передан и допустим
|
1834 |
-
if curator_index is None:
|
1835 |
-
logging.error("Curator index is required")
|
1836 |
-
return False
|
1837 |
-
|
1838 |
-
if not (0 <= curator_index < len(curators)):
|
1839 |
-
logging.error(f"Invalid curator index: {curator_index}")
|
1840 |
-
return False
|
1841 |
-
|
1842 |
-
# Логируем текущего куратора
|
1843 |
-
logging.info(f"Current curator set to {curators[curator_index]}")
|
1844 |
-
return True
|
1845 |
-
|
1846 |
def add_user_to_db(db_name, user_data):
|
1847 |
conn = sqlite3.connect(db_name)
|
1848 |
cursor = conn.cursor()
|
@@ -1871,25 +1860,21 @@ def add_user_to_db(db_name, user_data):
|
|
1871 |
|
1872 |
@app.route('/add_user_cur', methods=['GET'])
|
1873 |
def add_user_cur_route():
|
1874 |
-
|
1875 |
-
curator_index = request.args.get('curator', type=int)
|
1876 |
|
1877 |
-
#
|
1878 |
-
|
1879 |
-
return jsonify({'status': 'error', 'message': 'Curator index is required'}), 400
|
1880 |
|
1881 |
-
#
|
1882 |
-
|
1883 |
-
# Получаем данные пользователя из параметров запроса
|
1884 |
-
user_data = {key: request.args.get(key, "") for key in mapping_template_cur.keys()}
|
1885 |
-
user_data['curator'] = curators[curator_index]
|
1886 |
|
1887 |
-
|
1888 |
-
|
1889 |
|
1890 |
-
|
1891 |
-
|
1892 |
-
|
|
|
1893 |
|
1894 |
|
1895 |
|
|
|
1819 |
|
1820 |
|
1821 |
# Список кураторов
|
1822 |
+
curators = ["Anna", "Ekaterina", "Ivan", "Maria", "Sergey", "Olga", "Alex", "Natalia", "Dmitry", "Elena"]
|
1823 |
+
|
1824 |
+
# Переменная для отслеживания текущего куратора
|
1825 |
+
current_curator_index = 0
|
1826 |
|
1827 |
# Шаблон сопоставления для кураторов
|
1828 |
mapping_template_cur = {
|
|
|
1832 |
'curator': 'curator'
|
1833 |
}
|
1834 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1835 |
def add_user_to_db(db_name, user_data):
|
1836 |
conn = sqlite3.connect(db_name)
|
1837 |
cursor = conn.cursor()
|
|
|
1860 |
|
1861 |
@app.route('/add_user_cur', methods=['GET'])
|
1862 |
def add_user_cur_route():
|
1863 |
+
global current_curator_index
|
|
|
1864 |
|
1865 |
+
# Получаем данные пользователя из параметров запроса
|
1866 |
+
user_data = {key: request.args.get(key, "") for key in mapping_template_cur.keys()}
|
|
|
1867 |
|
1868 |
+
# Назначаем текущего куратора
|
1869 |
+
user_data['curator'] = curators[current_curator_index]
|
|
|
|
|
|
|
1870 |
|
1871 |
+
# Добавляем пользователя в базу данных
|
1872 |
+
add_user_to_db('data_gc.db', user_data)
|
1873 |
|
1874 |
+
# Переходим к следующему куратору
|
1875 |
+
current_curator_index = (current_curator_index + 1) % len(curators)
|
1876 |
+
|
1877 |
+
return jsonify({'status': 'success', 'message': f'User added with curator {user_data["curator"]}'})
|
1878 |
|
1879 |
|
1880 |
|