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 |
|