python基于tkinter制作m3u8视频下载工具

这是我为了学习tkinter用python 写的一个下载m3u8视频的小程序，程序使用了多线程下载，下载后自动合并成一个视频文件，方便播放。

目前的众多视频都是m3u8的播放类型，只要知道视频的m3u8地址，就可以完美下载整个视频。

m3u8地址获取

打开浏览器，点开你要获取地址的视频

重要的来了，右键>>审查元素或者按F12也可以

根据开发或测试的实际环境选择相应的设备，选择iphone6 plus

选择好了以后，刷新页面，点击漏斗，选择media，一定刷新之后再点击，没出来的话切换几下选项卡，就能出来了

点击播放视频，在下边就可以看到地址了

程序代码

# -*- coding: UTF-8 -*-
import sys

sys.path.append("C:\\Python36-32\\Lib\\site-packages")
import tkinter
import re
import urllib3
import threadpool
import threading
import os
import shutil
import time
import glob
from tkinter.ttk import *
from PIL import Image, ImageTk
import pyperclip
from tkinter.filedialog import askdirectory

def get_image(filename,width,height):
   im = Image.open(filename).resize((width,height))
   return ImageTk.PhotoImage(im)

def get_resource_path(relative_path):
   if hasattr(sys, '_MEIPASS'):
       return os.path.join(sys._MEIPASS, relative_path)
   return os.path.join(os.path.abspath("."), relative_path)

def getrealtask(link):
   global key
   rooturl1 = ''
   rooturl2 = ''
   pattern3 = re.compile(r'^.*[\/]', re.M)
   result11 = pattern3.findall(link)
   if result11:
       rooturl1 = result11[0]
   pattern4 = re.compile(r'^http[s]?:\/\/[^\/]*', re.M)
   result114 = pattern4.findall(link)
   if result114:
       rooturl2 = result114[0]
   res = http.request('GET', link)
   content = str(res.data, 'utf8')
   list = content.split('\n')
   reallist = []
   for one in list:
       if one.endswith('"key.key"'):
           keyurl = rooturl1 + "key.key"
           res = http.request('GET', keyurl)
           key = str(res.data, 'utf8')
       if one.endswith('.ts') or one.endswith('.image'):
           if re.match(r'http', one, re.M | re.I):
               reallist.append(one)
           elif re.match(r'\/', one, re.M | re.I):
               reallist.append(rooturl2 + one)
           else:
               reallist.append(rooturl1 + one)
       if one.endswith('.m3u8'):
           if re.match(r'\/', one, re.M | re.I):
               reallist = getrealtask(rooturl2 + one)
           else:
               reallist = getrealtask(rooturl1 + one)
           break
   return reallist

def download_ts(result):
   url = result['url']
   name = result['name']
   num = result['num']
   rootpath = result['root']
   m3u8Name = result['m3u8name']
   t= str(result['total'])
   if num ％ 10000 == 0:
       print(str(num)+' / '+t)
   basepath = os.path.join(rootpath,m3u8Name)
   fullpath = os.path.join(basepath,name)
   isExist = os.path.exists(fullpath)
   if not isExist:
       http = urllib3.PoolManager(timeout=10.0)
       while(1):
           try:
               f = http.request('GET', url)
               break
           except:
               print("URL ERRO: " + url)
               time.sleep(2)
       d = f.data
       with open(fullpath, "wb") as code:
           code.write(d)
       print("SAVE: " + url)
def clock2(num,path):
   global window
   global key
   v3 = tkinter.StringVar();
   v4 = tkinter.StringVar();
   l3 = tkinter.Label(window, text='', textvariable=v3, font=('Arial', 10))
   l4 = tkinter.Label(window, text='', textvariable=v4, font=('Arial', 10))
   l3.place(x=10, y=130, anchor='nw')
   l4.place(x=10, y=160, anchor='nw')
   v3.set("下载中。。。")
   while(1):
       path_file_number = len(glob.glob(path+'/*.ts'))
       mp4_file_number = len(glob.glob(path + '/*.mp4'))
       numberstr = str(path_file_number) + '/'+str(num)
       v4.set(numberstr)
       if mp4_file_number==1:
           v3.set("下载完成！")
           key = ''
           break

def clock1():
   global v
   global v2
   global rootpath
   m3u8Name = v2.get()
   url = v.get()
   print(url)
   urls = getrealtask(url)
   total = len(urls)
   i = 0
   tasks = []
   tsNames = []
   for one in urls:
       task = {}
       task['root'] = rootpath
       task['m3u8name'] = m3u8Name
       task['url'] = one
       task['num'] = i
       task['total'] = total
       task['name'] = str(i) + '.ts'
       tsNames.append(str(i) + '.ts')
       i = i + 1
       tasks.append(task)
   print('tasks: ' + str(len(tasks)))
   targetpath = os.path.join(rootpath, m3u8Name)
   if not os.path.exists(targetpath):
       os.makedirs(targetpath)
   timer2 = threading.Thread(target=clock2,args=(len(tasks),targetpath))
   timer2.daemon = True
   timer2.start()
   requests = threadpool.makeRequests(download_ts, tasks)
   [task_pool.putRequest(req) for req in requests]
   task_pool.wait()
   mp4targetfile = os.path.join(targetpath, m3u8Name + '.mp4')
   with open(mp4targetfile, 'wb') as f:
       for ts in tsNames:
           tstargetfile = os.path.join(targetpath, ts)
           with open(tstargetfile, 'rb') as mergefile:
               shutil.copyfileobj(mergefile, f)
           print(tstargetfile + ' merged.')
       for tts in tsNames:
           tstargetfile = os.path.join(targetpath, tts)
           os.remove(tstargetfile)
   print(total)

