v0 plus附加功能
- 新增 unlockmusic.py：解锁（解禁）歌曲 + 解禁 Master / Re:Master 难度，整个流程复用 upsert.py 的 run() - 买票前置闸门 api/ChargeGuard.py：必须先跑过 GetUserCharge，且同进程只允许买一次 - buy_ticket.py 票源改为 GetUserCharge，不再调用 GetGameCharge - UpsertUserAll 与买票的冷却统一为 15 秒 - api/ 文件名统一 PascalCase（文件名 = 函数名） - 项目改名 Astral，README 加 Logo、鸣谢与开发者说明 - 自检 284 项
This commit is contained in:
+311
@@ -0,0 +1,311 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
Chime.py —— 扫码登录（二维码 -> userId + token）
|
||||
================================================================================
|
||||
|
||||
功能：把微信二维码串（或二维码 URL）交给 aimedb 服务器，换取 userId 与 token。
|
||||
|
||||
对应接口：POST https://ai.sys-allnet.cn/wc_aime/api/get_data
|
||||
|
||||
--------------------------------------------------------------------------------
|
||||
协议依据
|
||||
--------------------------------------------------------------------------------
|
||||
source/ChimeLib.NET/ChimeLib/NET/COMM/CCommGetUserData.cs
|
||||
CCommGetUserData_Create(strGameID, strChipID, strCommonKey,
|
||||
strQRData, strTitleKey, serverURLIndex)
|
||||
strGameID = "MAID"
|
||||
strChipID = KeychipID
|
||||
strQRData = 二维码串[20:] <- 见 CCommQRCode.EraseHashIdentifier()
|
||||
strTitleKey= "SDGB"
|
||||
|
||||
source/ChimeLib.NET/ChimeLib/NET/COMM/CCommQRCode.cs
|
||||
IsValid() : 前缀 "SGWC"(Chime.WechatID) + 偏移 4 处为 gameID
|
||||
GetTimeStamp() : [8:20] 12 位 yymmddHHMMss
|
||||
EraseHashIdentifier() : [20:] 64 位载荷
|
||||
|
||||
source/Assembly-CSharp/Manager/ChimeReaderManager.cs:250-255
|
||||
KeychipID = AMDaemon.System.KeychipId.Value
|
||||
TitleKey = AMDaemon.System.GameId
|
||||
GameID = "MAID"
|
||||
|
||||
docs/舞萌DX_扫码登录协议技术规范.md
|
||||
- 签名 key = SHA-256(chipID + timestamp + SIGN_KEY) 大写 hex
|
||||
- ★ 请求头【绝不能】带 Content-Type，否则服务端返回 errorID=57
|
||||
- qrCode 字段传的是 64 位载荷，不是完整 84 字符串
|
||||
|
||||
--------------------------------------------------------------------------------
|
||||
调用示例
|
||||
--------------------------------------------------------------------------------
|
||||
from api import ParseQRCode
|
||||
|
||||
user_id, token = ParseQRCode("SGWCMAID2610022219463A...")
|
||||
# 也可以直接传二维码 URL
|
||||
user_id, token = ParseQRCode("https://wq.wahlap.net/qrcode/req/MAID261002....html?l=...")
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import hashlib
|
||||
import http.client
|
||||
import json
|
||||
import re
|
||||
import ssl
|
||||
from datetime import datetime
|
||||
from typing import Tuple
|
||||
from urllib.parse import urlparse
|
||||
|
||||
from config import KEYCHIP_ID
|
||||
|
||||
# =============================================================================
|
||||
# aimedb 扫码登录常量（原在 config.py，2026-10-06 下沉到本模块）
|
||||
# =============================================================================
|
||||
# 依据：source/ChimeLib.NET/ChimeLib/NET/COMM/CCommGetUserData.cs
|
||||
# + docs 分支《舞萌DX_扫码登录协议技术规范》
|
||||
|
||||
#: aimedb 服务器
|
||||
AIMEDB_HOST = "ai.sys-allnet.cn"
|
||||
AIMEDB_PORT = 443
|
||||
AIMEDB_PATH = "/wc_aime/api/get_data"
|
||||
|
||||
#: 签名密钥（含前导 X，32 字符）
|
||||
#: 来源：chimelib_dll.dll .rdata UTF-16 @0x00B040，抓包验证一致
|
||||
QR_SIGN_KEY = "XcW5FW4cPArBXEk4vzKz3CIrMuA5EVVW"
|
||||
|
||||
#: 固定字段
|
||||
QR_OPEN_GAME_ID = "MAID"
|
||||
QR_TITLE_KEY = "SDGB"
|
||||
QR_USER_AGENT = "WC_AIME_LIB"
|
||||
|
||||
from .LogSetup import log_debug, log_error, log_info, log_request, log_response
|
||||
|
||||
# =============================================================================
|
||||
# 二维码格式常量（CCommQRCode.cs + 技术规范）
|
||||
# =============================================================================
|
||||
|
||||
QR_PREFIX = "SGWC" # Chime.WechatID
|
||||
QR_GAME_ID = "MAID"
|
||||
QR_PREFIX_LEN = 20 # SGWC(4) + MAID(4) + timestamp(12)
|
||||
QR_PAYLOAD_LEN = 64
|
||||
QR_TOTAL_LEN = QR_PREFIX_LEN + QR_PAYLOAD_LEN # 84
|
||||
|
||||
#: 默认机台 chipID
|
||||
#:
|
||||
#: ★★ 必须用**完整形式**（带连字符），不是短值 —— 这是个容易搞错的点：
|
||||
#: source/Assembly-CSharp/Manager/ChimeReaderManager.cs:250
|
||||
#: KeychipID = AMDaemon.System.KeychipId.Value; // 完整形式
|
||||
#: source/Assembly-CSharp/Manager/ChimeReaderManager.cs:255
|
||||
#: Chime.Instance.Init(GameID, KeychipID, _chimeDevice, TitleKey);
|
||||
#: 而游戏服务器那边用的是短值：
|
||||
#: source/Assembly-CSharp/Net.Packet.Mai2/PacketGetGameSetting.cs:23
|
||||
#: clientId = AMDaemon.System.KeychipId.ShortValue; // 短值
|
||||
#:
|
||||
#: 即：aimedb 扫码用完整 KeychipId，游戏服务器用 ShortValue。
|
||||
#: docs 分支的《扫码登录协议技术规范》§3.4 里也有对应的抓包验证向量。
|
||||
#: token 与 chipID 绑定，用错会导致后续接口「returnCode=1 但不落库」。
|
||||
DEFAULT_CHIP_ID = KEYCHIP_ID
|
||||
|
||||
|
||||
# =============================================================================
|
||||
# 内部工具
|
||||
# =============================================================================
|
||||
|
||||
def _extract_qr_from_url(url: str) -> str:
|
||||
"""
|
||||
从二维码 URL 提取完整 84 字符串。
|
||||
|
||||
URL 形如：
|
||||
https://wq.wahlap.net/qrcode/req/MAID2610022219463A....html?l=...&t=...&d=...
|
||||
取 /qrcode/req/ 与 .html 之间的 80 字符，前面补 "SGWC"。
|
||||
"""
|
||||
path = urlparse(url).path
|
||||
match = re.search(r"/qrcode/req/([^/]+)\.html", path)
|
||||
if not match:
|
||||
raise ValueError("URL 路径格式不符合预期，无法提取二维码数据：" + url)
|
||||
return QR_PREFIX + match.group(1)
|
||||
|
||||
|
||||
def _normalize_qr(qr_input: str) -> str:
|
||||
"""URL 或字符串 -> 标准 84 字符二维码串（含格式校验）。"""
|
||||
qr = (qr_input or "").strip()
|
||||
|
||||
if qr.startswith("http://") or qr.startswith("https://"):
|
||||
log_debug("输入是 URL，先提取二维码字符串")
|
||||
qr = _extract_qr_from_url(qr)
|
||||
|
||||
if len(qr) != QR_TOTAL_LEN:
|
||||
raise ValueError(
|
||||
"二维码长度应为 " + str(QR_TOTAL_LEN) + " 字符，实际 " + str(len(qr)) + " 字符：" + qr
|
||||
)
|
||||
if not qr.startswith(QR_PREFIX):
|
||||
raise ValueError("二维码必须以 '" + QR_PREFIX + "' 开头：" + qr)
|
||||
if qr[4:8] != QR_GAME_ID:
|
||||
raise ValueError(
|
||||
"二维码 GameID 应为 '" + QR_GAME_ID + "'，实际 '" + qr[4:8] + "'：" + qr
|
||||
)
|
||||
if not all(c in "0123456789ABCDEF" for c in qr[QR_PREFIX_LEN:]):
|
||||
raise ValueError("二维码载荷必须为大写十六进制：" + qr)
|
||||
|
||||
log_debug(
|
||||
"二维码校验通过：timestamp=" + qr[8:20]
|
||||
+ " payload_len=" + str(len(qr[QR_PREFIX_LEN:]))
|
||||
)
|
||||
return qr
|
||||
|
||||
|
||||
def _build_body(qr_string: str, chip_id: str) -> Tuple[bytes, str, str]:
|
||||
"""
|
||||
构造请求体。
|
||||
|
||||
签名（技术规范 §4.1）：
|
||||
key = SHA-256(chipID + timestamp + SIGN_KEY).hexdigest().upper()
|
||||
★ 拼接里不含 qrCode、不含 titlekey
|
||||
|
||||
qrCode 字段（CCommQRCode.EraseHashIdentifier）：
|
||||
qr_string[20:] 即 64 位载荷
|
||||
|
||||
返回：
|
||||
(body_bytes, timestamp, key)
|
||||
"""
|
||||
timestamp = datetime.now().strftime("%y%m%d%H%M%S")
|
||||
key = hashlib.sha256((chip_id + timestamp + QR_SIGN_KEY).encode()).hexdigest().upper()
|
||||
|
||||
body_dict = {
|
||||
"chipID": chip_id,
|
||||
"openGameID": QR_OPEN_GAME_ID,
|
||||
"key": key,
|
||||
"qrCode": qr_string[QR_PREFIX_LEN:],
|
||||
"timestamp": timestamp,
|
||||
"titlekey": QR_TITLE_KEY,
|
||||
}
|
||||
body_bytes = json.dumps(body_dict, separators=(",", ":")).encode("utf-8")
|
||||
return body_bytes, timestamp, key
|
||||
|
||||
|
||||
def _send(body_bytes: bytes) -> dict:
|
||||
"""
|
||||
发送 HTTPS POST 并解析 JSON 响应。
|
||||
|
||||
★ 请求头只放 Connection / User-Agent / Content-Length。
|
||||
绝不能加 Content-Type —— 加了会得到 errorID=57（技术规范 §3.3）。
|
||||
"""
|
||||
ctx = ssl.create_default_context()
|
||||
ctx.check_hostname = False
|
||||
ctx.verify_mode = ssl.CERT_NONE
|
||||
|
||||
headers = {
|
||||
"Connection": "Keep-Alive",
|
||||
"User-Agent": QR_USER_AGENT,
|
||||
"Content-Length": str(len(body_bytes)),
|
||||
# ★ 此处刻意不加 Content-Type
|
||||
}
|
||||
|
||||
conn = http.client.HTTPSConnection(AIMEDB_HOST, AIMEDB_PORT, context=ctx, timeout=20)
|
||||
try:
|
||||
conn.request("POST", AIMEDB_PATH, body=body_bytes, headers=headers)
|
||||
resp = conn.getresponse()
|
||||
text = resp.read().decode("utf-8")
|
||||
status = resp.status
|
||||
finally:
|
||||
conn.close()
|
||||
|
||||
log_response(status=status, body=text, api_name="aimedb/get_data")
|
||||
try:
|
||||
return json.loads(text)
|
||||
except Exception as exc:
|
||||
raise RuntimeError("aimedb 响应不是合法 JSON：" + text[:300]) from exc
|
||||
|
||||
|
||||
def _parse(resp: dict) -> Tuple[int, str]:
|
||||
"""解析响应，返回 (user_id, token)。"""
|
||||
error_id = resp.get("errorID", -1)
|
||||
user_id = resp.get("userID", -1)
|
||||
token = resp.get("token", "") or ""
|
||||
|
||||
if error_id != 0:
|
||||
raise RuntimeError(
|
||||
"扫码登录失败：errorID=" + str(error_id) + "，响应=" + json.dumps(resp, ensure_ascii=False)
|
||||
+ "\n 提示：errorID=57 通常是二维码已失效/已被使用，或请求头带了 Content-Type"
|
||||
)
|
||||
if user_id == -1 or not token:
|
||||
raise RuntimeError(
|
||||
"扫码登录失败：userID=" + str(user_id) + ", token='" + token + "'"
|
||||
)
|
||||
return int(user_id), str(token)
|
||||
|
||||
|
||||
# =============================================================================
|
||||
# 公开 API
|
||||
# =============================================================================
|
||||
|
||||
def ParseQRCode(qr_input: str, *, chip_id: str = DEFAULT_CHIP_ID) -> Tuple[int, str]:
|
||||
"""
|
||||
解析二维码并换取 userId 与 token。
|
||||
|
||||
参数：
|
||||
qr_input: 84 字符二维码串（SGWCMAID...），或二维码 URL
|
||||
chip_id: 机台 Keychip 短值（可选，默认取 config.CLIENT_ID）
|
||||
|
||||
返回：
|
||||
(user_id, token) 元组，可直接传给 GetUserPreview / UserLogin 等
|
||||
|
||||
异常：
|
||||
ValueError: 二维码格式不合法
|
||||
RuntimeError: 服务器返回错误或网络异常
|
||||
"""
|
||||
log_info("解析二维码登录：chip_id=" + chip_id)
|
||||
|
||||
qr_string = _normalize_qr(qr_input)
|
||||
body_bytes, timestamp, key = _build_body(qr_string, chip_id)
|
||||
|
||||
log_request(
|
||||
url="https://" + AIMEDB_HOST + ":" + str(AIMEDB_PORT) + AIMEDB_PATH,
|
||||
body={
|
||||
"chipID": chip_id,
|
||||
"openGameID": QR_OPEN_GAME_ID,
|
||||
"key": key,
|
||||
"qrCode": qr_string[QR_PREFIX_LEN:],
|
||||
"timestamp": timestamp,
|
||||
"titlekey": QR_TITLE_KEY,
|
||||
},
|
||||
headers={"Connection": "Keep-Alive", "User-Agent": QR_USER_AGENT,
|
||||
"Content-Length": str(len(body_bytes))},
|
||||
api_name="aimedb/get_data",
|
||||
)
|
||||
|
||||
log_info("发送登录请求：timestamp=" + timestamp + " host=" + AIMEDB_HOST + AIMEDB_PATH)
|
||||
resp = _send(body_bytes)
|
||||
user_id, token = _parse(resp)
|
||||
|
||||
log_info("扫码登录成功：user_id=" + str(user_id) + " token=" + token[:8] + "...（长度 "
|
||||
+ str(len(token)) + "）")
|
||||
return user_id, token
|
||||
|
||||
|
||||
# =============================================================================
|
||||
# 单独运行（调试）
|
||||
# =============================================================================
|
||||
|
||||
if __name__ == "__main__":
|
||||
import sys
|
||||
|
||||
from .LogSetup import setup_logging, teardown_logging
|
||||
|
||||
def _main() -> int:
|
||||
setup_logging("chime_test")
|
||||
try:
|
||||
if len(sys.argv) < 2:
|
||||
print("用法: python -m api.chime <二维码串或URL> [chip_id]")
|
||||
return 1
|
||||
chip = sys.argv[2] if len(sys.argv) > 2 else DEFAULT_CHIP_ID
|
||||
user_id, token = ParseQRCode(sys.argv[1], chip_id=chip)
|
||||
print("")
|
||||
print("user_id =", user_id)
|
||||
print("token =", token)
|
||||
return 0
|
||||
except Exception as exc:
|
||||
log_error("扫码登录失败：" + str(exc))
|
||||
return 1
|
||||
finally:
|
||||
teardown_logging()
|
||||
|
||||
raise SystemExit(_main())
|
||||
Reference in New Issue
Block a user