from __future__ import annotations import re from datetime import date from decimal import Decimal, InvalidOperation, ROUND_HALF_UP LOCATION_BANDS = { "premium": ("北京", "上海", "广州", "深圳", "杭州", "南京", "苏州", "成都", "重庆", "天津"), "remote": ("新疆", "西藏", "青海", "甘肃", "宁夏", "内蒙古", "海南", "香港", "澳门", "台湾", "海外", "国外"), "coastal": ("上海", "广州", "深圳", "厦门", "福州", "青岛", "大连", "宁波", "舟山", "海口", "三亚", "天津"), } TRANSPORT_PRICE_BASE = { "火车": {"default": Decimal("360"), "premium": Decimal("520"), "remote": Decimal("900"), "coastal": Decimal("520")}, "飞机": {"default": Decimal("850"), "premium": Decimal("1100"), "remote": Decimal("1800"), "coastal": Decimal("1050")}, "轮船": {"default": Decimal("320"), "premium": Decimal("480"), "remote": Decimal("680"), "coastal": Decimal("520")}, } LODGING_DAILY_BASE = { "default": Decimal("420"), "premium": Decimal("600"), "remote": Decimal("520"), "coastal": Decimal("500"), } ALLOWANCE_DAILY_BASE = { "default": Decimal("100"), "premium": Decimal("120"), "remote": Decimal("120"), "coastal": Decimal("110"), } def parse_application_days(days_text: str) -> int: match = re.search(r"\d+", str(days_text or "")) if not match: return 1 return max(1, int(match.group(0))) def parse_application_money(value: object) -> Decimal: normalized = re.sub(r"[^\d.\-]", "", str(value or "").replace(",", "")) if not normalized: return Decimal("0") try: return Decimal(normalized) except (InvalidOperation, ValueError): return Decimal("0") def format_application_money(value: Decimal | int | float | str) -> str: amount = parse_application_money(value) if not isinstance(value, Decimal) else value quantized = amount.quantize(Decimal("0.01"), rounding=ROUND_HALF_UP) if quantized == quantized.to_integral(): return f"{int(quantized):,}" return f"{quantized:,.2f}".rstrip("0").rstrip(".") def normalize_application_transport_mode(value: str) -> str: text = str(value or "").strip() if re.search(r"飞机|机票|航班|乘机|坐飞机", text): return "飞机" if re.search(r"轮船|船票|客轮|渡轮|邮轮|坐船", text): return "轮船" if re.search(r"火车|高铁|动车|铁路|列车", text): return "火车" return text if text in TRANSPORT_PRICE_BASE else "" def resolve_application_location_band(location: str) -> str: text = str(location or "").strip() if any(keyword in text for keyword in LOCATION_BANDS["remote"]): return "remote" if any(keyword in text for keyword in LOCATION_BANDS["premium"]): return "premium" if any(keyword in text for keyword in LOCATION_BANDS["coastal"]): return "coastal" return "default" def _round_to_ten(value: Decimal) -> Decimal: return (value / Decimal("10")).quantize(Decimal("1"), rounding=ROUND_HALF_UP) * Decimal("10") def parse_application_start_date(time_text: object) -> str: match = re.search(r"(20\d{2})[年\-/.](\d{1,2})[月\-/.](\d{1,2})", str(time_text or "")) if not match: return "" try: return date(int(match.group(1)), int(match.group(2)), int(match.group(3))).isoformat() except ValueError: return "" def _resolve_ticket_price_factor(query_date: str) -> Decimal: if not query_date: return Decimal("1.00") try: parsed = date.fromisoformat(query_date) except ValueError: return Decimal("1.00") factor = Decimal("1.00") if parsed.weekday() == 0: factor += Decimal("0.04") if parsed.weekday() in (4, 6): factor += Decimal("0.08") if parsed.month in {1, 2, 7, 8, 10}: factor += Decimal("0.06") jitter = (parsed.year + parsed.month * 13 + parsed.day * 7) % 7 - 3 factor += Decimal(jitter) / Decimal("100") if factor < Decimal("0.88"): return Decimal("0.88") if factor > Decimal("1.22"): return Decimal("1.22") return factor def _resolve_mock_query_latency_ms(query_date: str, mode: str, location_band: str) -> int: try: parsed = date.fromisoformat(query_date) if query_date else None except ValueError: parsed = None seed = len(mode) * 43 + len(location_band) * 29 if parsed: seed += parsed.year + parsed.month * 17 + parsed.day * 31 return 360 + seed % 420 def build_application_system_estimate( *, transport_mode: str, location: str, days_text: str, time_text: object = "", lodging_amount: object = None, allowance_amount: object = None, ) -> dict[str, str]: mode = normalize_application_transport_mode(transport_mode) if not mode: return {} days = parse_application_days(days_text) location_band = resolve_application_location_band(location) query_date = parse_application_start_date(time_text) price_factor = _resolve_ticket_price_factor(query_date) simulated_latency_ms = _resolve_mock_query_latency_ms(query_date, mode, location_band) transport_one_way = TRANSPORT_PRICE_BASE[mode].get(location_band, TRANSPORT_PRICE_BASE[mode]["default"]) transport_amount = _round_to_ten(transport_one_way * Decimal("2") * price_factor) lodging = parse_application_money(lodging_amount) allowance = parse_application_money(allowance_amount) lodging_daily = LODGING_DAILY_BASE.get(location_band, LODGING_DAILY_BASE["default"]) allowance_daily = ALLOWANCE_DAILY_BASE.get(location_band, ALLOWANCE_DAILY_BASE["default"]) if lodging <= 0: lodging = lodging_daily * days if allowance <= 0: allowance = allowance_daily * days total_amount = transport_amount + lodging + allowance transport_display = format_application_money(transport_amount) lodging_display = format_application_money(lodging) allowance_display = format_application_money(allowance) total_display = format_application_money(total_amount) return { "amount": f"{total_display}元", "lodging_daily_cap": f"{format_application_money(lodging_daily)}元/天", "subsidy_daily_cap": f"{format_application_money(allowance_daily)}元/天", "transport_policy": f"预估交通费用 {transport_display}元", "policy_estimate": ( f"交通 {transport_display}元 + 住宿 {lodging_display}元" f" + 补贴 {allowance_display}元 = {total_display}元（{days}天）" ), "matched_city": str(location or "").strip(), "transport_estimated_amount": f"{transport_display}元", "transport_estimate_date": query_date, "transport_query_latency_ms": str(simulated_latency_ms), "policy_total_amount": f"{total_display}元", "estimate_source": "mock_ticket_price_query_v1", "estimate_confidence": "mock", } def _is_pending_application_amount(value: str) -> bool: normalized = str(value or "").strip() return not normalized or normalized in {"待测算", "待补充", "未知"} def apply_application_system_estimate_to_facts(facts: dict[str, str]) -> None: estimate = build_application_system_estimate( transport_mode=str(facts.get("transport_mode") or ""), location=str(facts.get("matched_city") or facts.get("location") or ""), days_text=str(facts.get("days") or ""), time_text=facts.get("time") or "", lodging_amount=facts.get("hotel_amount") or None, allowance_amount=facts.get("allowance_amount") or None, ) if not estimate: return if _is_pending_application_amount(facts.get("amount", "")): facts["amount"] = estimate["amount"] field_map = { "lodging_daily_cap": "lodging_daily_cap", "subsidy_daily_cap": "subsidy_daily_cap", "transport_policy": "transport_policy", "policy_estimate": "policy_estimate", "matched_city": "matched_city", "transport_estimated_amount": "transport_estimated_amount", "transport_estimate_date": "transport_estimate_date", "transport_query_latency_ms": "transport_query_latency_ms", "transport_estimate_source": "estimate_source", "transport_estimate_confidence": "estimate_confidence", "policy_total_amount": "policy_total_amount", } for target, source in field_map.items(): if not str(facts.get(target) or "").strip() and estimate.get(source): facts[target] = estimate[source]