132 lines
4.4 KiB
Python
132 lines
4.4 KiB
Python
|
# 号码分类工具
|
|||
|
|
# 按MEMORY.md最新规则 (2026-04-02)
|
|||
|
|
# 优先级：数字越小越有价值
|
|||
|
|
|
|||
|
|
CATEGORIES = [
|
|||
|
|
# 1. 圆圆号：无.123457，只能用0689
|
|||
|
|
{"name": "圆圆号", "cannot_use": ".123457", "must_have": ""},
|
|||
|
|
# 2. 倒置号：无23457，可用01689必须有1
|
|||
|
|
{"name": "倒置号", "cannot_use": "23457", "must_have": "1"},
|
|||
|
|
# 3. 金马王：无12347，可用05689必须有5
|
|||
|
|
{"name": "金马王", "cannot_use": "12347", "must_have": "5"},
|
|||
|
|
# 4. 金马号：无2347，可用015689必须有1和5
|
|||
|
|
{"name": "金马号", "cannot_use": "2347", "must_have": "15"},
|
|||
|
|
# 5. 金山王：无12457，可用03689必须有3
|
|||
|
|
{"name": "金山王", "cannot_use": "12457", "must_have": "3"},
|
|||
|
|
# 6. 天马王：无1247，可用035689必须有3和5
|
|||
|
|
{"name": "天马王", "cannot_use": "1247", "must_have": "35"},
|
|||
|
|
# 7. 金山号：无2457，可用013689必须有1和3
|
|||
|
|
{"name": "金山号", "cannot_use": "2457", "must_have": "13"},
|
|||
|
|
# 8. 天马号：无247，可用0135689必须有1、3和5
|
|||
|
|
{"name": "天马号", "cannot_use": "247", "must_have": "135"},
|
|||
|
|
# 9. 朦胧王：无13457
|
|||
|
|
{"name": "朦胧王", "cannot_use": "13457", "must_have": ""},
|
|||
|
|
# 10. 朦胧号：无3457
|
|||
|
|
{"name": "朦胧号", "cannot_use": "3457", "must_have": ""},
|
|||
|
|
# 11. 如意号：无1347
|
|||
|
|
{"name": "如意号", "cannot_use": "1347", "must_have": ""},
|
|||
|
|
# 12. 钻石号：无347
|
|||
|
|
{"name": "钻石号", "cannot_use": "347", "must_have": ""},
|
|||
|
|
# 13. 永恒号：无47
|
|||
|
|
{"name": "永恒号", "cannot_use": "47", "must_have": ""},
|
|||
|
|
# 14. 无4号：不包含4
|
|||
|
|
{"name": "无4号", "cannot_use": "4", "must_have": ""},
|
|||
|
|
# 15. 通货：含4
|
|||
|
|
{"name": "通货", "cannot_use": "", "must_have": "4"},
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
|
|||
|
|
def extract_digits(serial: str) -> dict:
|
|||
|
|
"""提取冠字号中的数字部分"""
|
|||
|
|
if not serial:
|
|||
|
|
return {"digits": "", "type": "single"}
|
|||
|
|
|
|||
|
|
nums = serial.replace("J", "").replace(",", "").replace(".", "").strip()
|
|||
|
|
nums = "".join(c for c in nums if c.isdigit())
|
|||
|
|
|
|||
|
|
if nums.endswith("01"):
|
|||
|
|
return {"digits": nums[:-2], "type": "hundred"}
|
|||
|
|
elif nums.endswith("1"):
|
|||
|
|
return {"digits": nums[:-1], "type": "ten"}
|
|||
|
|
else:
|
|||
|
|
return {"digits": nums, "type": "single"}
|
|||
|
|
|
|||
|
|
|
|||
|
|
def matches_category(digits: str, category: dict) -> bool:
|
|||
|
|
cannot_use = category.get("cannot_use", "")
|
|||
|
|
must_have = category.get("must_have", "")
|
|||
|
|
|
|||
|
|
# 检查不能用的数字
|
|||
|
|
for n in cannot_use:
|
|||
|
|
if n in digits:
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
# 检查必须有的数字
|
|||
|
|
if must_have:
|
|||
|
|
for n in must_have:
|
|||
|
|
if n not in digits:
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_number_category(serial: str) -> str:
|
|||
|
|
"""号码分类函数"""
|
|||
|
|
if not serial:
|
|||
|
|
return ""
|
|||
|
|
|
|||
|
|
# 提取数字
|
|||
|
|
info = extract_digits(serial)
|
|||
|
|
digits = info["digits"]
|
|||
|
|
|
|||
|
|
if not digits:
|
|||
|
|
return ""
|
|||
|
|
|
|||
|
|
# 根据类型取对应位数
|
|||
|
|
digits_type = info["type"]
|
|||
|
|
if digits_type == "hundred":
|
|||
|
|
# 标百看后6位
|
|||
|
|
check_digits = digits[-6:] if len(digits) >= 6 else digits
|
|||
|
|
elif digits_type == "ten":
|
|||
|
|
# 标十看后7位
|
|||
|
|
check_digits = digits[-7:] if len(digits) >= 7 else digits
|
|||
|
|
else:
|
|||
|
|
# 散钞看全部
|
|||
|
|
check_digits = digits
|
|||
|
|
|
|||
|
|
# 按优先级匹配分类
|
|||
|
|
for category in CATEGORIES:
|
|||
|
|
if matches_category(check_digits, category):
|
|||
|
|
return category["name"]
|
|||
|
|
|
|||
|
|
return "通货"
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_number_category_color(category: str) -> str:
|
|||
|
|
"""获取分类颜色"""
|
|||
|
|
colors = {
|
|||
|
|
"圆圆号": "#ef4444", # 红
|
|||
|
|
"倒置号": "#f97316", # 橙
|
|||
|
|
"金马王": "#eab308", # 黄
|
|||
|
|
"金马号": "#84cc16", # 绿
|
|||
|
|
"金山王": "#22c55e", # 深绿
|
|||
|
|
"天马王": "#14b8a6", # 青
|
|||
|
|
"金山号": "#06b6d4", # 蓝
|
|||
|
|
"天马号": "#0ea5e9", # 浅蓝
|
|||
|
|
"朦胧王": "#6366f1", # 靛蓝
|
|||
|
|
"朦胧号": "#8b5cf6", # 紫
|
|||
|
|
"如意号": "#a855f7", # 深紫
|
|||
|
|
"钻石号": "#d946ef", # 品红
|
|||
|
|
"永恒号": "#ec4899", # 粉红
|
|||
|
|
"无4号": "#64748b", # 灰
|
|||
|
|
"通货": "#9ca3af", # 浅灰
|
|||
|
|
}
|
|||
|
|
return colors.get(category, "#9ca3af")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_category_priority(category: str) -> int:
|
|||
|
|
"""获取分类优先级（数字越小越高级）"""
|
|||
|
|
for i, cat in enumerate(CATEGORIES, 1):
|
|||
|
|
if cat["name"] == category:
|
|||
|
|
return i
|
|||
|
|
return 999
|