lzf_0626
8ac0b794ff
接 b3da1b6。上一条只改了凌晨规则与日报日界，剩下几处一并收掉： 1. risk_query_service.py:77-78：REST 的 start_time/end_time 是**裸 datetime**， 原先原样透传去比库内 UTC 列，而 Agent 路径本来就带时区 （risk_natural_language.py:117）——同一条筛选条件在界面与对话里会查出不同结果。 timeutil 新增 from_local：裸值按**北京时间**解释（面向中国客户的业务系统， 填表人的预期就是本地时间），带时区的按其自身时区处理。它与 to_utc_naive 的区别 正在裸值上：取库里的值用后者，接客户端输入用这个。 2. risk_scan_service.py 与 risk_judgement_service.py 的 _age：一处用 UTC 日期、 一处用服务器 date.today()，生日边界上同一客户会差一岁、65 岁阈值可能翻面。 统一走 local_date(北京时间)。 3. risk_daily_report_service.py:122 的 report_date 与 :244 的 created_today： 原先取 UTC 日期，北京 08:00 之前会把"今天新增的预警"算成昨天。 ruff / mypy(135 文件) / 603 unit+contract 全绿。
84 lines
3.5 KiB
Python
84 lines
3.5 KiB
Python
"""时区换算：库内存 UTC naive，边界判断与展示按配置时区（默认北京时间）。
|
||
|
||
**为什么需要它**：风控的定时规则曾直接取 `confirmed_at.hour` 判断"凌晨"，而库里存的是
|
||
UTC —— `[0,6)` UTC 实际是北京时间 **08:00–14:00**，于是「凌晨时段小额操作」这条规则把
|
||
整个上午的交易都判成了凌晨（`docs/25` P1 #6）。日报也按 UTC 日切，却按北京时间展示，
|
||
两处对同一天的理解差 8 小时。
|
||
|
||
**约定**（与 `app/infrastructure/db.py:15-21` 的说明一致）：
|
||
|
||
- 库内 DATETIME 为 **UTC naive**，不带时区；
|
||
- 任何"这是本地几点 / 哪一天"的判断，都必须先经过本模块换算；
|
||
- 展示同样走这里，保证与 `get_settings().timezone` 一致。
|
||
"""
|
||
|
||
from datetime import UTC, date, datetime, time, timedelta
|
||
from zoneinfo import ZoneInfo
|
||
|
||
from app.core.config import get_settings
|
||
|
||
|
||
def local_zone() -> ZoneInfo:
|
||
"""业务判断与展示统一使用的本地时区（默认 `Asia/Shanghai`）。"""
|
||
return ZoneInfo(get_settings().timezone)
|
||
|
||
|
||
def to_local(value: datetime) -> datetime:
|
||
"""把库内的 UTC naive 时间换算成本地时区。
|
||
|
||
已经带时区的值按其原时区处理——那说明它来自库外（例如请求参数），
|
||
按它自己声明的时区解释才是对的。
|
||
"""
|
||
aware = value if value.tzinfo is not None else value.replace(tzinfo=UTC)
|
||
return aware.astimezone(local_zone())
|
||
|
||
|
||
def to_utc_naive(value: datetime) -> datetime:
|
||
"""把任意时区的时间换算回**库内格式**（UTC naive），用于构造查询条件。
|
||
|
||
查询参数必须经过这一步：拿本地时间直接去比库内的 UTC 列，会整体差 8 小时。
|
||
"""
|
||
aware = value if value.tzinfo is not None else value.replace(tzinfo=UTC)
|
||
return aware.astimezone(UTC).replace(tzinfo=None)
|
||
|
||
|
||
def from_local(value: datetime) -> datetime:
|
||
"""把**客户端传来的时间参数**换算成库内格式（UTC naive）。
|
||
|
||
REST 的时间参数是裸 `datetime`（`api/schemas/risk.py:32-33`），不带时区信息。
|
||
对面向中国客户的业务系统，裸值按**本地（北京）时间**解释才符合填表人的预期：
|
||
当成 UTC 会让前端填的"今天 00:00"实际查到昨天 08:00 起的数据。
|
||
带时区的值仍按它自己声明的时区处理。
|
||
|
||
与 `to_utc_naive` 的区别就在裸值上：这个把裸值当**本地**，那个当 **UTC**。
|
||
取库里的值用后者，接客户端输入用这个。
|
||
"""
|
||
aware = value if value.tzinfo is not None else value.replace(tzinfo=local_zone())
|
||
return aware.astimezone(UTC).replace(tzinfo=None)
|
||
|
||
|
||
def local_hour(value: datetime) -> int:
|
||
"""库内时间对应的**本地**小时（0-23）。
|
||
|
||
用于"是否凌晨"这类按时段判断的规则——绝不能用 `value.hour`，
|
||
那是 UTC 小时。
|
||
"""
|
||
return to_local(value).hour
|
||
|
||
|
||
def local_date(value: datetime) -> date:
|
||
"""库内时间对应的**本地**日期。"""
|
||
return to_local(value).date()
|
||
|
||
|
||
def local_day_bounds(value: datetime) -> tuple[datetime, datetime]:
|
||
"""库内时间所在的**本地自然日**，换算回库内格式的起止时刻 `[start, end)`。
|
||
|
||
返回的是 UTC naive：它要拿去查库内的 UTC 列。若返回本地时间，
|
||
区间会与库内值整体错开 8 小时——这正是原先日报日界出错的成因。
|
||
"""
|
||
local = to_local(value)
|
||
start_local = datetime.combine(local.date(), time.min, tzinfo=local.tzinfo)
|
||
start = to_utc_naive(start_local)
|
||
return start, start + timedelta(days=1)
|