225 lines
7.5 KiB
Python
225 lines
7.5 KiB
Python
"""纯函数工具：归一化/解析/格式化/数据提取，可单测。"""
|
|
from __future__ import annotations
|
|
|
|
import re
|
|
from urllib.parse import quote
|
|
|
|
try:
|
|
from .constants import (
|
|
GAMEMODE_REVERSE_MAPPING,
|
|
HERO_NAME_MAPPING,
|
|
HERO_NAME_REVERSE_MAPPING,
|
|
PLATFORM_MAPPING,
|
|
PLATFORM_REVERSE_MAPPING,
|
|
RANK_MAPPING,
|
|
REGION_MAPPING,
|
|
REGION_REVERSE_MAPPING,
|
|
ROLE_MAPPING,
|
|
)
|
|
except ImportError: # 作为独立模块导入时回退
|
|
from constants import (
|
|
GAMEMODE_REVERSE_MAPPING,
|
|
HERO_NAME_MAPPING,
|
|
HERO_NAME_REVERSE_MAPPING,
|
|
PLATFORM_MAPPING,
|
|
PLATFORM_REVERSE_MAPPING,
|
|
RANK_MAPPING,
|
|
REGION_MAPPING,
|
|
REGION_REVERSE_MAPPING,
|
|
ROLE_MAPPING,
|
|
)
|
|
|
|
_PLAYER_ID_RE = re.compile(r"^[A-Za-z0-9\u4e00-\u9fa5_\- ]{3,32}$")
|
|
_TIER_ROMAN = ["", "I", "II", "III", "IV", "V"]
|
|
|
|
DEFAULT_API_BASE_URL = "https://overfast-api.tekrop.fr"
|
|
|
|
|
|
def normalize_api_base_url(raw: str | None) -> str:
|
|
"""归一化 API 地址：去空格/去尾斜杠，无效时回退官方地址。"""
|
|
url = (raw or "").strip().rstrip("/")
|
|
if not url or not url.lower().startswith(("http://", "https://")):
|
|
return DEFAULT_API_BASE_URL
|
|
return url
|
|
|
|
|
|
def normalize_player_id(player_id: str) -> str:
|
|
"""# -> - 并 strip，不做合法性校验。"""
|
|
return player_id.replace("#", "-").strip()
|
|
|
|
|
|
def validate_player_id(player_id: str) -> str:
|
|
"""归一化 + 合法性校验，非法时抛 ValueError。
|
|
|
|
允许字母/数字/中文/_/-/空格，长度归一化后 3~32，且必须含 - 分隔符。
|
|
返回可直接拼 URL path 的 quote 结果。
|
|
"""
|
|
norm = normalize_player_id(player_id)
|
|
if len(norm) < 3 or len(norm) > 32:
|
|
raise ValueError(f"玩家 ID 长度非法: {player_id!r}")
|
|
if "/" in player_id or "\\" in player_id or ".." in norm:
|
|
raise ValueError(f"玩家 ID 非法: {player_id!r}")
|
|
# BattleTag 归一化后应为 Name-XXXX 形式；宽松校验：至少含 - 或纯字母数字
|
|
if not _PLAYER_ID_RE.match(norm):
|
|
raise ValueError(f"玩家 ID 含非法字符: {player_id!r}")
|
|
return norm
|
|
|
|
|
|
def quote_player_id(player_id: str) -> str:
|
|
"""返回 URL-safe 的玩家 ID（已归一化+quote）。"""
|
|
return quote(validate_player_id(player_id), safe="-")
|
|
|
|
|
|
def get_rank_display(division: str | None, tier: int | None) -> str:
|
|
if not division:
|
|
return "未定位"
|
|
rank_info = RANK_MAPPING.get(division.lower(), (division, ""))
|
|
tier_str = _TIER_ROMAN[tier] if tier and 1 <= tier <= 5 else ""
|
|
name = rank_info[0]
|
|
return f"{name}{' ' + tier_str if tier_str else ''}".strip()
|
|
|
|
|
|
def build_rank_data(summary: dict, platform: str) -> tuple[str, list[dict]]:
|
|
comp = summary.get("competitive")
|
|
if not isinstance(comp, dict):
|
|
comp = summary.get("competitive_ranks")
|
|
comp = comp if isinstance(comp, dict) else {}
|
|
|
|
shown_platform = platform
|
|
container = comp.get(platform)
|
|
if not isinstance(container, dict) or not container:
|
|
other_platform = "console" if platform == "pc" else "pc"
|
|
other = comp.get(other_platform)
|
|
if isinstance(other, dict) and other:
|
|
shown_platform = other_platform
|
|
container = other
|
|
else:
|
|
container = None
|
|
|
|
rows: list[dict] = []
|
|
if container:
|
|
for role_key in ["tank", "damage", "support"]:
|
|
role_cn = ROLE_MAPPING.get(role_key, (role_key, ""))[0]
|
|
role_text = f"{role_cn}"
|
|
role_data = container.get(role_key)
|
|
if isinstance(role_data, dict) and role_data.get("division"):
|
|
rows.append({
|
|
"role_text": role_text,
|
|
"rank_text": get_rank_display(role_data.get("division"), role_data.get("tier")),
|
|
"rank_icon": role_data.get("rank_icon") or "",
|
|
})
|
|
else:
|
|
rows.append({"role_text": role_text, "rank_text": "未定位", "rank_icon": ""})
|
|
return shown_platform, rows
|
|
|
|
|
|
def extract_top_heroes(stats: dict, platform: str, count: int = 3) -> list[tuple[str, int]]:
|
|
if not isinstance(stats, dict):
|
|
return []
|
|
for plat in (platform, "console" if platform == "pc" else "pc"):
|
|
plat_stats = stats.get(plat)
|
|
if not isinstance(plat_stats, dict):
|
|
continue
|
|
totals: dict[str, int] = {}
|
|
for gamemode in ("quickplay", "competitive"):
|
|
gm_stats = plat_stats.get(gamemode)
|
|
if not isinstance(gm_stats, dict):
|
|
continue
|
|
comparisons = gm_stats.get("heroes_comparisons")
|
|
if not isinstance(comparisons, dict):
|
|
continue
|
|
time_played = comparisons.get("time_played")
|
|
if not isinstance(time_played, dict):
|
|
continue
|
|
for item in time_played.get("values") or []:
|
|
if not isinstance(item, dict):
|
|
continue
|
|
hero = item.get("hero")
|
|
value = item.get("value") or 0
|
|
if hero:
|
|
try:
|
|
totals[hero] = totals.get(hero, 0) + int(value)
|
|
except (TypeError, ValueError):
|
|
continue
|
|
if totals:
|
|
return sorted(totals.items(), key=lambda kv: kv[1], reverse=True)[:count]
|
|
return []
|
|
|
|
|
|
def get_role_display(role_key: str) -> str:
|
|
return ROLE_MAPPING.get(role_key.lower(), (role_key, ""))[0]
|
|
|
|
|
|
def get_hero_name_cn(hero_key: str) -> str:
|
|
return HERO_NAME_MAPPING.get(hero_key.lower(), hero_key)
|
|
|
|
|
|
def resolve_hero_name(hero_input: str) -> str:
|
|
hero_input = hero_input.strip()
|
|
lowered = hero_input.lower()
|
|
if lowered in HERO_NAME_MAPPING:
|
|
return lowered
|
|
if hero_input in HERO_NAME_REVERSE_MAPPING:
|
|
return HERO_NAME_REVERSE_MAPPING[hero_input]
|
|
for cn_name, en_key in HERO_NAME_REVERSE_MAPPING.items():
|
|
if lowered in cn_name.lower() or cn_name.lower() in lowered:
|
|
return en_key
|
|
return lowered
|
|
|
|
|
|
def resolve_platform(platform_input: str) -> str | None:
|
|
if not platform_input:
|
|
return None
|
|
return PLATFORM_REVERSE_MAPPING.get(platform_input.strip().lower())
|
|
|
|
|
|
def get_platform_display(platform: str) -> str:
|
|
return PLATFORM_MAPPING.get(platform, platform)
|
|
|
|
|
|
def resolve_region(region_input: str) -> str | None:
|
|
if not region_input:
|
|
return None
|
|
return REGION_REVERSE_MAPPING.get(region_input.strip().lower())
|
|
|
|
|
|
def get_region_display(region: str) -> str:
|
|
return REGION_MAPPING.get(region, region)
|
|
|
|
|
|
def resolve_gamemode(mode_input: str) -> str | None:
|
|
if not mode_input:
|
|
return None
|
|
return GAMEMODE_REVERSE_MAPPING.get(mode_input.strip().lower())
|
|
|
|
|
|
def format_time_played(seconds: int | float | None) -> str:
|
|
if not seconds:
|
|
return "0小时"
|
|
total_minutes = int(seconds) // 60
|
|
hours = total_minutes // 60
|
|
minutes = total_minutes % 60
|
|
if hours > 0 and minutes > 0:
|
|
return f"{hours}小时{minutes}分钟"
|
|
elif hours > 0:
|
|
return f"{hours}小时"
|
|
return f"{minutes}分钟"
|
|
|
|
|
|
def format_time_short(seconds: int | float | None) -> str:
|
|
if not seconds:
|
|
return "0分钟"
|
|
total_minutes = int(seconds) // 60
|
|
hours = total_minutes / 60
|
|
if hours >= 1:
|
|
return f"{hours:.1f}小时"
|
|
return f"{total_minutes}分钟"
|
|
|
|
|
|
def format_number(num: int | float | None) -> str:
|
|
if num is None:
|
|
return "0"
|
|
if isinstance(num, float):
|
|
return f"{num:,.1f}" if num != int(num) else f"{int(num):,}"
|
|
return f"{num:,}"
|