DmitrMakeev commited on
Commit
8f688ea
1 Parent(s): 2e7a805

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -30
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
- if curator_index is None:
1879
- return jsonify({'status': 'error', 'message': 'Curator index is required'}), 400
1880
 
1881
- # Устанавливаем текущего куратора
1882
- if add_user_cur(curator_index):
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
- add_user_to_db('data_gc.db', user_data)
1889
 
1890
- return jsonify({'status': 'success', 'message': f'User added with curator {curators[curator_index]}'})
1891
- else:
1892
- return jsonify({'status': 'error', 'message': f'Invalid curator index: {curator_index}'}), 400
 
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