"""PMFlow 时间工具：统一使用北京时间（Asia/Shanghai）。"""

from __future__ import annotations

from datetime import datetime, timedelta, timezone

CN_TZ = timezone(timedelta(hours=8), name="Asia/Shanghai")


def now_cn_aware() -> datetime:
    """返回带时区的北京时间。"""
    return datetime.now(CN_TZ)


def now_cn_naive() -> datetime:
    """返回不带时区的北京时间（用于 TIMESTAMP WITHOUT TIME ZONE）。"""
    return now_cn_aware().replace(tzinfo=None)