sennrisyunnga / video-editing-auxiliary-tool-based-on-barrage-analysis- Goto Github PK

import requests    #第三方库，需通过pip安装
def getDanmu(vid):
    if vid.startswith("BV"):    #BV号
        resp = requests.get(f"https://api.bilibili.com/x/player/pagelist?bvid={vid}")
    elif vid.startswith("av"):    #AV号
        resp = requests.get(f"https://api.bilibili.com/x/player/pagelist?aid={vid[2:]}")
    elif vid.isdigit():    #AV号（纯数字）
        resp = requests.get(f"https://api.bilibili.com/x/player/pagelist?aid={vid}")
    else:
        raise Exception("无效的av号或bv号")
    if resp.ok:
        data = resp.json()
        return requests.get(f"https://api.bilibili.com/x/v1/dm/list.so?oid={data['data'][0]['cid']}")
    else:
        raise Exception(resp.reason)

import xml.etree.ElementTree as ET
root = ET.fromstring(getDanmu("vid").content)
for danmu in root.findall('d'):
    print(danmu.attrib['p'].split(',')[0], danmu.text)    #时间，弹幕