dltest 发表于 2022-10-15 10:17

[原码]直播流录制-在线录制你喜欢的直播节目永久保存,随时观看

本帖最后由 dltest 于 2022-10-15 10:20 编辑

区别于之前我发的贴子,本程序需要你,提前获得直播流地址,将此地址复制到框内,再输入主播名用于创建保存文件夹。
程序运行后会自动保存本次输入框内数据,方便下次开启软件直接加载
开始录制后,屏幕会出现一个黑框,是本程序调用ffmpeg.exe,关闭黑框,录制进程也会关屏

打包后的程序下载如下: 同样,本程序同时打包了ffmpeg.exe,所以包比较大
https://wwz.lanzoub.com/ibgJd0dw58uj
密码:52pj


代码如下

# -*- 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()



czp13 发表于 2022-10-15 12:19

能录加密视频码

nagra 发表于 2022-10-15 12:48

文件删除了?

FBIdeal999 发表于 2022-10-15 13:11

直播的不好找,用fiddler可以,但是用fiddler就不需要你这个软件了吧?

rufan321 发表于 2022-10-15 13:12

想评分提示要过24小时

s7700 发表于 2022-10-15 13:23

试试看效果如何

mainblog 发表于 2022-10-15 13:32

谢谢分享。

wsxb 发表于 2022-10-15 14:48

支持支持,看起来很好用

rhci 发表于 2022-10-15 15:28

这个不错,看看。谢谢

paihan 发表于 2022-10-16 10:28

有点问题,我测试黑框一闪而过,文件夹下也没有ts文件
页: [1] 2
查看完整版本: [原码]直播流录制-在线录制你喜欢的直播节目永久保存,随时观看