mirror of
https://gitee.com/houhuan/TrendRadar.git
synced 2025-12-21 16:07:15 +08:00
508 lines
18 KiB
Python
508 lines
18 KiB
Python
"""
|
||
日期解析工具
|
||
|
||
支持多种自然语言日期格式解析，包括相对日期和绝对日期。
|
||
"""
|
||
|
||
import re
|
||
from datetime import datetime, timedelta
|
||
from typing import Tuple, Dict, Optional
|
||
|
||
from .errors import InvalidParameterError
|
||
|
||
|
||
class DateParser:
|
||
"""日期解析器类"""
|
||
|
||
# 中文日期映射
|
||
CN_DATE_MAPPING = {
|
||
"今天": 0,
|
||
"昨天": 1,
|
||
"前天": 2,
|
||
"大前天": 3,
|
||
}
|
||
|
||
# 英文日期映射
|
||
EN_DATE_MAPPING = {
|
||
"today": 0,
|
||
"yesterday": 1,
|
||
}
|
||
|
||
# 日期范围表达式（用于 resolve_date_range_expression）
|
||
RANGE_EXPRESSIONS = {
|
||
# 中文表达式
|
||
"今天": "today",
|
||
"昨天": "yesterday",
|
||
"本周": "this_week",
|
||
"这周": "this_week",
|
||
"当前周": "this_week",
|
||
"上周": "last_week",
|
||
"本月": "this_month",
|
||
"这个月": "this_month",
|
||
"当前月": "this_month",
|
||
"上月": "last_month",
|
||
"上个月": "last_month",
|
||
"最近3天": "last_3_days",
|
||
"近3天": "last_3_days",
|
||
"最近7天": "last_7_days",
|
||
"近7天": "last_7_days",
|
||
"最近一周": "last_7_days",
|
||
"过去一周": "last_7_days",
|
||
"最近14天": "last_14_days",
|
||
"近14天": "last_14_days",
|
||
"最近两周": "last_14_days",
|
||
"过去两周": "last_14_days",
|
||
"最近30天": "last_30_days",
|
||
"近30天": "last_30_days",
|
||
"最近一个月": "last_30_days",
|
||
"过去一个月": "last_30_days",
|
||
# 英文表达式
|
||
"today": "today",
|
||
"yesterday": "yesterday",
|
||
"this week": "this_week",
|
||
"current week": "this_week",
|
||
"last week": "last_week",
|
||
"this month": "this_month",
|
||
"current month": "this_month",
|
||
"last month": "last_month",
|
||
"last 3 days": "last_3_days",
|
||
"past 3 days": "last_3_days",
|
||
"last 7 days": "last_7_days",
|
||
"past 7 days": "last_7_days",
|
||
"past week": "last_7_days",
|
||
"last 14 days": "last_14_days",
|
||
"past 14 days": "last_14_days",
|
||
"last 30 days": "last_30_days",
|
||
"past 30 days": "last_30_days",
|
||
"past month": "last_30_days",
|
||
}
|
||
|
||
# 星期映射
|
||
WEEKDAY_CN = {
|
||
"一": 0, "二": 1, "三": 2, "四": 3,
|
||
"五": 4, "六": 5, "日": 6, "天": 6
|
||
}
|
||
|
||
WEEKDAY_EN = {
|
||
"monday": 0, "tuesday": 1, "wednesday": 2, "thursday": 3,
|
||
"friday": 4, "saturday": 5, "sunday": 6
|
||
}
|
||
|
||
@staticmethod
|
||
def parse_date_query(date_query: str) -> datetime:
|
||
"""
|
||
解析日期查询字符串
|
||
|
||
支持的格式：
|
||
- 相对日期（中文）：今天、昨天、前天、大前天、N天前
|
||
- 相对日期（英文）：today、yesterday、N days ago
|
||
- 星期（中文）：上周一、上周二、本周三
|
||
- 星期（英文）：last monday、this friday
|
||
- 绝对日期：2025-10-10、10月10日、2025年10月10日
|
||
|
||
Args:
|
||
date_query: 日期查询字符串
|
||
|
||
Returns:
|
||
datetime对象
|
||
|
||
Raises:
|
||
InvalidParameterError: 日期格式无法识别
|
||
|
||
Examples:
|
||
>>> DateParser.parse_date_query("今天")
|
||
datetime(2025, 10, 11)
|
||
>>> DateParser.parse_date_query("昨天")
|
||
datetime(2025, 10, 10)
|
||
>>> DateParser.parse_date_query("3天前")
|
||
datetime(2025, 10, 8)
|
||
>>> DateParser.parse_date_query("2025-10-10")
|
||
datetime(2025, 10, 10)
|
||
"""
|
||
if not date_query or not isinstance(date_query, str):
|
||
raise InvalidParameterError(
|
||
"日期查询字符串不能为空",
|
||
suggestion="请提供有效的日期查询，如：今天、昨天、2025-10-10"
|
||
)
|
||
|
||
date_query = date_query.strip().lower()
|
||
|
||
# 1. 尝试解析中文常用相对日期
|
||
if date_query in DateParser.CN_DATE_MAPPING:
|
||
days_ago = DateParser.CN_DATE_MAPPING[date_query]
|
||
return datetime.now() - timedelta(days=days_ago)
|
||
|
||
# 2. 尝试解析英文常用相对日期
|
||
if date_query in DateParser.EN_DATE_MAPPING:
|
||
days_ago = DateParser.EN_DATE_MAPPING[date_query]
|
||
return datetime.now() - timedelta(days=days_ago)
|
||
|
||
# 3. 尝试解析 "N天前" 或 "N days ago"
|
||
cn_days_ago_match = re.match(r'(\d+)\s*天前', date_query)
|
||
if cn_days_ago_match:
|
||
days = int(cn_days_ago_match.group(1))
|
||
if days > 365:
|
||
raise InvalidParameterError(
|
||
f"天数过大: {days}天",
|
||
suggestion="请使用小于365天的相对日期或使用绝对日期"
|
||
)
|
||
return datetime.now() - timedelta(days=days)
|
||
|
||
en_days_ago_match = re.match(r'(\d+)\s*days?\s+ago', date_query)
|
||
if en_days_ago_match:
|
||
days = int(en_days_ago_match.group(1))
|
||
if days > 365:
|
||
raise InvalidParameterError(
|
||
f"天数过大: {days}天",
|
||
suggestion="请使用小于365天的相对日期或使用绝对日期"
|
||
)
|
||
return datetime.now() - timedelta(days=days)
|
||
|
||
# 4. 尝试解析星期（中文）：上周一、本周三
|
||
cn_weekday_match = re.match(r'(上|本)周([一二三四五六日天])', date_query)
|
||
if cn_weekday_match:
|
||
week_type = cn_weekday_match.group(1) # 上 或 本
|
||
weekday_str = cn_weekday_match.group(2)
|
||
target_weekday = DateParser.WEEKDAY_CN[weekday_str]
|
||
return DateParser._get_date_by_weekday(target_weekday, week_type == "上")
|
||
|
||
# 5. 尝试解析星期（英文）：last monday、this friday
|
||
en_weekday_match = re.match(r'(last|this)\s+(monday|tuesday|wednesday|thursday|friday|saturday|sunday)', date_query)
|
||
if en_weekday_match:
|
||
week_type = en_weekday_match.group(1) # last 或 this
|
||
weekday_str = en_weekday_match.group(2)
|
||
target_weekday = DateParser.WEEKDAY_EN[weekday_str]
|
||
return DateParser._get_date_by_weekday(target_weekday, week_type == "last")
|
||
|
||
# 6. 尝试解析绝对日期：YYYY-MM-DD
|
||
iso_date_match = re.match(r'(\d{4})-(\d{1,2})-(\d{1,2})', date_query)
|
||
if iso_date_match:
|
||
year = int(iso_date_match.group(1))
|
||
month = int(iso_date_match.group(2))
|
||
day = int(iso_date_match.group(3))
|
||
try:
|
||
return datetime(year, month, day)
|
||
except ValueError as e:
|
||
raise InvalidParameterError(
|
||
f"无效的日期: {date_query}",
|
||
suggestion=f"日期值错误: {str(e)}"
|
||
)
|
||
|
||
# 7. 尝试解析中文日期：MM月DD日 或 YYYY年MM月DD日
|
||
cn_date_match = re.match(r'(?:(\d{4})年)?(\d{1,2})月(\d{1,2})日', date_query)
|
||
if cn_date_match:
|
||
year_str = cn_date_match.group(1)
|
||
month = int(cn_date_match.group(2))
|
||
day = int(cn_date_match.group(3))
|
||
|
||
# 如果没有年份，使用当前年份
|
||
if year_str:
|
||
year = int(year_str)
|
||
else:
|
||
year = datetime.now().year
|
||
# 如果月份大于当前月份，说明是去年
|
||
current_month = datetime.now().month
|
||
if month > current_month:
|
||
year -= 1
|
||
|
||
try:
|
||
return datetime(year, month, day)
|
||
except ValueError as e:
|
||
raise InvalidParameterError(
|
||
f"无效的日期: {date_query}",
|
||
suggestion=f"日期值错误: {str(e)}"
|
||
)
|
||
|
||
# 8. 尝试解析斜杠格式：YYYY/MM/DD 或 MM/DD
|
||
slash_date_match = re.match(r'(?:(\d{4})/)?(\d{1,2})/(\d{1,2})', date_query)
|
||
if slash_date_match:
|
||
year_str = slash_date_match.group(1)
|
||
month = int(slash_date_match.group(2))
|
||
day = int(slash_date_match.group(3))
|
||
|
||
if year_str:
|
||
year = int(year_str)
|
||
else:
|
||
year = datetime.now().year
|
||
current_month = datetime.now().month
|
||
if month > current_month:
|
||
year -= 1
|
||
|
||
try:
|
||
return datetime(year, month, day)
|
||
except ValueError as e:
|
||
raise InvalidParameterError(
|
||
f"无效的日期: {date_query}",
|
||
suggestion=f"日期值错误: {str(e)}"
|
||
)
|
||
|
||
# 如果所有格式都不匹配
|
||
raise InvalidParameterError(
|
||
f"无法识别的日期格式: {date_query}",
|
||
suggestion=(
|
||
"支持的格式:\n"
|
||
"- 相对日期: 今天、昨天、前天、3天前、today、yesterday、3 days ago\n"
|
||
"- 星期: 上周一、本周三、last monday、this friday\n"
|
||
"- 绝对日期: 2025-10-10、10月10日、2025年10月10日"
|
||
)
|
||
)
|
||
|
||
@staticmethod
|
||
def _get_date_by_weekday(target_weekday: int, is_last_week: bool) -> datetime:
|
||
"""
|
||
根据星期几获取日期
|
||
|
||
Args:
|
||
target_weekday: 目标星期 (0=周一, 6=周日)
|
||
is_last_week: 是否是上周
|
||
|
||
Returns:
|
||
datetime对象
|
||
"""
|
||
today = datetime.now()
|
||
current_weekday = today.weekday()
|
||
|
||
# 计算天数差
|
||
if is_last_week:
|
||
# 上周的某一天
|
||
days_diff = current_weekday - target_weekday + 7
|
||
else:
|
||
# 本周的某一天
|
||
days_diff = current_weekday - target_weekday
|
||
if days_diff < 0:
|
||
days_diff += 7
|
||
|
||
return today - timedelta(days=days_diff)
|
||
|
||
@staticmethod
|
||
def format_date_folder(date: datetime) -> str:
|
||
"""
|
||
将日期格式化为文件夹名称
|
||
|
||
Args:
|
||
date: datetime对象
|
||
|
||
Returns:
|
||
文件夹名称，格式: YYYY-MM-DD
|
||
|
||
Examples:
|
||
>>> DateParser.format_date_folder(datetime(2025, 10, 11))
|
||
'2025-10-11'
|
||
"""
|
||
return date.strftime("%Y-%m-%d")
|
||
|
||
@staticmethod
|
||
def validate_date_not_future(date: datetime) -> None:
|
||
"""
|
||
验证日期不在未来
|
||
|
||
Args:
|
||
date: 待验证的日期
|
||
|
||
Raises:
|
||
InvalidParameterError: 日期在未来
|
||
"""
|
||
if date.date() > datetime.now().date():
|
||
raise InvalidParameterError(
|
||
f"不能查询未来的日期: {date.strftime('%Y-%m-%d')}",
|
||
suggestion="请使用今天或过去的日期"
|
||
)
|
||
|
||
@staticmethod
|
||
def validate_date_not_too_old(date: datetime, max_days: int = 365) -> None:
|
||
"""
|
||
验证日期不太久远
|
||
|
||
Args:
|
||
date: 待验证的日期
|
||
max_days: 最大天数
|
||
|
||
Raises:
|
||
InvalidParameterError: 日期太久远
|
||
"""
|
||
days_ago = (datetime.now().date() - date.date()).days
|
||
if days_ago > max_days:
|
||
raise InvalidParameterError(
|
||
f"日期太久远: {date.strftime('%Y-%m-%d')} ({days_ago}天前)",
|
||
suggestion=f"请查询{max_days}天内的数据"
|
||
)
|
||
|
||
@staticmethod
|
||
def resolve_date_range_expression(expression: str) -> Dict:
|
||
"""
|
||
将自然语言日期表达式解析为标准日期范围
|
||
|
||
这是专门为 MCP 工具设计的方法，用于在服务器端解析日期表达式，
|
||
避免 AI 模型自己计算日期导致的不一致问题。
|
||
|
||
Args:
|
||
expression: 自然语言日期表达式，支持：
|
||
- 单日: "今天", "昨天", "today", "yesterday"
|
||
- 本周/上周: "本周", "上周", "this week", "last week"
|
||
- 本月/上月: "本月", "上月", "this month", "last month"
|
||
- 最近N天: "最近7天", "最近30天", "last 7 days", "last 30 days"
|
||
- 动态N天: "最近5天", "last 10 days"
|
||
|
||
Returns:
|
||
解析结果字典：
|
||
{
|
||
"success": True,
|
||
"expression": "本周",
|
||
"normalized": "this_week",
|
||
"date_range": {
|
||
"start": "2025-11-18",
|
||
"end": "2025-11-24"
|
||
},
|
||
"current_date": "2025-11-26",
|
||
"description": "本周（周一到周日）"
|
||
}
|
||
|
||
Raises:
|
||
InvalidParameterError: 无法识别的日期表达式
|
||
|
||
Examples:
|
||
>>> DateParser.resolve_date_range_expression("本周")
|
||
{"success": True, "date_range": {"start": "2025-11-18", "end": "2025-11-24"}, ...}
|
||
|
||
>>> DateParser.resolve_date_range_expression("最近7天")
|
||
{"success": True, "date_range": {"start": "2025-11-20", "end": "2025-11-26"}, ...}
|
||
"""
|
||
if not expression or not isinstance(expression, str):
|
||
raise InvalidParameterError(
|
||
"日期表达式不能为空",
|
||
suggestion="请提供有效的日期表达式，如：本周、最近7天、last week"
|
||
)
|
||
|
||
expression_lower = expression.strip().lower()
|
||
today = datetime.now()
|
||
today_str = today.strftime("%Y-%m-%d")
|
||
|
||
# 1. 尝试匹配预定义表达式
|
||
normalized = DateParser.RANGE_EXPRESSIONS.get(expression_lower)
|
||
|
||
# 2. 尝试匹配动态 "最近N天" / "last N days" 模式
|
||
if not normalized:
|
||
# 中文: 最近N天
|
||
cn_match = re.match(r'最近(\d+)天', expression_lower)
|
||
if cn_match:
|
||
days = int(cn_match.group(1))
|
||
normalized = f"last_{days}_days"
|
||
|
||
# 英文: last N days
|
||
en_match = re.match(r'(?:last|past)\s+(\d+)\s+days?', expression_lower)
|
||
if en_match:
|
||
days = int(en_match.group(1))
|
||
normalized = f"last_{days}_days"
|
||
|
||
if not normalized:
|
||
# 提供支持的表达式列表
|
||
supported_cn = ["今天", "昨天", "本周", "上周", "本月", "上月",
|
||
"最近7天", "最近30天", "最近N天"]
|
||
supported_en = ["today", "yesterday", "this week", "last week",
|
||
"this month", "last month", "last 7 days", "last N days"]
|
||
raise InvalidParameterError(
|
||
f"无法识别的日期表达式: {expression}",
|
||
suggestion=f"支持的表达式:\n中文: {', '.join(supported_cn)}\n英文: {', '.join(supported_en)}"
|
||
)
|
||
|
||
# 3. 根据 normalized 类型计算日期范围
|
||
start_date, end_date, description = DateParser._calculate_date_range(
|
||
normalized, today
|
||
)
|
||
|
||
return {
|
||
"success": True,
|
||
"expression": expression,
|
||
"normalized": normalized,
|
||
"date_range": {
|
||
"start": start_date.strftime("%Y-%m-%d"),
|
||
"end": end_date.strftime("%Y-%m-%d")
|
||
},
|
||
"current_date": today_str,
|
||
"description": description
|
||
}
|
||
|
||
@staticmethod
|
||
def _calculate_date_range(
|
||
normalized: str,
|
||
today: datetime
|
||
) -> Tuple[datetime, datetime, str]:
|
||
"""
|
||
根据标准化的日期类型计算实际日期范围
|
||
|
||
Args:
|
||
normalized: 标准化的日期类型
|
||
today: 当前日期
|
||
|
||
Returns:
|
||
(start_date, end_date, description) 元组
|
||
"""
|
||
# 单日类型
|
||
if normalized == "today":
|
||
return today, today, "今天"
|
||
|
||
if normalized == "yesterday":
|
||
yesterday = today - timedelta(days=1)
|
||
return yesterday, yesterday, "昨天"
|
||
|
||
# 本周（周一到周日）
|
||
if normalized == "this_week":
|
||
# 计算本周一
|
||
weekday = today.weekday() # 0=周一, 6=周日
|
||
start = today - timedelta(days=weekday)
|
||
end = start + timedelta(days=6)
|
||
# 如果本周还没结束，end 不能超过今天
|
||
if end > today:
|
||
end = today
|
||
return start, end, f"本周（周一到周日，{start.strftime('%m-%d')} 至 {end.strftime('%m-%d')}）"
|
||
|
||
# 上周（上周一到上周日）
|
||
if normalized == "last_week":
|
||
weekday = today.weekday()
|
||
# 本周一
|
||
this_monday = today - timedelta(days=weekday)
|
||
# 上周一
|
||
start = this_monday - timedelta(days=7)
|
||
end = start + timedelta(days=6)
|
||
return start, end, f"上周（{start.strftime('%m-%d')} 至 {end.strftime('%m-%d')}）"
|
||
|
||
# 本月（本月1日到今天）
|
||
if normalized == "this_month":
|
||
start = today.replace(day=1)
|
||
return start, today, f"本月（{start.strftime('%m-%d')} 至 {today.strftime('%m-%d')}）"
|
||
|
||
# 上月（上月1日到上月最后一天）
|
||
if normalized == "last_month":
|
||
# 上月最后一天 = 本月1日 - 1天
|
||
first_of_this_month = today.replace(day=1)
|
||
end = first_of_this_month - timedelta(days=1)
|
||
start = end.replace(day=1)
|
||
return start, end, f"上月（{start.strftime('%Y-%m-%d')} 至 {end.strftime('%Y-%m-%d')}）"
|
||
|
||
# 最近N天 (last_N_days 格式)
|
||
match = re.match(r'last_(\d+)_days', normalized)
|
||
if match:
|
||
days = int(match.group(1))
|
||
start = today - timedelta(days=days - 1) # 包含今天，所以是 days-1
|
||
return start, today, f"最近{days}天（{start.strftime('%m-%d')} 至 {today.strftime('%m-%d')}）"
|
||
|
||
# 兜底：返回今天
|
||
return today, today, "今天（默认）"
|
||
|
||
@staticmethod
|
||
def get_supported_expressions() -> Dict[str, list]:
|
||
"""
|
||
获取支持的日期表达式列表
|
||
|
||
Returns:
|
||
分类的表达式列表
|
||
"""
|
||
return {
|
||
"单日": ["今天", "昨天", "today", "yesterday"],
|
||
"周": ["本周", "上周", "this week", "last week"],
|
||
"月": ["本月", "上月", "this month", "last month"],
|
||
"最近N天": ["最近3天", "最近7天", "最近14天", "最近30天",
|
||
"last 3 days", "last 7 days", "last 14 days", "last 30 days"],
|
||
"动态天数": ["最近N天", "last N days"]
|
||
}
|