bv2av.py

import requests
import sys
import locale

headers = {
    'Connection': 'keep-alive',
    'Cache-Control': 'max-age=0',
    'DNT': '1',
    'Upgrade-Insecure-Requests': '1',
    'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_3) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/82.0.4083.0 Safari/537.36 Edg/82.0.458.0',
    'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.9',
    'Sec-Fetch-Site': 'none',
    'Sec-Fetch-Mode': 'navigate',
    'Sec-Fetch-User': '?1',
    'Sec-Fetch-Dest': 'document',
    'Accept-Language': 'en-US,en;q=0.9'
}


chs = [
    '请输入BV号：',
    '输入错误！BV号应由\'BV\'开头！',
    '查询出错了！\n可能的原因：\n\t该视频可能并不存在AV号',
    '查询出错了！\n可能的原因：\n\t1、你调用的次数太多暂时冻结了，要过一会儿才可以继续查询！\n\t2、你的网络可能出现了一些异常',
    '抱歉，请重试！',
    '对应的AV号为：AV'
]

eng = [
    'Please input BV number:',
    'Input Error! BV number should start with \'BV\'!',
    'Query Error!\nPossible Reason:\n\tThis video might not have an AV number',
    'Query Error!\nPossible Reasons:\n\t1. You use this too much in short period, please query later!' +
    '\n\t2.You might encountered with some internet issues',
    'Sorry, please try again!',
    'The AV number for the video you input is: AV'
]


# Function when it is short link
def get_url(link):
    response = requests.get(link, headers=headers)
    return response.url


language = locale.getdefaultlocale()[0]

try:
    temp = sys.argv[1]
    int(sys.argv[2])
except IndexError:
    if language == 'zh_CN':
        print('你当前的系统语言是简体中文，需要更改程序显示语言吗？')
        print('Your current language display is Chinese Simplified, do you want to switch language?')
        lang = chs
    else:
        print('你当前的系统语言不是简体中文，需要更改程序显示语言吗？')
        print('Your current language display is English, do you want to switch language?')
        lang = eng
    while True:
        switch_confirmation = input('(Y/N)')
        if switch_confirmation.lower() == 'y' or switch_confirmation.lower() == 'yes' \
                or switch_confirmation.lower() == 't' or switch_confirmation.lower() == 'true'\
                or switch_confirmation == '是' or switch_confirmation == '确定':
            switch = True
            break
        elif switch_confirmation.lower() == 'n' or switch_confirmation.lower() == 'no' \
                or switch_confirmation.lower() == 'f' or switch_confirmation.lower() == 'false'\
                or switch_confirmation == '否':
            switch = False
            break
        else:
            print('输入错误！')
            print('Unknown Selection, please try again!')

    if switch:
        while True:
            print('1. 简体中文\n2. English')
            language_selection = input('请选择语言(Select language):')
            if language_selection == '1' or language_selection.lower() == 'chs' or language_selection == '简体中文' \
                    or language_selection == '中文' or language_selection.lower() == 'cn' \
                    or language_selection.lower() == 'china' or language_selection.lower() == 'chinese':
                lang = chs
                break
            elif language_selection == '2' or language_selection.lower() == 'eng' \
                    or language_selection.lower() == 'english':
                lang = eng
                break
            else:
                print('输入错误，请重试\n(Unknown Selection, please try again)')
else:
    if language == 'zh_CN':
        lang = chs
    else:
        lang = eng

try:
    temp = sys.argv[1]
    int(sys.argv[2])
except IndexError:
    BV_Number = input(lang[0])
    while True:
        if BV_Number.lower().find('bilibili.com') > 0 \
                or BV_Number.lower().find('b23.tv') > 0:
            if get_url(BV_Number).upper().find('BV') < len(get_url(BV_Number)) - 2:
                BV_Number = get_url(BV_Number)
                break
            else:
                print(lang[1])
                BV_Number = input(lang[0])
        else:
            if BV_Number.upper().find('BV') == 0:
                break
            else:
                print(lang[1])
                BV_Number = input(lang[0])
else:
    BV_Number = sys.argv[1]
    while True:
        if BV_Number.lower().find('bilibili.com') > 0 \
                or BV_Number.lower().find('b23.tv') > 0:

            if get_url(BV_Number).upper().find('BV') < len(get_url(BV_Number)) - 2:
                BV_Number = get_url(BV_Number)
                break
            else:
                print(lang[1])
                BV_Number = input(lang[0])
        else:
            if BV_Number.upper().find('BV') == 0:
                break
            else:
                print(lang[1])
                BV_Number = input(lang[0])

if BV_Number.find('?') != -1:
    BV_Number = "BV" + BV_Number[BV_Number.upper().find('BV')+2:BV_Number.find('?')]
else:
    BV_Number = "BV" + BV_Number[BV_Number.upper().find('BV')+2:]

url = 'https://api.bilibili.com/x/web-interface/archive/stat?bvid={}'

r = requests.get(url.format(BV_Number), headers=headers)
if r.status_code == 200:
    try:
        j = r.json()['data']
        AV_Number = j['aid']
    except Exception:
        print(lang[2])
else:
    print(lang[3])
try:
    AV_Number
except Exception:
    print(lang[4])
else:
    print(lang[5] + str(AV_Number))