Steven
b5170ad1ce
CI / backend (push) Canceled after 0s
短信登录关掉后，生产没有任何注册入口——这一版补上最普通的那条路：
填邮箱、收验证码、设密码，注册完直接进首页。
先发码再注册，不是"注册时顺带发信"
POST /auth/email/code 只做三件事：查重、限流、发码。
填完密码才被告知"这个邮箱注册过了"最气人，所以占用（2019）在发码前就回。
邮箱既是账号名也是唯一联系方式，所以必须真验一次码
验证通过时间落 app_user.email_verified_at，/auth/me 回 email_verified。
"随手填一个"会让账号变成没人能找回的孤岛。
app_user.phone 改为可空（迁移 0011）
邮箱注册的账号本来就没有手机号。user.phone 对这类账号回空串，
成员列表同理，不再伪装成一串星号。
防刷
同邮箱 60 秒一次（2022）、验证码 10 分钟有效、错 5 次即作废（2021）。
6 位数可以被刮，不设上限等于把注册口子交给脚本。
本机联调开关 EMAIL_CODE_DEV_ECHO
为 true 时把码回给调用方且不发真信——否则每跑一次 e2e 就往真实收件箱塞一封。
生产打开会让 validate_runtime() 拒绝启动。
前端
新增 /register（邮箱、验证码、密码、确认、昵称；60 秒倒计时；
服务端关着注册时整页只说"联系管理员开通"），登录页按 /auth/methods
决定要不要显示「还没有账号？用邮箱注册」。
验证
pytest 139 passed、ruff check / format 全绿
tsc --noEmit 0、vite build 通过
e2e register.mjs 7 项：登录页入口 → 拿码 → 注册落进首页 → 退出用邮箱+密码登回来
→ 同一邮箱再发码回 2019 → 控制台零报错
e2e login-email-smtp.mjs 15 项回归通过（含真发一封测试信）
e2e account.mjs 14 项、knowledge.mjs 全绿
顺带修的两处测试
test_account 的字段白名单补 email_verified（接口确实多了这个字段）；
test_config_prod 用 Settings(_env_file=None, ...) 构造，别被本机 .env 里的
EMAIL_CODE_DEV_ECHO=true 带偏——那个开关在生产是硬错误，本机却是跑 e2e 必需的。
126 lines
3.7 KiB
Python
126 lines
3.7 KiB
Python
"""短生命周期状态存储：验证码、幂等键。
|
|
|
|
dev/test 用进程内实现（不依赖 Redis 也能跑测试），prod 强制走 Redis。
|
|
接口刻意做小，将来换实现不影响调用方。
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import time
|
|
from typing import Protocol
|
|
|
|
from .config import get_settings
|
|
|
|
settings = get_settings()
|
|
|
|
|
|
class KVStore(Protocol):
|
|
async def set(self, key: str, value: str, ttl_seconds: int) -> None: ...
|
|
|
|
async def get(self, key: str) -> str | None: ...
|
|
|
|
async def delete(self, key: str) -> None: ...
|
|
|
|
async def incr(self, key: str, ttl_seconds: int) -> int: ...
|
|
|
|
|
|
class MemoryStore:
|
|
"""进程内 TTL 字典。仅限单进程 dev/test——多 worker 下不共享，别在生产用。"""
|
|
|
|
def __init__(self) -> None:
|
|
self._data: dict[str, tuple[str, float]] = {}
|
|
|
|
def _purge(self, key: str) -> None:
|
|
item = self._data.get(key)
|
|
if item and item[1] < time.time():
|
|
self._data.pop(key, None)
|
|
|
|
async def set(self, key: str, value: str, ttl_seconds: int) -> None:
|
|
self._data[key] = (value, time.time() + ttl_seconds)
|
|
|
|
async def get(self, key: str) -> str | None:
|
|
self._purge(key)
|
|
item = self._data.get(key)
|
|
return item[0] if item else None
|
|
|
|
async def delete(self, key: str) -> None:
|
|
self._data.pop(key, None)
|
|
|
|
async def incr(self, key: str, ttl_seconds: int) -> int:
|
|
self._purge(key)
|
|
current = int(self._data[key][0]) if key in self._data else 0
|
|
current += 1
|
|
expire_at = self._data[key][1] if key in self._data else time.time() + ttl_seconds
|
|
self._data[key] = (str(current), expire_at)
|
|
return current
|
|
|
|
|
|
class RedisStore:
|
|
def __init__(self, url: str) -> None:
|
|
from redis.asyncio import Redis # 延迟导入：dev 环境没有 redis 包也能起
|
|
|
|
self._client = Redis.from_url(url, decode_responses=True)
|
|
|
|
async def set(self, key: str, value: str, ttl_seconds: int) -> None:
|
|
await self._client.set(key, value, ex=ttl_seconds)
|
|
|
|
async def get(self, key: str) -> str | None:
|
|
return await self._client.get(key)
|
|
|
|
async def delete(self, key: str) -> None:
|
|
await self._client.delete(key)
|
|
|
|
async def incr(self, key: str, ttl_seconds: int) -> int:
|
|
value = await self._client.incr(key)
|
|
if value == 1:
|
|
await self._client.expire(key, ttl_seconds)
|
|
return int(value)
|
|
|
|
|
|
_store: KVStore | None = None
|
|
|
|
|
|
def get_store() -> KVStore:
|
|
global _store
|
|
if _store is None:
|
|
_store = RedisStore(settings.redis_url) if settings.env == "prod" else MemoryStore()
|
|
return _store
|
|
|
|
|
|
def reset_store_for_tests() -> None:
|
|
"""测试夹具用：清掉进程内状态。"""
|
|
global _store
|
|
_store = None
|
|
|
|
|
|
def dumps(value: object) -> str:
|
|
return json.dumps(value, ensure_ascii=False, separators=(",", ":"))
|
|
|
|
|
|
# ── 业务键约定 ──────────────────────────────────────────
|
|
def sms_code_key(phone: str) -> str:
|
|
return f"sms:login:{phone}"
|
|
|
|
|
|
def sms_throttle_key(phone: str) -> str:
|
|
return f"sms:throttle:{phone}"
|
|
|
|
|
|
# ── 邮箱注册 ────────────────────────────────────────────
|
|
def email_code_key(email: str) -> str:
|
|
return f"email:code:{email}"
|
|
|
|
|
|
def email_tries_key(email: str) -> str:
|
|
"""试错计数：错一次算一次，超过上限就把验证码作废，别让人拿 6 位数当彩票刮。"""
|
|
return f"email:tries:{email}"
|
|
|
|
|
|
def email_throttle_key(email: str) -> str:
|
|
return f"email:throttle:{email}"
|
|
|
|
|
|
def idempotency_key(user_id: str, key: str) -> str:
|
|
return f"idem:{user_id}:{key}"
|