Python异步下载m3u8_python 异步下载文件

在Python中，你可以使用aiohttp库和asyncio进行异步HTTP请求，以及使用ffmpeg库对m3u8文件进行处理。首先，确保安装了以下库：

pip install aiohttp
pip install ffmpeg-python

1
2
3

然后，你可以编写一个异步下载m3u8的程序：

import os
import asyncio
import aiohttp
import ffmpeg

async def download_segment(session, url, filename):
    async with session.get(url) as response:
        with open(filename, 'wb') as f:
            while True:
                chunk = await response.content.read(1024)
                if not chunk:
                    break
                f.write(chunk)

async def download_m3u8(url, output):
    async with aiohttp.ClientSession() as session:
        async with session.get(url) as response:
            m3u8_content = await response.text()

        base_url = url.rsplit('/', 1)[0] + '/'
        lines = m3u8_content.splitlines()
        tasks = []

        for i, line in enumerate(lines):
            if line.startswith('#'):
                continue
            filename = f'segment_{i}.ts'
            segment_url = base_url + line
            tasks.append(asyncio.ensure_future(download_segment(session, segment_url, filename)))

        await asyncio.gather(*tasks)

        input_files = '|'.join([f'segment_{i}.ts' for i in range(len(tasks))])
        ffmpeg.concat(input_files, output, vcodec='copy', acodec='copy', safe=0).run()
        for i in range(len(tasks)):
            os.remove(f'segment_{i}.ts')

if __name__ == "__main__":
    m3u8_url = 'https://your-m3u8-url.m3u8'
    output_file = 'output.mp4'

    loop = asyncio.get_event_loop()
    loop.run_until_complete(download_m3u8(m3u8_url, output_file))

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

确保将m3u8_url替换为你要下载的m3u8文件的URL。这个程序将下载m3u8文件中的所有片段，并将它们合并到一个名为output.mp4的输出文件中。

请注意，这个程序可能需要根据你的需求进行一定的修改，例如处理加密的m3u8文件。
如果您想要下载加密的m3u8文件，您需要处理KEY文件。以下是一个修改后的程序，用于处理加密的m3u8文件。为了解密，我们将使用cryptography库，因此请确保安装了这个库：

pip install cryptography

1
2

以下是一个处理加密的m3u8文件的示例程序：

import os
import re
import asyncio
import aiohttp
import ffmpeg
from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
from cryptography.hazmat.primitives import serialization, hashes
from cryptography.hazmat.backends import default_backend

async def download_key(session, url):
    async with session.get(url) as response:
        key_data = await response.read()
    return key_data

async def download_and_decrypt_segment(session, url, key, iv, filename):
    async with session.get(url) as response:
        encrypted_data = await response.read()

    cipher = Cipher(algorithms.AES(key), modes.CBC(iv), backend=default_backend())
    decryptor = cipher.decryptor()
    decrypted_data = decryptor.update(encrypted_data) + decryptor.finalize()

    with open(filename, 'wb') as f:
        f.write(decrypted_data)

async def download_m3u8(url, output):
    async with aiohttp.ClientSession() as session:
        async with session.get(url) as response:
            m3u8_content = await response.text()

        base_url = url.rsplit('/', 1)[0] + '/'
        lines = m3u8_content.splitlines()

        key = None
        iv = None
        tasks = []

        for i, line in enumerate(lines):
            if line.startswith('#EXT-X-KEY'):
                key_url = re.search(r'URI="(.+?)"', line).group(1)
                if not key_url.startswith('http'):
                    key_url = base_url + key_url
                key = await download_key(session, key_url)
                iv = bytes.fromhex(re.search(r'IV=0x(.+)', line).group(1))
            elif line.startswith('#'):
                continue
            else:
                filename = f'segment_{i}.ts'
                segment_url = base_url + line
                tasks.append(asyncio.ensure_future(download_and_decrypt_segment(session, segment_url, key, iv, filename)))

        await asyncio.gather(*tasks)

        input_files = '|'.join([f'segment_{i}.ts' for i in range(len(tasks))])
        ffmpeg.concat(input_files, output, vcodec='copy', acodec='copy', safe=0).run()
        for i in range(len(tasks)):
            os.remove(f'segment_{i}.ts')

if __name__ == "__main__":
    m3u8_url = 'https://your-encrypted-m3u8-url.m3u8'
    output_file = 'output.mp4'

    loop = asyncio.get_event_loop()
    loop.run_until_complete(download_m3u8(m3u8_url, output_file))

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

这个程序将下载加密的m3u8文件中的所有片段，并解密它们。然后将解密后的片段合并到名为output.mp4的输出文件中。请确保将m3u8_url替换为要下载的加密m3u8文件的URL。

请注意，此代码可能需要根据您的需求进行调整，例如处理不同加密类型、IV的来源等。