• 爬虫
  • 视频处理
  • 文件处理
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
# -*- coding: utf-8 -*-

import os
import requests
# import subprocess
from time import sleep
from multiprocessing import Pool

# 创建文件夹
def mkdir(path):
    isExists = os.path.exists(path)
    if not isExists:
        os.makedirs(path) # 不存在则创建
        print('文件夹创建成功')
    else:
        print('文件夹已存在')
# 循环加零并转换数字为字符串
def forNum(n):
    n = int(n)
    if n < 10:
        a = '00000' + str(n)
    else:
        if n < 100:
            a = '0000' + str(n)
        else:
            if n < 1000:
                a = '000' + str(n)
            else:
                if n < 10000:
                    a = '00' + str(n)
                else:
                    if n < 100000:
                        a = '0' + str(n)
                    else:
                        if n < 1000000:
                            a = str(n)
    return a
# 分离m3u8文件 修改下载文件为 | 提取下载文件名称
def editM3u8(url_m3u8):
    f = open(url_m3u8, 'r', encoding='utf-8')
    # 分解文件行数据
    text_list = f.readlines()
    # 下载列表
    files = []
    # 提取下载文件列表
    for i in text_list:
        if i.find('#EX') == -1:
            files.append(str.strip(i))
    f.close()
    return files
# 调用线程下载文件
def download_init(files, tit_link):
    # 循环列表下载
    # 多线程
    if __name__ == "__main__":
        pool = Pool(4)
        for n in range(0, len(files)):
            a = forNum(n)
            url = tit_link + str(files[n])
            pool.apply_async(download, (url, a))
        pool.close()
        pool.join()
    # 单线程
    # for n in range(0, len(files)):
    #     # 数字加零并转换成字符串
    #     a = forNum(n)
    #     url = tit_link + str(n) + '.ts'
    #     # print(url, a)
    #     # 下载文件
    #     download(url, a)
# 下载文件
def download(url, num):
    # print(num + "开始下载")
    headers = {
        "User-Agent": "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/69.0.3497.100 Safari/537.36"
    }
    print(url)
    response = requests.get(url, headers=headers)
    f = open("./ts/" + num + '.ts', "wb")
    f.write(response.content)
    f.close()
    print(num + "下载完成")
# 删除文件
def delFile(file):
    os.remove(file)
# m3u8地址
url_m3u8 = './ts/boss.m3u8' # 文件路径
tit_link = "https://baikevideo.cdn.bcebos.com/media/mda-XzWxczOmPXU0VoI4/" # 下载前缀
mkdir('./ts/')
# 提取m3u8文件
files = editM3u8(url_m3u8)
# 启动下载进程
download_init(files, tit_link)
# sleep(2)
# 合并视频命令
# subprocess.Popen('cd ts && copy/b *.ts video.mp4', shell=True)
# 在windows系统下面，直接可以使用:copy/b *.ts video.mp4  把所有ts文件合成一个mp4格式文件