Update app.py

app.py

@@ -1819,7 +1819,85 @@ def add_user_bot_route():
 
 
 
+# Список кураторов
+curators = ["Anna", "Ekaterina", "Ivan"]
 
+# Переменная для отслеживания текущего куратора
+current_curator_index = 0
+
+# Пример шаблона сопоставления для настройки переменных записываемых в базу данных.
+mapping_template3 = {
+    'name': 'name',
+    'phone': 'phone',
+    'email': 'email'
+}
+
+def add_user_bot(db_name, user_data, mapping_template3):
+    global current_curator_index
+
+    conn = sqlite3.connect(db_name)
+    cursor = conn.cursor()
+
+    # Проверка наличия пользователя по email
+    email = user_data.get('email')
+    cursor.execute("SELECT 1 FROM contacts WHERE email = ?", (email,))
+    if cursor.fetchone() is not None:
+        logging.warning(f"User with email {email} already exists. Skipping insert.")
+        conn.close()
+        return
+
+    # Преобразование данных пользователя на основе шаблона сопоставления
+    transformed_data = {db_column: user_data.get(json_key, "") for json_key, db_column in mapping_template3.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] = ""
+
+    # Назначение куратора
+    transformed_data['curator'] = curators[current_curator_index]
+    current_curator_index = (current_curator_index + 1) % len(curators)
+
+    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_bot', methods=['GET'])
+def add_user_bot_route():
+    db_name = request.args.get('db_name', 'data_gc.db')  # Имя базы данных из параметра запроса
+    user_data = {key: request.args.get(key, "") for key in mapping_template3.keys()}
+
+    logging.debug(f"User data: {user_data}")
+
+    try:
+        add_user_bot(db_name, user_data, mapping_template3)
+        return jsonify({'status': 'success', 'message': f'User added to {db_name}'})
+    except Exception as e:
+        logging.error(f"Error adding user: {e}")
+        return jsonify({'status': 'error', 'message': str(e)}), 500
+
+@app.route('/add_user_cur', methods=['POST'])
+def add_user_cur_route():
+    global current_curator_index
+
+    data = request.json
+    curator_name = data.get('curator_name')
+
+    if curator_name not in curators:
+        return jsonify({'status': 'error', 'message': f'Curator {curator_name} not found in the list of curators'}), 400
+
+    current_curator_index = curators.index(curator_name)
+    return jsonify({'status': 'success', 'message': f'Current curator set to {curator_name}'})