q792602257andClaude Opus 4.6
65e3ed31f8
trading 不再管理 ラクマ 的购物车购买、支付与订单监控。删除 auth_site / auth_session / login_runner / models.AuthSite 中的 rakuma 分支与常量、 account.yaml.example 的 rakuma 段，并清理相关测试。scraping 侧的 ラクマ 抓取 API（/api/rakuma/*）保留不动。 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
90 lines
3.2 KiB
Python
90 lines
3.2 KiB
Python
"""登录态路由：查询与重新加载账号登录态
|
|
|
|
抓取接口全部匿名，只有加购与下单需要账号。登录本身不在这里做——乐天登录要过
|
|
reCAPTCHA 与设备验证，由 `scripts/login.py` 起有头浏览器人工完成一次，
|
|
本服务只读取落盘的 cookie。这里提供的是运维视角的两个动作：
|
|
|
|
- `/api/auth/status`：现在还登录着吗（默认真实打一次请求探测，不看缓存）
|
|
- `/api/auth/reload`：人工重新登录后，免重启服务重新读取登录态
|
|
"""
|
|
from fastapi import APIRouter, Depends
|
|
|
|
from app.shared.api import ApiResponse, get_container, require_bearer_token
|
|
from app.trading.container import TradingContainer
|
|
from app.trading.models import (
|
|
AuthReloadData,
|
|
AuthReloadRequest,
|
|
AuthSiteStatus,
|
|
AuthStatusData,
|
|
AuthStatusRequest,
|
|
)
|
|
from app.trading.services.auth_session import AuthStatus
|
|
|
|
router = APIRouter(prefix="/api/auth", tags=["auth"])
|
|
|
|
|
|
def _to_model(status: AuthStatus) -> AuthSiteStatus:
|
|
return AuthSiteStatus(**status.to_dict())
|
|
|
|
|
|
@router.post(
|
|
"/status",
|
|
response_model=ApiResponse[AuthStatusData],
|
|
dependencies=[Depends(require_bearer_token)],
|
|
)
|
|
async def auth_status(
|
|
payload: AuthStatusRequest,
|
|
container: TradingContainer = Depends(get_container),
|
|
) -> ApiResponse[AuthStatusData]:
|
|
"""查询账号登录态
|
|
|
|
`refresh=true`（默认）时会真实访问站点探测——登录态过期没有可靠的本地判据，
|
|
cookie 上的 expires 与服务端会话不是一回事，只能问站点。
|
|
不需要这次网络往返时传 `refresh=false`，此时返回上一次探测的缓存结果
|
|
（`logged_in` 为 null 表示从未探测过）。
|
|
|
|
`logged_in=false` 时，加购与下单接口会直接返回 5001，需要重新跑
|
|
`scripts/login.py --site <site>` 后调用 `/api/auth/reload`。
|
|
"""
|
|
sites = [payload.site.value] if payload.site else list(container.auth_session.sites)
|
|
statuses = []
|
|
for site in sites:
|
|
status = (
|
|
await container.auth_session.check(site)
|
|
if payload.refresh
|
|
else container.auth_session.status(site)
|
|
)
|
|
statuses.append(_to_model(status))
|
|
|
|
return ApiResponse[AuthStatusData](
|
|
success=True,
|
|
msg="success",
|
|
data=AuthStatusData(sites=statuses),
|
|
code=0,
|
|
)
|
|
|
|
|
|
@router.post(
|
|
"/reload",
|
|
response_model=ApiResponse[AuthReloadData],
|
|
dependencies=[Depends(require_bearer_token)],
|
|
)
|
|
async def auth_reload(
|
|
payload: AuthReloadRequest,
|
|
container: TradingContainer = Depends(get_container),
|
|
) -> ApiResponse[AuthReloadData]:
|
|
"""重新从磁盘加载登录态并立即探测
|
|
|
|
人工跑完 `scripts/login.py` 后调用，避免为了换一套 cookie 重启整个服务。
|
|
"""
|
|
sites = [payload.site.value] if payload.site else list(container.auth_session.sites)
|
|
reloaded = {site: container.auth_session.reload(site) for site in sites}
|
|
statuses = [_to_model(await container.auth_session.check(site)) for site in sites]
|
|
|
|
return ApiResponse[AuthReloadData](
|
|
success=True,
|
|
msg="success",
|
|
data=AuthReloadData(reloaded=reloaded, sites=statuses),
|
|
code=0,
|
|
)
|