Простая регистрация: /register, автовход, ALLOW_REGISTRATION, лимит для новых

webui.py

@@ -22,7 +22,7 @@ from functools import wraps
 
 from flask import (Flask, jsonify, render_template, request, session,
                    redirect, url_for, send_file)
-from werkzeug.security import check_password_hash
+from werkzeug.security import check_password_hash, generate_password_hash
 
 from hh_search import (search_vacancies, fetch_vacancy_details, fetch_my_resumes,
                        parse_cookies, get_proxies)
@@ -208,6 +208,47 @@ def login_page():
     return render_template("login.html")
 
 
+@app.route("/register", methods=["GET", "POST"])
+def register_page():
+    """Простая регистрация: логин + пароль. Отключается через ALLOW_REGISTRATION=0."""
+    if os.environ.get("ALLOW_REGISTRATION", "1") != "1":
+        if request.method == "POST":
+            return jsonify({"error": "Регистрация отключена"}), 403
+        return redirect(url_for("login_page"))
+
+    if request.method == "POST":
+        data = request.get_json(force=True)
+        username = (data.get("username") or "").strip()
+        password = data.get("password") or ""
+        confirm = data.get("confirm") or ""
+
+        # Проверки
+        if not re.match(r"^[a-zA-Z0-9_.-]{3,30}$", username):
+            return jsonify({"error": "Логин: 3-30 символов, буквы/цифры/точка/дефис/подчёркивание"}), 400
+        if len(password) < 6:
+            return jsonify({"error": "Пароль: минимум 6 символов"}), 400
+        if password != confirm:
+            return jsonify({"error": "Пароли не совпадают"}), 400
+
+        users = load_users()
+        if username in users:
+            return jsonify({"error": "Логин уже занят"}), 409
+
+        limit = int(os.environ.get("REGISTER_LLM_LIMIT", "50") or 0)
+        users[username] = {
+            "password_hash": generate_password_hash(password),
+            "llm_limit": limit,
+            "llm_usage": {},
+            "created": date.today().isoformat(),
+            "registered": True,
+        }
+        save_users(users)
+        os.makedirs(os.path.join(USERS_DIR, username, "accounts"), exist_ok=True)
+        session["user"] = username  # сразу входим
+        return jsonify({"ok": True, "limit": limit})
+    return render_template("register.html")
+
+
 @app.route("/logout")
 def logout():
     session.clear()