bili.py

import requests
import json
import os
import sys
from os import path
from moviepy.editor import VideoFileClip
from io import BytesIO
import tempfile
import uuid
from urllib.parse import quote_plus

headers = {
    'User-Agent': 'PostmanRuntime/7.26.8',
    'Referer': 'https://www.bilibili.com/',
}

def fname_escape(name):
    return name.replace('\\', '＼') \
               .replace('/', '／') \
               .replace(':', '：') \
               .replace('*', '＊') \
               .replace('?', '？') \
               .replace('"', '＂') \
               .replace('<', '＜') \
               .replace('>', '＞') \
               .replace('|', '｜')
    

def batch_home(mid, st, ed, to_audio=False):
    for i in range(st, ed + 1):
        url = f'https://api.bilibili.com/x/space/arc/search?search_type=video&mid={mid}&pn={i}&order=pubdate'
        j = requests.get(url, headers=headers).json()
        if j['code'] != 0:
            print('解析失败：' + j['message'])
            return
        for it in j['data']['list']['vlist']:
            bv = it['bvid']
            download_safe(bv, to_audio)
            
    print('bvid')

def batch_kw(kw, st, ed, to_audio=False):
    kw_enco = quote_plus(kw)
    for i in range(st, ed + 1):
        url = f'https://api.bilibili.com/x/web-interface/search/type?search_type=video&keyword={kw_enco}&page={i}&order=pubdate'
        j = requests.get(url, headers=headers).json()
        if j['code'] != 0:
            print('解析失败：' + j['message'])
            return
        for it in j['data']['result']:
            bv = it['bvid']
            download_safe(bv, to_audio)
            
    print('bvid')

def download_safe(id, to_audio=False):
    try: download(id, to_audio)
    except Exception as ex: print(ex)

def download(id, to_audio=False):
    av = ''
    bv = ''
    if id.lower().startswith('av'):
        av = id[2:]
    else:
        bv = id
        
    url = f'https://api.bilibili.com/x/web-interface/view?bvid={bv}&aid={av}'
    j = requests.get(url, headers=headers).json()
    if j['code'] != 0:
        print('获取 CID 失败：' + j['message'])
        return
    av = j['data']['aid']
    bv = j['data']['bvid']
    cid = j['data']['cid']
    title = fname_escape(j['data']['title'])
    author = fname_escape(j['data']['owner']['name'])
    print(title, author)
    name = f'{title} - {author} - {bv}'
    fname = f'out/{name}'
    if to_audio:
        fname += '.mp3'
    else:
        fname += '.flv'
    if path.isfile(fname):
        print(f'{fname} 已存在')
        return
    try: os.mkdir('out')
    except: pass
    url = f'https://api.bilibili.com/x/player/playurl?cid={cid}&otype=json&bvid={bv}&aid={av}'
    j = requests.get(url, headers=headers).json()
    if j['code'] != 0:
        print('解析失败：' + j['message'])
        return
    video_url = j['data']['durl'][0]['url']
    video = requests.get(video_url, headers=headers).content
    if not to_audio:
        open(fname, 'wb').write(video)
        return
    tmp_fname = path.join(tempfile.gettempdir(), uuid.uuid4().hex + '.flv')
    open(tmp_fname, 'wb').write(video)
    vc = VideoFileClip(tmp_fname)
    vc.audio.write_audiofile(fname)
    vc.reader.close()
    os.unlink(tmp_fname)
    
def main():
    cmd = sys.argv[1]
    if cmd in ['dl', 'download']:
        download(sys.argv[2])
    elif cmd in ['dlau', 'dlaudio']:
        download(sys.argv[2], True)
    elif cmd == 'batchkw':
        batch_kw(
            sys.argv[2],
            int(sys.argv[3]) if len(sys.argv) > 3 else 1,
            int(sys.argv[4]) if len(sys.argv) > 4 else 50,
        )
    elif cmd == 'batchkwau':
        batch_kw(
            sys.argv[2],
            int(sys.argv[3]) if len(sys.argv) > 3 else 1,
            int(sys.argv[4]) if len(sys.argv) > 4 else 50,
            True,
        )
    elif cmd == 'batchhome':
        batch_home(
            sys.argv[2],
            int(sys.argv[3]) if len(sys.argv) > 3 else 1,
            int(sys.argv[4]) if len(sys.argv) > 4 else 50,
        )
    elif cmd == 'batchhomeau':
        batch_home(
            sys.argv[2],
            int(sys.argv[3]) if len(sys.argv) > 3 else 1,
            int(sys.argv[4]) if len(sys.argv) > 4 else 50,
            True,
        )
        
if __name__ == '__main__': main()