feat: 导入 CapacityReport 初始源码

scripts/gen_license_code.py

@@ -0,0 +1,72 @@
+#!/usr/bin/env python3
+"""根据授权 key 生成激活码（与 app/services/license.py 算法一致）。"""
+
+from __future__ import annotations
+
+import argparse
+import sys
+from datetime import date, datetime
+from pathlib import Path
+
+ROOT = Path(__file__).resolve().parents[1]
+if str(ROOT) not in sys.path:
+    sys.path.insert(0, str(ROOT))
+
+from app.services.license import activation_hash, format_key_label
+
+
+def parse_key(raw: str) -> date:
+    text = raw.strip()
+    for fmt in ("%Y/%m/%d", "%Y-%m-%d"):
+        try:
+            return datetime.strptime(text, fmt).date()
+        except ValueError:
+            continue
+    raise ValueError("无法解析 key，请使用 YYYY/MM/DD 格式，例如 2026/06/20")
+
+
+def generate_code(key: str) -> tuple[str, str]:
+    expires_on = parse_key(key)
+    key_label = format_key_label(expires_on)
+    return key_label, activation_hash(expires_on)
+
+
+def build_parser() -> argparse.ArgumentParser:
+    parser = argparse.ArgumentParser(
+        description="输入授权 key（到期日期），输出对应激活码。",
+    )
+    parser.add_argument(
+        "key",
+        nargs="?",
+        help="授权 key，格式 YYYY/MM/DD 或 YYYY-MM-DD，例如 2026/06/20",
+    )
+    return parser
+
+
+def main(argv: list[str] | None = None) -> int:
+    args = build_parser().parse_args(argv)
+    key_input = args.key
+    if not key_input:
+        try:
+            key_input = input("请输入 key (YYYY/MM/DD): ").strip()
+        except EOFError:
+            print("未输入 key。", file=sys.stderr)
+            return 1
+
+    if not key_input:
+        print("key 不能为空。", file=sys.stderr)
+        return 1
+
+    try:
+        key_label, code = generate_code(key_input)
+    except ValueError as exc:
+        print(str(exc), file=sys.stderr)
+        return 1
+
+    print(f"key:    {key_label}")
+    print(f"激活码: {code}")
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())