951c12a208
FastAPI + Jinja2 + MariaDB web application with registration, login, profile, password reset, and email confirmation flows. All UI in Russian. Styled with Evotor brand colors. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
109 lines
4.3 KiB
Python
109 lines
4.3 KiB
Python
import uuid
|
||
from datetime import datetime, timedelta, timezone
|
||
|
||
from fastapi import APIRouter, Request, Depends
|
||
from fastapi.responses import RedirectResponse
|
||
from fastapi.templating import Jinja2Templates
|
||
from sqlalchemy.orm import Session
|
||
|
||
from web.auth import hash_password
|
||
from web.config import settings
|
||
from web.database import get_db
|
||
from web.models import User
|
||
from web.schemas import validate_reset_password
|
||
|
||
router = APIRouter()
|
||
templates = Jinja2Templates(directory="web/templates")
|
||
|
||
|
||
@router.get("/forgot-password")
|
||
def forgot_form(request: Request):
|
||
return templates.TemplateResponse("forgot_password.html", {"request": request, "user": None})
|
||
|
||
|
||
@router.post("/forgot-password")
|
||
async def forgot_submit(request: Request, db: Session = Depends(get_db)):
|
||
form = await request.form()
|
||
email = form.get("email", "").strip()
|
||
|
||
if email:
|
||
user = db.query(User).filter(User.email == email).first()
|
||
if user:
|
||
token = uuid.uuid4().hex
|
||
user.password_reset_token = token
|
||
user.password_reset_expires = datetime.now(timezone.utc) + timedelta(
|
||
minutes=settings.PASSWORD_RESET_EXPIRE_MINUTES
|
||
)
|
||
db.commit()
|
||
|
||
reset_url = f"{settings.BASE_URL}/reset-password?token={token}"
|
||
print("=" * 40)
|
||
print("СБРОС ПАРОЛЯ")
|
||
print(f"Пользователь: {user.email}")
|
||
print(f"Ссылка: {reset_url}")
|
||
print(f"Действительна: {settings.PASSWORD_RESET_EXPIRE_MINUTES} мин.")
|
||
print("=" * 40)
|
||
|
||
return templates.TemplateResponse("message.html", {
|
||
"request": request, "user": None,
|
||
"title": "Сброс пароля",
|
||
"message": "Если аккаунт с таким email существует, ссылка для сброса пароля выведена в консоль сервера.",
|
||
})
|
||
|
||
|
||
@router.get("/reset-password")
|
||
def reset_form(request: Request, token: str, db: Session = Depends(get_db)):
|
||
user = db.query(User).filter(User.password_reset_token == token).first()
|
||
if not user or not user.password_reset_expires:
|
||
return templates.TemplateResponse("message.html", {
|
||
"request": request, "user": None,
|
||
"title": "Ошибка", "message": "Неверная или устаревшая ссылка.",
|
||
})
|
||
|
||
if datetime.now(timezone.utc) > user.password_reset_expires.replace(tzinfo=timezone.utc):
|
||
return templates.TemplateResponse("message.html", {
|
||
"request": request, "user": None,
|
||
"title": "Ошибка", "message": "Срок действия ссылки истек.",
|
||
})
|
||
|
||
return templates.TemplateResponse("reset_password.html", {
|
||
"request": request, "user": None, "token": token,
|
||
})
|
||
|
||
|
||
@router.post("/reset-password")
|
||
async def reset_submit(request: Request, token: str, db: Session = Depends(get_db)):
|
||
user = db.query(User).filter(User.password_reset_token == token).first()
|
||
if not user or not user.password_reset_expires:
|
||
return templates.TemplateResponse("message.html", {
|
||
"request": request, "user": None,
|
||
"title": "Ошибка", "message": "Неверная или устаревшая ссылка.",
|
||
})
|
||
|
||
if datetime.now(timezone.utc) > user.password_reset_expires.replace(tzinfo=timezone.utc):
|
||
return templates.TemplateResponse("message.html", {
|
||
"request": request, "user": None,
|
||
"title": "Ошибка", "message": "Срок действия ссылки истек.",
|
||
})
|
||
|
||
form = await request.form()
|
||
data = dict(form)
|
||
errors = validate_reset_password(data)
|
||
|
||
if errors:
|
||
return templates.TemplateResponse("reset_password.html", {
|
||
"request": request, "user": None, "token": token, "errors": errors,
|
||
})
|
||
|
||
user.password_hash = hash_password(data["password"])
|
||
user.password_reset_token = None
|
||
user.password_reset_expires = None
|
||
db.commit()
|
||
|
||
return templates.TemplateResponse("message.html", {
|
||
"request": request, "user": None,
|
||
"title": "Пароль изменен",
|
||
"message": "Ваш пароль успешно изменен. Теперь вы можете войти.",
|
||
"link": "/login", "link_text": "Войти",
|
||
})
|