python 下载m3u8视频「建议收藏」开发者社区

python 下载m3u8视频「建议收藏」

import os
import requests
headers = {
    "accept": "*/*",
    "accept-encoding": "gzip, deflate, br",
    "accept-language": "zh-CN,zh;q=0.9",
    "cache-control": "no-cache",
    "origin": "https://leetcode-cn.com",
    "pragma": "no-cache",
    "referer": "https://leetcode-cn.com/problems/course-schedule/solution/ke-cheng-biao-by-leetcode-solution/",
    "sec-fetch-dest": "empty",
    "sec-fetch-mode": "cors",
    "sec-fetch-site": "same-site",
    "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/84.0.4147.89 Safari/537.36",
def get_m3u8_file(m3u8_url, file_path):
    下载m3u8文件
    :param m3u8_url: m3u8文件的URL
    :param file_path: 要下载的文件位置
    :return:
    resp = requests.get(m3u8_url, headers=headers)
    if resp.status_code == 200:
        content = resp.text
        with open(file_path, "w") as f:
            f.write(content)

import os
def get_ts_name_list(file_path):
    获取ts文件名称列表
    :param file_path: m3u8文件路径
    :return: ts文件名称列表
    ts_name_list = []
    with open(file_path, "rb") as f:
        cont_list = f.readlines()
    for cont in cont_list:
        cont = cont.decode().strip()
        if cont.endswith(".ts"):
            ts_name_list.append(cont.split("/")[-1])
            # if not cont.startswith("#"):
            #     ts_name_list.append(cont.split("/")[-1])
    return ts_name_list

def get_ts_files(file_dir, ts_url_template, ts_name_list):
    循环下载ts文件
    :param file_dir: 文件下载所在文件夹
    :param ts_url_template: ts文件请求URL模板
    :param ts_name_list: ts文件名称列表
    :return: 
    for ts_name in ts_name_list:
        ts_url = ts_url_template + ts_name
        resp = requests.get(ts_url, headers=headers)
        if resp.status_code == 200:
            with open(os.path.join(file_dir, ts_name), "wb") as f:
                f.write(resp.content)
            print("%s-->下载成功！" % ts_name)
        else:
            print("%s-->下载失败！" % ts_name)

def merge_ts_files(file_dir, file_name, ts_name_list):
    将多个ts文件进行合并
    :param file_dir: ts文件所在文件夹
    :param file_name: 合并后的文件名称
    :param ts_name_list: ts文件名称列表
    :return: 
    file_out = os.path.join(file_dir, file_name)
    with open(file_out, 'wb') as f_out:
        for ts_name in ts_name_list:
            with open(os.path.join(file_dir, ts_name), "rb") as f_in:
                f_out.write(f_in.read())
    print("合并ts文件成功！")

import os
import requests
headers = {
    "accept": "*/*",
    "accept-encoding": "gzip, deflate, br",
    "accept-language": "zh-CN,zh;q=0.9",
    "cache-control": "no-cache",
    "origin": "https://leetcode-cn.com",
    "pragma": "no-cache",
    "referer": "https://leetcode-cn.com/problems/course-schedule/solution/ke-cheng-biao-by-leetcode-solution/",
    "sec-fetch-dest": "empty",
    "sec-fetch-mode": "cors",
    "sec-fetch-site": "same-site",
    "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/84.0.4147.89 Safari/537.36",
def get_m3u8_file(m3u8_url, file_path):
    下载m3u8文件
    :param m3u8_url: m3u8文件的URL
    :param file_path: 要下载的文件路径
    :return:
    resp = requests.get(m3u8_url, headers=headers)
    if resp.status_code == 200:
        content = resp.text
        with open(file_path, "w") as f:
            f.write(content)
def get_ts_name_list(file_path):
    获取ts文件名称列表
    :param file_path: m3u8文件路径
    :return: ts文件名称列表
    ts_name_list = []
    with open(file_path, "rb") as f:
        cont_list = f.readlines()
    for cont in cont_list:
        cont = cont.decode().strip()
        if cont.endswith(".ts"):
            ts_name_list.append(cont.split("/")[-1])
            # if not cont.startswith("#"):
            #     ts_name_list.append(cont.split("/")[-1])
    return ts_name_list
def get_ts_files(file_dir, ts_url_template, ts_name_list):
    循环下载ts文件
    :param file_dir: 文件下载所在文件夹
    :param ts_url_template: ts文件请求URL模板
    :param ts_name_list: ts文件名称列表
    :return:
    for ts_name in ts_name_list:
        ts_url = ts_url_template + ts_name
        resp = requests.get(ts_url, headers=headers)
        if resp.status_code == 200:
            with open(os.path.join(file_dir, ts_name), "wb") as f:
                f.write(resp.content)
            print("%s-->下载成功！" % ts_name)
        else:
            print("%s-->下载失败！" % ts_name)
def merge_ts_files(file_dir, file_name, ts_name_list):
    将多个ts文件进行合并
    :param file_dir: ts文件所在文件夹
    :param file_name: 合并后的文件名称
    :param ts_name_list: ts文件名称列表
    :return:
    file_out = os.path.join(file_dir, file_name)
    with open(file_out, 'wb') as f_out:
        for ts_name in ts_name_list:
            with open(os.path.join(file_dir, ts_name), "rb") as f_in:
                f_out.write(f_in.read())
    print("合并ts文件成功！")
if __name__ == '__main__':
    # 用来保存ts文件
    file_dir = './ts_file'
    if not os.path.exists(file_dir):
        os.mkdir(file_dir)
    # m3u8文件URL
    m3u8_url = "https://video.leetcode-cn.com/5d1c2e6731b448aaa9965202c997a26c/c5c5c3d84d7b4c70bdef21cd7ef011e6-1380cfc54b15bf9d5088f6deaff5a053-hd.m3u8"
    # 提取文件名
    file_name = m3u8_url.split('/')[-1]
    file_path = os.path.join(file_dir, file_name)
    # 下载m3u8文件
    get_m3u8_file(m3u8_url, file_path)
    # 获取文件中ts文件名称列表
    ts_name_list = get_ts_name_list(file_path)
    # ts文件URL模板