带参数的py文件执行方式，减少多次交互。方便做成脚本运行

bilibili_video_download_v3-linux-oneline.py

@@ -0,0 +1,275 @@
+# !/usr/bin/python
+# -*- coding:utf-8 -*-
+# time: 2019/07/02--08:12
+__author__ = 'Henry'
+
+
+'''
+项目: B站视频下载 - 多线程下载
+
+版本1: 加密API版,不需要加入cookie,直接即可下载1080p视频
+
+20190422 - 增加多P视频单独下载其中一集的功能
+20190702 - 增加视频多线程下载 速度大幅提升
+'''
+
+import requests, time, hashlib, urllib.request, re, json
+from moviepy.editor import *
+import os, sys, threading
+import signal
+import imageio
+import argparse
+imageio.plugins.ffmpeg.download()
+
+
+parser = argparse.ArgumentParser()
+parser.add_argument('url', default='60025097' ,type=str, help='B站url或者av号')
+parser.add_argument('--quality', default=64 ,type=int, help='videoQuality(1080p:80;720p:64;480p:32;360p:16)(input:80,64,32or16)')
+parser.add_argument('--localpath', default='', type=str, help='local path,eg: /test ') 
+args = parser.parse_args(sys.argv[1:])
+print('下载地址：' + args.url)
+print('清晰度：'+ str(args.quality))
+print('本地路径'+os.path.join(sys.path[0], 'bilibili_video'+args.localpath))
+
+# 线程信号量, 限制并发数
+S = threading.Semaphore(5)
+
+# 正在下载的视频
+currentPage = []
+
+# 清屏函数
+def Clear():
+    Hide()
+    print('\033[2J', end='')
+
+# 显示光标
+def Show():
+    print('\033[?25h', end='')
+
+# 隐藏光标
+def Hide():
+    print('\033[?25l', end='')
+
+# 移动到位置,且清除这一行
+def POS(x=0,y=0):
+    print('\033[{};{}H\033[K'.format(y,x), end='')
+
+def signal_handler(signal,frame):
+    Show()
+    sys.exit(0)
+
+signal.signal(signal.SIGINT,signal_handler)
+
+# 访问API地址
+def get_play_list(start_url, cid, quality):
+    entropy = 'rbMCKn@KuamXWlPMoJGsKcbiJKUfkPF_8dABscJntvqhRSETg'
+    appkey, sec = ''.join([chr(ord(i) + 2) for i in entropy[::-1]]).split(':')
+    params = 'appkey=%s&cid=%s&otype=json&qn=%s&quality=%s&type=' % (appkey, cid, quality, quality)
+    chksum = hashlib.md5(bytes(params + sec, 'utf8')).hexdigest()
+    url_api = 'https://interface.bilibili.com/v2/playurl?%s&sign=%s' % (params, chksum)
+    headers = {
+        'Referer': start_url,  # 注意加上referer
+        'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/55.0.2883.87 Safari/537.36'
+    }
+    # print(url_api)
+    html = requests.get(url_api, headers=headers).json()
+    # print(json.dumps(html))
+    video_list = []
+    for i in html['durl']:
+        video_list.append(i['url'])
+    # print(video_list)
+    return video_list
+
+
+# 下载视频
+'''
+ urllib.urlretrieve 的回调函数：
+def callbackfunc(blocknum, blocksize, totalsize):
+    @blocknum:  已经下载的数据块
+    @blocksize: 数据块的大小
+    @totalsize: 远程文件的大小
+'''
+
+
+def Schedule_cmd(title, page):
+    start_time = time.time()
+    def Schedule(blocknum, blocksize, totalsize):
+        # 进度条打印在第几行
+        lineNum = currentPage.index(page)+1
+        POS(0, lineNum)
+        speed = (blocknum * blocksize) / (time.time() - start_time)
+        # speed_str = " Speed: %.2f" % speed
+        speed_str = " Speed: %s" % format_size(speed)
+        recv_size = blocknum * blocksize
+
+        # 设置下载进度条
+        percent = recv_size / totalsize
+        percent_str = "%.2f%%" % (percent * 100)
+        n = round(percent * 50)
+        s = ('#' * n).ljust(50, '-')
+        print('P{}:'.format(page) + '[' + s + ']  ' + percent_str.ljust(6, ' ') + speed_str)
+    return Schedule
+
+
+# 字节bytes转化K\M\G
+def format_size(bytes):
+    try:
+        bytes = float(bytes)
+        kb = bytes / 1024
+    except:
+        print("传入的字节格式不对")
+        return "Error"
+    if kb >= 1024:
+        M = kb / 1024
+        if M >= 1024:
+            G = M / 1024
+            return "%.3fG" % (G)
+        else:
+            return "%.3fM" % (M)
+    else:
+        return "%.3fK" % (kb)
+
+
+#  下载视频
+def down_video(video_list, title, start_url, page):
+    S.acquire()
+    num = 1
+    currentVideoPath = os.path.join(sys.path[0], 'bilibili_video'+args.localpath, title)  # 当前目录作为下载目录
+    if not os.path.exists(currentVideoPath):
+        os.makedirs(currentVideoPath)
+    for i in video_list:
+        opener = urllib.request.build_opener()
+        # 请求头
+        opener.addheaders = [
+            # ('Host', 'upos-hz-mirrorks3.acgvideo.com'),  #注意修改host,不用也行
+            ('User-Agent', 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10.13; rv:56.0) Gecko/20100101 Firefox/56.0'),
+            ('Accept', '*/*'),
+            ('Accept-Language', 'en-US,en;q=0.5'),
+            ('Accept-Encoding', 'gzip, deflate, br'),
+            ('Range', 'bytes=0-'),  # Range 的值要为 bytes=0- 才能下载完整视频
+            ('Referer', start_url),  # 注意修改referer,必须要加的!
+            ('Origin', 'https://www.bilibili.com'),
+            ('Connection', 'keep-alive'),
+        ]
+        urllib.request.install_opener(opener)
+        # 创建文件夹存放下载的视频
+        if not os.path.exists(currentVideoPath):
+            os.makedirs(currentVideoPath)
+        # 开始下载
+        reporthook = Schedule_cmd(title, page)
+        currentPage.append(page)
+        if len(video_list) > 1:
+            urllib.request.urlretrieve(url=i, filename=os.path.join(currentVideoPath, r'{}-{}.flv'.format(title, num)),reporthook=reporthook)  # 写成mp4也行  title + '-' + num + '.flv'
+        else:
+            urllib.request.urlretrieve(url=i, filename=os.path.join(currentVideoPath, r'{}.flv'.format(title)),reporthook=reporthook)  # 写成mp4也行  title + '-' + num + '.flv'
+        currentPage.remove(page)
+        num += 1
+    S.release()
+
+# 合并视频(20190802新版)
+def combine_video(title_list):
+    video_path = os.path.join(sys.path[0], 'bilibili_video'+args.localpath)  # 下载目录
+    for title in title_list:
+        current_video_path = os.path.join(video_path ,title)
+        if len(os.listdir(current_video_path)) >= 2:
+            # 视频大于一段才要合并
+            print('[下载完成,正在合并视频...]:' + title)
+            # 定义一个数组
+            L = []
+            # 遍历所有文件
+            for file in sorted(os.listdir(current_video_path), key=lambda x: int(x[x.rindex("-") + 1:x.rindex(".")])):
+                # 如果后缀名为 .mp4/.flv
+                if os.path.splitext(file)[1] == '.flv':
+                    # 拼接成完整路径
+                    filePath = os.path.join(current_video_path, file)
+                    # 载入视频
+                    video = VideoFileClip(filePath)
+                    # 添加到数组
+                    L.append(video)
+            # 拼接视频
+            final_clip = concatenate_videoclips(L)
+            # 生成目标视频文件
+            final_clip.to_videofile(os.path.join(current_video_path, r'{}.mp4'.format(title)), fps=24, remove_temp=False)
+            print('[视频合并完成]' + title)
+        else:
+            # 视频只有一段则直接打印下载完成
+            print('[视频合并完成]:' + title)
+
+
+if __name__ == '__main__':
+    start_time = time.time()
+    # 用户输入av号或者视频链接地址
+    print('*' * 30 + 'B站视频下载小助手' + '*' * 30)
+    #start = input('请输入您要下载的B站av号或者视频链接地址:')
+    start =args.url
+    if start.isdigit() == True:  # 如果输入的是av号
+        # 获取cid的api, 传入aid即可
+        start_url = 'https://api.bilibili.com/x/web-interface/view?aid=' + start
+    else:
+        # https://www.bilibili.com/video/av46958874/?spm_id_from=333.334.b_63686965665f7265636f6d6d656e64.16
+        start_url = 'https://api.bilibili.com/x/web-interface/view?aid=' + re.search(r'/av(\d+)/*', start).group(1)
+
+    # 视频质量
+    # <accept_format><![CDATA[flv,flv720,flv480,flv360]]></accept_format>
+    # <accept_description><![CDATA[高清 1080P,高清 720P,清晰 480P,流畅 360P]]></accept_description>
+    # <accept_quality><![CDATA[80,64,32,16]]></accept_quality>
+    #quality = input('请输入您要下载视频的清晰度(1080p:80;720p:64;480p:32;360p:16)(填写80或64或32或16):')
+    quality=args.quality
+    # 获取视频的cid,title
+    headers = {
+        'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/55.0.2883.87 Safari/537.36'
+    }
+    html = requests.get(start_url, headers=headers).json()
+    data = html['data']
+    cid_list = []
+    if '?p=' in start:
+        # 单独下载分P视频中的一集
+        p = re.search(r'\?p=(\d+)',start).group(1)
+        cid_list.append(data['pages'][int(p) - 1])
+    else:
+        # 如果p不存在就是全集下载
+        cid_list = data['pages']
+    # print(cid_list)
+    # 创建线程池
+    threadpool = []
+    title_list = []
+    Hide()
+    for i, item in enumerate(cid_list):
+        cid = str(item['cid'])
+        title = item['part']
+        title = re.sub(r'[\/\\:*?"<>|]', '', title)  # 替换为空的
+        # s是进度条
+        s = ('#' * round(i/len(cid_list)*50)).ljust(50, '-')
+        print('加载视频cid:[{}] {}/{}\r'.format(s,i,len(cid_list)), end='')
+        title_list.append(title)
+        page = str(item['page'])
+        start_url = start_url + "/?p=" + page
+        video_list = get_play_list(start_url, cid, quality)
+        # down_video(video_list, title, start_url, page)
+        # 定义线程
+        th = threading.Thread(target=down_video, args=(video_list, title, start_url, page))
+        # 将线程加入线程池
+        threadpool.append(th)
+
+    Clear()
+    # 开始线程
+    for th in threadpool:
+        th.start()
+    # 等待所有线程运行完毕
+    for th in threadpool:
+        th.join()
+    Show()
+    # 最后合并视频
+    print(title_list)
+    combine_video(title_list)
+
+    end_time = time.time()  # 结束时间
+    print('下载总耗时%.2f秒,约%.2f分钟' % (end_time - start_time, int(end_time - start_time) / 60))
+    # 如果是windows系统，下载完成后打开下载目录
+    currentVideoPath = os.path.join(sys.path[0], 'bilibili_video',+args.localpath)  # 当前目录作为下载目录
+    if (sys.platform.startswith('win')):
+        os.startfile(currentVideoPath)
+
+
+# 分P视频下载测试: https://www.bilibili.com/video/av19516333/
+# 下载总耗时14.21秒,约0.23分钟