python将下载到本地m3u8视频合成MP4的代码详解

import os
import requests
import datetime
from Crypto.Cipher import AES

def decode_key_file(key_file_name):
  with open(key_file_name,\"r\") as f:
    data=f.read()
  return data
def decode_m_file(m_file_name):
  with open(m_file_name,\"r\") as f:
    data=f.read()
  return data.split(\"\\n\")[6::2]

def decode_video(filename):
  data_list=[[],[],[]]
  file_list=os.listdir(filename)
  for one_file_name in file_list:
    if \"m3u8\" in one_file_name:
      data_list[0].append(filename+\"/\"+one_file_name)
    elif \"key\" in one_file_name:
      data_list[1].append(filename + \"/\" + one_file_name)
    else:
      data_list[2].append(filename + \"/\" + one_file_name)
  key=decode_key_file(data_list[1][0])
  name_list=decode_m_file(data_list[0][0])

  for one_name in name_list:
    for path_name in data_list[2]:
      if one_name in path_name:
        with open(path_name,\"rb\") as f:
          data_name=f.read()

        if len(key): # AES 解密
          cryptor = AES.new(key, AES.MODE_CBC, key)
          with open(key + \".mp4\", \'ab\') as f:
            f.write(cryptor.decrypt(data_name))


if __name__ == \'__main__\':
  decode_video(\"./data\")

def decode_key_file(key_file_name):
  with open(key_file_name, \"r\") as f:
    data = f.read()
  return data


def decode_m_file(m_file_name):
  with open(m_file_name, \"r\") as f:
    data = f.read()
  return data.split(\"\\n\")[6::2]


def decode_video(filename):
  data_list = [[], [], []]
  file_list = os.listdir(filename)
  for one_file_name in file_list:
    if \"m3u8\" in one_file_name:
      data_list[0].append(filename + \"/\" + one_file_name)
    elif \"key\" in one_file_name:
      data_list[1].append(filename + \"/\" + one_file_name)
    else:
      data_list[2].append(filename + \"/\" + one_file_name)
  try:
    key = decode_key_file(data_list[1][0])
  except:
    key = one_file_name
  name_list = decode_m_file(data_list[0][0])

  for one_name in name_list:
    for path_name in data_list[2]:
      if one_name in path_name:
        with open(path_name, \"rb\") as f:
          data_name = f.read()

        if len(key): # AES 解密
          cryptor = AES.new(key, AES.MODE_CBC, key)
          with open(key + \".mp4\", \'ab\') as f:
            f.write(cryptor.decrypt(data_name))
        else:
          with open(key + \".mp4\", \'ab\') as f:
            f.write(data_name)
            f.flush()


if __name__ == \'__main__\':
  data_list = os.listdir(\"/home/dfy/Videos/Download\")
  index=0
  for path_one in data_list:
    index+=1
    if index<18:
      continue
    decode_video(\"/home/dfy/Videos/Download\" + \"/\" + path_one)