Spaces:
Sleeping
Sleeping
File size: 1,662 Bytes
0aee47a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
"""
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])
|