def test_get_video_info():
bvid = "BV1vZ4y1M7mQ"
assert get_video_info(bvid=bvid)["bvid"] == bvid
# 匹配资源的 id 以及其对应所属类型
# fmt: off
resource_id = {
"avid": "",
"bvid": "",
"episode_id": "",
"season_id": "",
} >> AttrDict()
# fmt: off
if (avid_match := regex.acg_video.av.origin.match(args.url)) or \
(avid_match := regex.acg_video.av.short.match(args.url)):
from bilili.api.acg_video import get_video_info
avid = avid_match.group("avid")
if episode_id := get_video_info(avid=avid)["episode_id"]:
resource_id.episode_id = episode_id
else:
resource_id.avid = avid
elif (bvid_match := regex.acg_video.bv.origin.match(args.url)) or \
(bvid_match := regex.acg_video.bv.short.match(args.url)):
from bilili.api.acg_video import get_video_info
bvid = bvid_match.group("bvid")
if episode_id := get_video_info(bvid=bvid)["episode_id"]:
resource_id.episode_id = episode_id
else:
resource_id.bvid = bvid
elif media_id_match := regex.bangumi.md.origin.match(args.url):
from bilili.api.bangumi import get_season_id
media_id = media_id_match.group("media_id")
resource_id.season_id = get_season_id(media_id=media_id)