[Python] 纯文本查看 复制代码
# -*- coding:utf-8 -*-
#@FileName :m3u_flv_2_ts.py
# @ Author :dltest@52pojie
import configparser
import datetime
import os
import subprocess
import threading
import time
import tkinter as tk
import tkinter.font as tkFont
thd = set()
LOG_LINE_NUM =0
class App:
def __init__(self, root):
self.editdate = '2022/10/15'
self.initUI(root)
self.initData()
def initData(self):
self.tag = 'odd'
self.count = 0
self.dir = ''
self.flag = True
if not os.path.exists('ZBconfig.ini'):
f = open("ZBconfig.ini",'w', encoding='utf-8-sig')
f.close()
self._log("创建配置文件")
else:
self._log("加载配置文件")
zhubo_address,zhubo_name= self.getzhubo()
if zhubo_address:
self._log(f'加载地址【{zhubo_address}】')
self._log(f'加载主播名【{zhubo_name}】')
self.GLineEdit_492.insert(0,zhubo_address)
self.GLineEdit_956.insert(0,zhubo_name)
else:
self._log('请输入直播间地址')
def getzhubo(self):
try:
config = configparser.ConfigParser()
config.read('ZBconfig.ini', encoding='utf-8-sig')
zhubo_address=config.get('1', '直播地址')
zhubo_name=config.get('1', '主播名')
return zhubo_address,zhubo_name
except:
pass
return None,None
def initUI(self,root):
#setting title
root.title(f"直播流录制 {self.editdate} dltest@52pojie")
#setting window size
width=770
height=383
screenwidth = root.winfo_screenwidth()
screenheight = root.winfo_screenheight()
alignstr = '%dx%d+%d+%d' % (width, height, (screenwidth - width) / 2, (screenheight - height) / 2)
root.geometry(alignstr)
root.resizable(width=False, height=False)
self.GLabel_597=tk.Label(root)
ft = tkFont.Font(family='Times',size=10)
self.GLabel_597["font"] = ft
self.GLabel_597["fg"] = "#333333"
self.GLabel_597["justify"] = "center"
self.GLabel_597["text"] = "直播地址"
self.GLabel_597.place(x=0,y=10,width=61,height=30)
self.GLineEdit_492=tk.Entry(root)
self.GLineEdit_492["borderwidth"] = "1px"
ft = tkFont.Font(family='Times',size=10)
self.GLineEdit_492["font"] = ft
self.GLineEdit_492["fg"] = "#333333"
self.GLineEdit_492["justify"] = "center"
self.GLineEdit_492["text"] = "httpurl"
self.GLineEdit_492.place(x=60,y=10,width=309,height=30)
self.GLineEdit_956=tk.Entry(root)
self.GLineEdit_956["borderwidth"] = "1px"
ft = tkFont.Font(family='Times',size=10)
self.GLineEdit_956["font"] = ft
self.GLineEdit_956["fg"] = "#333333"
self.GLineEdit_956["justify"] = "center"
self.GLineEdit_956["text"] = "zhoboname"
self.GLineEdit_956.place(x=440,y=10,width=81,height=30)
self.GLabel_603=tk.Label(root)
ft = tkFont.Font(family='Times',size=10)
self.GLabel_603["font"] = ft
self.GLabel_603["fg"] = "#333333"
self.GLabel_603["justify"] = "center"
self.GLabel_603["text"] = "主播名"
self.GLabel_603.place(x=380,y=10,width=47,height=30)
self.GButton_316=tk.Button(root)
self.GButton_316["bg"] = "#efefef"
ft = tkFont.Font(family='Times',size=10)
self.GButton_316["font"] = ft
self.GButton_316["fg"] = "#000000"
self.GButton_316["justify"] = "center"
self.GButton_316["text"] = "开始"
self.GButton_316.place(x=530,y=10,width=72,height=30)
self.GButton_316["command"] = self.GButton_316_command
self.GButton_35=tk.Button(root)
self.GButton_35["bg"] = "#efefef"
ft = tkFont.Font(family='Times',size=10)
self.GButton_35["font"] = ft
self.GButton_35["fg"] = "#000000"
self.GButton_35["justify"] = "center"
self.GButton_35["text"] = "停止"
self.GButton_35.place(x=610,y=10,width=71,height=30)
self.GButton_35["command"] = self.GButton_35_command
self.GButton_610=tk.Button(root)
self.GButton_610["bg"] = "#efefef"
ft = tkFont.Font(family='Times',size=10)
self.GButton_610["font"] = ft
self.GButton_610["fg"] = "#000000"
self.GButton_610["justify"] = "center"
self.GButton_610["text"] = "打开"
self.GButton_610.place(x=690,y=10,width=73,height=30)
self.GButton_610["command"] = self.GButton_610_command
self.GLineEdit_50=tk.Text(root)
self.GLineEdit_50["borderwidth"] = "1px"
ft = tkFont.Font(family='Times',size=10)
self.GLineEdit_50["font"] = ft
self.GLineEdit_50["fg"] = "#333333"
self.GLineEdit_50.place(x=10,y=50,width=751,height=323)
def GButton_316_command(self): #开始
now = time.strftime("%Y-%m-%d-%H-%M-%S",time.localtime(time.time()))
zhubo = self.GLineEdit_956.get()
if len(zhubo) < 1:
zhubo = '未命名'
if not os.path.exists(zhubo):
os.makedirs('./'+zhubo)
self.dir =os.getcwd() + '\\' + zhubo
filename=zhubo + '_' + now + ".ts"
ffmpeg_path = "ffmpeg"
file = os.getcwd() + '\\' +zhubo +'\\'+ filename
self._log(f'保存路径{file}')
start = datetime.datetime.now()
hls_pull_url = self.GLineEdit_492.get()
if len(hls_pull_url) <1:
self._log(f'未输入直播地址')
return
self._log(f'直播地址：{hls_pull_url}')
self._log(f'--------开始录制-----------')
self.download(ffmpeg_path,hls_pull_url,file)
self.flag = True
config = configparser.ConfigParser()
config.read('ZBconfig.ini', encoding='utf-8-sig')
listx = config.sections()# 获取到配置文件中所有分组名称
if '1' not in listx:# 如果分组type不存在则插入type分组
config.add_section('1')
config.set('1', '直播地址', hls_pull_url)# 给type分组设置值
config.set('1', '主播名', zhubo)# 给type分组设置值
o = open('ZBconfig.ini', 'w',encoding='utf-8-sig')
config.write(o)
o.close()#不要忘记关闭
obj1 = threading.Thread(target=self.displayinfo, args=({start,zhubo}))
obj1.setDaemon(True)
obj1.start()
def GButton_35_command(self): #停止
self.flag = False
global thd
for i in thd:
i.stdin.write('q'.encode("GBK"))
i.communicate()
thd.clear()
def GButton_610_command(self): #打开
path = self.dir
if path:
self.open_fp(path)
else:
path =os.getcwd() + '\\未命名'
self.open_fp(path)
def open_fp(self, fp):
import platform
systemType: str = platform.platform() # 获取系统类型
if 'mac' in systemType: # 判断以下当前系统类型
fp: str = fp.replace("\\", "/") # mac系统下,遇到`\\`让路径打不开,不清楚为什么哈,觉得没必要的话自己可以删掉啦,18行那条也是
subprocess.call(["open", fp])
else:
fp: str = fp.replace("/", "\\") # win系统下,有时`/`让路径打不开
try:
os.startfile(fp)
except:
self._log(f'{"-" * 20}文件还未下载{"-" * 20}')
def download(self,ffmpeg_path,hls_pull_url,file):
try:
_output = subprocess.Popen([
ffmpeg_path, "-y",
"-v","verbose",
"-rw_timeout","10000000", # 10s
"-loglevel","error",
"-hide_banner",
# "-user_agent",headers["User-Agent"],
"-analyzeduration","2147483647",
"-probesize","2147483647",
"-i",hls_pull_url,
'-bufsize','5000k',
"-map","0",
"-sn","-dn",
"-f","mpegts",
# "-bsf:v","h264_mp4toannexb",
# "-c","copy",
"-c:v","copy",
'-max_muxing_queue_size','64',
"{path}".format(path=file),
], stderr = subprocess.STDOUT,stdin=subprocess.PIPE)
except subprocess.CalledProcessError as exc:
print(str(exc.output) +" 发生错误的行数: "+str(exc.__traceback__.tb_lineno) )
self.flag = False
global thd
thd.add(_output)
def displayinfo(self,start:datetime.datetime,zhubo):
start1 = datetime.datetime.now()
time.sleep(10)
while self.flag:
time.sleep(30)
self._log('循环值守录制抖音直播')
# self._log("x"*60)
end = datetime.datetime.now()
self._log(f"{zhubo} 正在录制中")
self._log('总共录制时间: ' +str(end - start1))
self._log("x"*60)
return
def _log(self, logmsg):
# logmsg = logmsg[:25] if len(logmsg) > 25 else logmsg
global LOG_LINE_NUM
current_time = self.get_current_time()
logmsg_in = str(current_time) + " " +str(logmsg) + "\n" # 换行
self.GLineEdit_50.tag_config("even", background='#e0e0e0')
self.GLineEdit_50.tag_config("odd", background='#ffffff')
self.tag = 'odd' if self.tag == 'even' else 'even'
if LOG_LINE_NUM <= 20:
self.GLineEdit_50.insert('end', logmsg_in, self.tag)
LOG_LINE_NUM = LOG_LINE_NUM + 1
else:
self.GLineEdit_50.delete(1.0, 2.0)
self.GLineEdit_50.insert('end', logmsg_in, self.tag)
def get_current_time(self):
current_time = time.strftime('%H:%M:%S', time.localtime(time.time()))
return current_time
def on_closing() :
global thd
if len(thd)>0:
for i in thd:
i.stdin.write('q'.encode("GBK"))
i.communicate()
print ("Window closed")
root.destroy()
if __name__ == "__main__":
root = tk.Tk()
app = App(root)
root.protocol("WM_DELETE_WINDOW", on_closing)
root.mainloop()
发表于 2022-10-15 10:17