def hit_me():
   global on_hit
   timer = threading.Thread(target=clock1)
   timer.daemon = True
   timer.start()
   return

def choose_dir():
   global v5
   global rootpath
   rootpath = askdirectory()
   v5.set('文件夹: '+rootpath+'/')
   return

def about():
   window = tkinter.Toplevel()
   window.geometry('600x100')# Note Toplevel, NOT Tk.
   msg = 'Rax m3u8下载器 v1.4\n写这个程序主要是为了学习Tk，顺便满足下自己看视频的需求。\n家里的移动网络看在线视频还是有些卡顿的。 '
   label = tkinter.Label(window, text=msg,font=('Arial', 15))
   label.grid()
def update():
   window = tkinter.Toplevel()
   window.geometry('250x200')
   msg = 'Rax m3u8下载器 v1.5\n可以选择保存的目录了\nRax m3u8下载器 v1.4\n增加了菜单栏'
   label = tkinter.Label(window, text=msg,font=('Arial', 13))
   label.place(x=30, y=30, anchor='nw')
def donate():
   window = tkinter.Toplevel()
   window.geometry('500x400')
   msg = '软件免费使用\n欢迎喜欢此软件的各位大佬打赏，谢谢。'
   label = tkinter.Label(window, text=msg, font=('Arial', 20))

i1 = tkinter.PhotoImage(file=get_resource_path("images\\wx.png"))
   i2 = tkinter.PhotoImage(file=get_resource_path("images\\zfb.png"))
   imagelabel = tkinter.Label(window, text='aaa', image=i1, font=('Arial', 10))
   imagelabel2 = tkinter.Label(window, text='vvv', image=i2, font=('Arial', 10))
   imagelabel.place(x=10, y=145, anchor='nw')
   imagelabel2.place(x=230, y=145, anchor='nw')
   label.place(x=40, y=50, anchor='nw')
   window.mainloop()
def clear():
   global v
   v.set("")
def paste():
   global v
   v.set(pyperclip.paste())

key = ''
on_hit = False
rootpath = "d:\\"
#最高50线程
task_pool = threadpool.ThreadPool(50)
http = urllib3.PoolManager(timeout=5.0)
urllib3.disable_warnings()

#主窗口初始化
window = tkinter.Tk()
window.style = Style()
window.style.theme_use("clam")
window.title("Rax m3u8视频下载器")
window.geometry('500x300')
window.resizable(0,0)

#飞机背景图
canvas_root = tkinter.Canvas(window,width=500,height=300)
im_root = get_image(get_resource_path('images\\feiji.jpeg'),500,300)
canvas_root.create_image(250,240,image=im_root)
canvas_root.pack()

#各控件初始状态
l1 = tkinter.Label(window, text='m3u8地址：', font=('Arial', 10))
l1.place(x=10, y=0, anchor='nw')

#   地址栏
v = tkinter.StringVar();
e2 = tkinter.Entry(window, show=None, textvariable = v,font=('Arial', 10),width=40)
v.set('')
e2.place(x=10, y=30, anchor='nw')

#   视频名称
l6 = tkinter.Label(window, text = ' 视频文件名称：', font=('Arial', 10))
l6.place(x=0, y=90, anchor='nw')

#   视频名称栏
v2 = tkinter.StringVar();
e3 = tkinter.Entry(window, show=None, textvariable = v2,font=('Arial', 10),width=15)
v2.set('')
e3.place(x=105, y=90, anchor='nw')

#   保存位置
v5 = tkinter.StringVar();
l2 = tkinter.Label(window, textvariable = v5, font=('Arial', 10))
v5.set('文件夹: D:/')
l2.place(x=10, y=60, anchor='nw')

#   下载按钮
b = tkinter.Button(window, text='下载', font=('Arial', 10), width=10, command=hit_me)
b.place(x=400, y=100, anchor='nw')

#   选择路径按钮
pathselectButton = tkinter.Button(window, text='选择路径', font=('Arial', 10), width=10, command=choose_dir)
pathselectButton.place(x=400, y=60, anchor='nw')

#   清空按钮
b2 = tkinter.Button(window, text='清空', font=('Arial', 10), width=10, command=clear)
b2.place(x=300, y=25, anchor='nw')

#   粘贴地址按钮
b3 = tkinter.Button(window, text='粘贴地址', font=('Arial', 10), width=10, command=paste)
b3.place(x=400, y=25, anchor='nw')

#   求捐赠按钮
l5 = tkinter.Label(window, text='软件免费使用，欢迎各位喜欢此软件的大佬打赏，谢谢。\nQQ讨论群：519565890', font=('Arial', 10))
l5.place(x=100, y=160, anchor='nw')

window.option_add('*tearOff', False)

#菜单栏
menubar = tkinter.Menu(window)
window['menu'] = menubar
help_menu = tkinter.Menu(menubar)
menubar.add_command(label='捐助作者',command=lambda: donate())
menubar.add_cascade(menu=help_menu, label='帮助')

#帮助 下拉菜单
help_menu.add_command(label='更新内容',command=lambda: update())
help_menu.add_command(label='关于',command=lambda: about())

# 进入消息循环
window.mainloop()

项目地址

https://github.com/raxar81/rax_m3u8_downloader

来源：https://github.com/raxar81/rax_m3u8_downloader