feat: handle 'av' format in BiliVideo initialization

Author: Ovler-Young

biliarchiver/rest_api/bilivid.py

@@ -2,6 +2,7 @@
 from enum import Enum
 import time
 from typing import Optional
+from biliarchiver.utils.avbv import av2bv, bv2av
 
 class VideoStatus(str, Enum):
     pending = "pending"
@@ -13,7 +14,11 @@ class VideoStatus(str, Enum):
 
 class BiliVideo:
     def __init__(self, bvid: str, status: VideoStatus):
-        if not bvid.startswith("BV"):
+        if bvid.startswith("av"):
+            bvid = av2bv(int(bvid[2:]))
+        elif bvid.isdecimal():
+            bvid = av2bv(int(bvid))
+        elif not bvid.startswith("BV"):
             bvid = "BV" + bvid
         self.added_time = int(time.time())
         self.bvid = bvid

biliarchiver/utils/avbv.py

@@ -0,0 +1,31 @@
+# https://github.com/SocialSisterYi/bilibili-API-collect/issues/847#issuecomment-1807020675
+# Original code by @12345-mcpython
+XOR_CODE = 23442827791579
+MASK_CODE = 2251799813685247
+MAX_AID = 1 << 51
+ALPHABET = "FcwAPNKTMug3GV5Lj7EJnHpWsx4tb8haYeviqBz6rkCy12mUSDQX9RdoZf"
+ENCODE_MAP = 8, 7, 0, 5, 1, 3, 2, 4, 6
+DECODE_MAP = tuple(reversed(ENCODE_MAP))
+
+BASE = len(ALPHABET)
+PREFIX = "BV1"
+PREFIX_LEN = len(PREFIX)
+CODE_LEN = len(ENCODE_MAP)
+
+def av2bv(aid: int) -> str:
+    bvid = [""] * 9
+    tmp = (MAX_AID | aid) ^ XOR_CODE
+    for i in range(CODE_LEN):
+        bvid[ENCODE_MAP[i]] = ALPHABET[tmp % BASE]
+        tmp //= BASE
+    return PREFIX + "".join(bvid)
+
+def bv2av(bvid: str) -> int:
+    assert bvid[:3] == PREFIX
+
+    bvid = bvid[3:]
+    tmp = 0
+    for i in range(CODE_LEN):
+        idx = ALPHABET.index(bvid[DECODE_MAP[i]])
+        tmp = tmp * BASE + idx
+    return (tmp & MASK_CODE) ^ XOR_CODE