moviepy_with_manga_test / bilibili_api /utils /aid_bvid_transformer.py
rogerxavier's picture
Upload 258 files
0aee47a verified
"""
bilibili_api.utils.aid_bvid_transformer
av 号和 bv 号互转,代码来源:https://www.zhihu.com/question/381784377/answer/1099438784。
此部分代码以 WTFPL 开源。
"""
XOR_CODE = 23442827791579
MASK_CODE = 2251799813685247
MAX_AID = 1 << 51
data = [b'F', b'c', b'w', b'A', b'P', b'N', b'K', b'T', b'M', b'u', b'g', b'3', b'G', b'V', b'5', b'L', b'j', b'7', b'E', b'J', b'n', b'H', b'p', b'W', b's', b'x', b'4', b't', b'b', b'8', b'h', b'a', b'Y', b'e', b'v', b'i', b'q', b'B', b'z', b'6', b'r', b'k', b'C', b'y', b'1', b'2', b'm', b'U', b'S', b'D', b'Q', b'X', b'9', b'R', b'd', b'o', b'Z', b'f']
BASE = 58
BV_LEN = 12
PREFIX = "BV1"
def bvid2aid(bvid: str) -> int:
"""
BV 号转 AV 号。
Args:
bvid (str): BV 号。
Returns:
int: AV 号。
"""
bvid = list(bvid)
bvid[3], bvid[9] = bvid[9], bvid[3]
bvid[4], bvid[7] = bvid[7], bvid[4]
bvid = bvid[3:]
tmp = 0
for i in bvid:
idx = data.index(i.encode())
tmp = tmp * BASE + idx
return (tmp & MASK_CODE) ^ XOR_CODE
def aid2bvid(aid: int) -> str:
"""
AV 号转 BV 号。
Args:
aid (int): AV 号。
Returns:
str: BV 号。
"""
bytes = [b'B', b'V', b'1', b'0', b'0', b'0', b'0', b'0', b'0', b'0', b'0', b'0']
bv_idx = BV_LEN - 1
tmp = (MAX_AID | aid) ^ XOR_CODE
while int(tmp) != 0:
bytes[bv_idx] = data[int(tmp % BASE)]
tmp /= BASE
bv_idx -= 1
bytes[3], bytes[9] = bytes[9], bytes[3]
bytes[4], bytes[7] = bytes[7], bytes[4]
return "".join([i.decode() for i in bytes])