download and merge m3u8

download_encrypted_m3u8.py

#!/usr/bin/env python3

import time
import m3u8
import sys
import subprocess
from os.path import abspath
from os import makedirs


def getNewDirName():
    localtime = time.localtime()
    name = time.strftime("%Y%m%d_%H%M%S", localtime)
    return name


class M3u8Session:
    def __init__(self, url, dir=None):
        self.dir = dir if dir else getNewDirName()
        try:
            makedirs(self.dir)
        except:
            pass
        self.m3u8 = m3u8.load(url)
        self.localM3u8Path = abspath(self.dir + "/ffmpeg.m3u8")
        self.aria2InputLines = []

    def rewriteKeyUrls(self):
        for i, key in enumerate(self.m3u8.keys):
            if key is None: continue
            self.aria2InputLines.append("%s" % key.absolute_uri)
            self.aria2InputLines.append("\tout=%d.key" % i)
            key.uri = "%d.key" % i

    def getSegmentUrls(self):
        self.aria2InputLines += list(
            map(lambda s: s.absolute_uri, self.m3u8.segments))

    def dumpM3u8(self):
        self.rewriteKeyUrls()
        self.getSegmentUrls()
        self.m3u8.dump(self.localM3u8Path)

    def aria2AndWait(self):
        aria2 = aria2 = subprocess.Popen(
            [
                'aria2c', '--console-log-level', 'warn', '-d',
                abspath(self.dir), '-x', '16', '-j', '16', '-c', '-i', '-'
            ],
            stdin=subprocess.PIPE)
        aria2.communicate("\n".join(self.aria2InputLines).encode("utf-8"))
        return aria2.returncode == 0

    def ffmpegAndWait(self):
        ffmpeg = subprocess.Popen([
            'ffmpeg', '-allowed_extensions', 'ALL', '-i', self.localM3u8Path,
            '-c', 'copy', 'output.mp4'
        ])
        ffmpeg.communicate()
        return ffmpeg.returncode == 0

    def run(self):
        self.dumpM3u8()
        assert self.aria2AndWait()
        assert self.ffmpegAndWait()
        return


def main():
    url = sys.argv[1]
    directory = sys.argv[2] if len(sys.argv) >= 3 else None
    m = M3u8Session(url, directory)
    m.run()


if __name__ == '__main__':
    main()