網上的TS流視頻文件下載，解密，合成一個文件的python方法（轉的別人大佬的，自己存一份~~）

# -*- coding:utf-8 -*- 
import os
import sys
import requests
import datetime
from Crypto.Cipher import AES
from binascii import b2a_hex, a2b_hex

# reload(sys)
# sys.setdefaultencoding('utf-8')

if 1:
    def debug(s):
        print(s)
else:
    def debug(s):
        pass

import importlib
importlib.reload(sys)

def download(url, savefile_path=''):
    # 創建download目錄
    download_path = os.getcwd() + "\download"
    if not os.path.exists(download_path):
        os.mkdir(download_path)

    #新建日期文件夾，如果沒有指定文件存儲位置，就用年月日創建文件夾，否則用指定的文件夾
    if not savefile_path:
        savefile_path = os.path.join(download_path, datetime.datetime.now().strftime('%Y%m%d'))
    else:
        savefile_path = os.path.join(download_path, savefile_path)
    print("savefile_path = " + savefile_path)
    if not os.path.exists(savefile_path):
        os.mkdir(savefile_path)
    

    #如果不存在first.m3u8，就重新獲取，並保存，否則證明之前已經獲取過，直接讀取就行了
    if not os.path.exists(os.path.join(download_path,'first.m3u8')):
        all_content = requests.get(url).text # 獲取第一層M3U8文件內容
        with open( os.path.join(download_path,'first.m3u8'),'w' ) as f:
            f.write(all_content)
        print("save first.m3u8")
    else:
        print("first.m3u8 is exist,just read it")
        all_content = open( os.path.join(download_path,'first.m3u8'),'r' ).read()
    
    # 不是M3U8文件，直接報錯退出
    if "#EXTM3U" not in all_content:
        raise BaseException("非M3U8的鏈接")
    # 
    if "EXT-X-STREAM-INF" in all_content: # 第一層
        # 從第一層中讀取所有的文件組成行列表 file_line
        print("we need to get the second line")
        file_line = all_content.split("\n")
        # 找到 .m3u8 ，拼出第二層的鏈接
        for line in file_line:
            if '.m3u8' in line:
                url = url.rsplit("/", 1)[0] + "/" + line # 拼出第二層m3u8的URL
                print("second line url = " + url)
                # 沒存，就存一下，同first.m3u8的邏輯
                if not os.path.exists(os.path.join(download_path,'second.m3u8')):
                    all_content = requests.get(url).text
                    with open( os.path.join(download_path,'second.m3u8'),'w' ) as f:
                        f.write(all_content)
                    print("second.m3u8 has been saved")
                else:
                    all_content = open( os.path.join(download_path,'second.m3u8'),'r' ).read()
                break
    # 到此為止，all_content里面的內容已經更新成了包含所有文件名的最終文件
    #           url 更新成了下載最終m3u8的URL
    
    # 把里面的元素分割出來
    file_line = all_content.split("\n")

    unknow = True
    key = ""
    
    for line in file_line:
        # 先找到解密key
        if "#EXT-X-KEY" in line: # 找解密Key
            method_pos = line.find("METHOD")
            comma_pos = line.find(",")
            method = line[method_pos:comma_pos].split('=')[1]
            print("Decode Method：", method)
         
            uri_pos = line.find("URI")
            quotation_mark_pos = line.rfind('"')
            key_path = line[uri_pos:quotation_mark_pos].split('"')[1]
         
            key_url = url.rsplit("/", 1)[0] + "/" + key_path # 拼出key解密密鑰URL
            res = requests.get(key_url)
            key = res.content
            
            with open( os.path.join(download_path,'key.key'),'wb' ) as f:
                f.write(key)

            cryptor = AES.new(key, AES.MODE_CBC, key) 
            print( "key：" , key)
            break
            print("you cann't see me")
    # 再進行下載和解密
    for index, line in enumerate(file_line):    # 第二層
        if "EXTINF" in line: # 找ts地址並下載
            unknow = False
            pd_url = url.rsplit("/", 1)[0] + "/" + file_line[index + 1] # 拼出ts片段的URL
         
            c_fule_name = file_line[index + 1].rsplit("/", 1)[-1]
            # 如果文件已經存在，就下一個
            if os.path.exists( os.path.join(savefile_path, c_fule_name) ):
                print("file %s exist, next" % c_fule_name)
                continue
            # 網絡不好的時候會失敗，所以重復3次
            for i in range(3):
                # 獲取視頻內容
                print("get video " + datetime.datetime.now().strftime('%H:%M:%S'))
                res = requests.get(pd_url)
                try:
                    if len(key): # AES 解密，有key就是需要解密
                        with open(os.path.join(savefile_path, c_fule_name), 'ab') as f:
                            f.write(cryptor.decrypt(res.content))
                        print(c_fule_name + " success")
                        break
                    else:  # 沒有key就不需要解密了，直接保存文件就行了
                        with open(os.path.join(savefile_path, c_fule_name), 'ab') as f:
                            f.write(res.content)
                            f.flush()
                        print(c_fule_name + " success")
                        break
                except:  # 網絡不好導致下載失敗，清除文件
                    print( str(i+1) + " download error, file: " + c_fule_name)
                    os.remove(os.path.join(savefile_path, c_fule_name))
            else:  # 下載失敗，先跳過這個文件，后面重新執行的時候在重新下載
                print("download file has failed 3 times, jump it")
                #exit()

    if unknow:
        raise BaseException("未找到對應的下載鏈接")
    else:
        print( "下載完成")
        # merge_file(savefile_path)

def merge_file(path):
    os.chdir(path)
    cmd = "copy /b * new.ts"
    #os.system(cmd)
    #os.system('del /Q *.ts')
    #os.system('del /Q *.mp4')
    #os.rename("new.tmp", "new.mp4")
 
if __name__ == '__main__':
    url = input("please input the url of index.m3u8 file:\n")
    savefile_path = input("give me a dir name to save all the video:\n(or you can just press Enter and I will create one for you)\n")
    # url = r''
    # savefile_path = r''
    download(url,savefile_path)