Update app.py

app.py

@@ -1818,6 +1818,7 @@ def add_user_bot_route():
 # Конец примера шаблона сопоставления для настройки переменных записываемых в базу данных.
 
 
+# Список кураторов
 curators = ["Anna", "Ekaterina", "Ivan"]
 
 # Шаблон сопоставления для кураторов
@@ -1829,6 +1830,7 @@ mapping_template_cur = {
 }
 
 def add_user_cur(curator_index):
+    # Проверяем, что индекс куратора передан и допустим
     if curator_index is None:
         logging.error("Curator index is required")
         return False
@@ -1837,18 +1839,55 @@ def add_user_cur(curator_index):
         logging.error(f"Invalid curator index: {curator_index}")
         return False
 
+    # Логируем текущего куратора
     logging.info(f"Current curator set to {curators[curator_index]}")
     return True
 
+def add_user_to_db(db_name, user_data):
+    conn = sqlite3.connect(db_name)
+    cursor = conn.cursor()
+
+    # Преобразование данных пользователя на основе шаблона сопоставления
+    transformed_data = {db_column: user_data.get(json_key, "") for json_key, db_column in mapping_template_cur.items()}
+
+    # Добавление недостающих полей с пустыми строками
+    required_fields = [
+        "ad_url", "b_ban", "b_baners", "b_butt", "b_city", "b_fin", "b_ign", "b_mess",
+        "canal", "chat_id", "curator", "data_t", "fin_prog", "key_pr", "n_con",
+        "pr1", "pr2", "pr3", "pr4", "pr5", "shop_st", "vk_id", "web_st", "ws_st", "ws_stop"
+    ]
+    for field in required_fields:
+        if field not in transformed_data:
+            transformed_data[field] = ""
+
+    columns = ', '.join(transformed_data.keys())
+    placeholders = ', '.join('?' for _ in transformed_data)
+    insert_query = f"INSERT INTO contacts ({columns}) VALUES ({placeholders})"
+    insert_values = list(transformed_data.values())
+
+    cursor.execute(insert_query, insert_values)
+    conn.commit()
+    conn.close()
+
 @app.route('/add_user_cur', methods=['GET'])
 def add_user_cur_route():
+    # Получаем индекс куратора из параметров запроса
     curator_index = request.args.get('curator', type=int)
 
+    # Проверяем, что индекс куратора передан
     if curator_index is None:
         return jsonify({'status': 'error', 'message': 'Curator index is required'}), 400
 
+    # Устанавливаем текущего куратора
     if add_user_cur(curator_index):
-        return jsonify({'status': 'success', 'message': f'Current curator set to {curators[curator_index]}'})
+        # Получаем данные пользователя из параметров запроса
+        user_data = {key: request.args.get(key, "") for key in mapping_template_cur.keys()}
+        user_data['curator'] = curators[curator_index]
+
+        # Добавляем пользователя в базу данных
+        add_user_to_db('data_gc.db', user_data)
+
+        return jsonify({'status': 'success', 'message': f'User added with curator {curators[curator_index]}'})
     else:
         return jsonify({'status': 'error', 'message': f'Invalid curator index: {curator_index}'}), 400