多线程的音频打标记的python实现（原创）

 

技术难度：

①需要一个UI界面，并且其中可进行相关参数的自调，最开始使用的是pygame的框架，后来转用tk界面；

②需要可以播放音频文件，MP3、WMA等格式；

③需要在播放音频的同时进行打标签操作，多线程技术；

④对音频进行相关拆分、叠加、重组，使用ffmpeg库进行相关操作；

 

一，UI界面

 

二，文件内容

使用 双击启动音频标记程序.bat 文件进行相对位置的文件启动，并且在 lib 中把所需的 ffmpeg 文件都已加载进去

 

三，程序源码

其中使用 threading 库的多线程功能，并且也使用到tk按钮中传递对象的功能。

#-*- coding:utf-8 -*-
#edited by Mufasa
#文本标记软件

# import pygame     #已经确定
import time
import threading
import tkinter,time,decimal,math,string,os
import tkinter.filedialog
import shutil
import tkinter as tk
import subprocess
from pydub import AudioSegment     #已经确定


def out_comma(data,num_1,num_2):
  txt=''
  if len(data)==1:
    for n in range(num_2):
      txt=txt+data[0]+'，'
    txt=txt[:-1]+'。'
  else:
    for i in data:
      for n in range(num_1):
        txt=txt+i+'，'
    if num_1!=0:
      txt=txt[:-1]+'。'
    for n in range(num_2):
      for i in data:
        txt=txt+i+'，'
      txt=txt[:-1]+'。'
  return txt




class assist():
  def write_txt_data(data,path):
    ls = os.linesep
    #data 为数组类型
    fobj = open(path,'w')
    fobj.writelines(['%s%s'%(x,ls)for x in data])
    fobj.close()
  def read_txt_data(path):
    fobj = open(path,'r')
    string = []
    for eachline in fobj:
      if eachline[:-1] != '':
        string.append(eachline[:-1])
    fobj.close()
    return string
    
    
class out_aboutcode(tk.Toplevel):
  def __init__(self):
    super().__init__()
    self.title('关于程序')# 弹窗界面
    self.setup_UI()
  def setup_UI(self):
    row1 = tk.Frame(self)
    row1.pack(fill="x")
    # tk.Label(row1, text='欢迎使用音频标记软件！软件基于Python36设计\n\n使用说明：\n1,先选择需要进行标记的音频文件；\n2,点击——标记音频按钮，开始进行标记其中[ , ]符为小段标记点，[ . ]为大段标记点；\n3,标记的音频播放完毕后，点击—生成标记按钮，直接转换成标记文件并存储在\\data\\文件名\\目录下；\n\n欢迎使用').pack(side=tk.LEFT)
    tk.Label(row1, text='欢迎使用音频标记软件！软件基于Python36设计\n\n使用说明：\n1，先点击[ 选择音频 ]，选定您要进行标记的音频文件（默认支持MP3和wav格式）\n2，点击[ 标记音频 ]，在文件的文本输入框中输入 , . 这两种符号分别进行小段和大段标记\n3，点击[ 生成标记 ]，将音频的标记信息进行存储\n4，点击[ 设置音频 ]，对音频的播放模式进行个性化设置\n\n祝使用愉快(*^_^*)').pack(side=tk.LEFT)
    
    row4 = tk.Frame(self)
    row4.pack(fill="x")
    tk.Button(row4, text="关闭窗口", command=self.cancel).pack(side=tk.RIGHT)
  def cancel(self):
    self.destroy()

'''
data_set.txt 
[0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15]
程序A    ["逗号次数","逗号停顿"，"句号次数"，"句号停顿" , "大段次数" , "大段停顿" , "全体音频文件播放次数" , "产生此音频"]
[3,0,3,0,1,1,2,1]
A 0 1 2 3
[3,0,3,0,1,1,2,1,3,0,0,0,1,1,0,1]

["3","0","3","0","1","1","2","1","3","0","0","0","1","1","0","1"]

[3,0,0,0,1,1,0,1]
程序B    ["逗号次数","逗号停顿"，"句号次数"，"句号停顿" , "大段次数" , "大段停顿" , "全体音频文件播放次数" , "产生此音频"]
'''

class out_setconfig(tk.Toplevel):
  def __init__(self):
    super().__init__()
    self.title('参数设置')# 弹窗界面
    self.data_set_path = os.path.dirname(os.path.realpath(__file__))+"\\data_set.txt"
    isExists = os.path.exists(self.data_set_path )
    if isExists == False:     #bug,之前没有设置
      self.setup_eror()
    else:
      self.setup_UI()

  def setup_eror(self):
    row1 = tk.Frame(self)
    row1.pack(fill="x")
    tk.Label(row1, text='程序第一次运行！\n\n已进行参数初始化操作\n').pack(side=tk.LEFT)
    row4 = tk.Frame(self)
    row4.pack(fill="x")
    assist.write_txt_data(["3","0","3","0","1","1","2","1","3","0","1","0","2","0","0","1"],self.data_set_path )
    tk.Button(row4, text="关闭窗口", command=self.cancel).pack(side=tk.RIGHT)
  def setup_UI(self):
    
    self.A00 = tk.StringVar()
    self.A01 = tk.StringVar()
    self.A10 = tk.StringVar()
    self.A11 = tk.StringVar()
    self.A20 = tk.StringVar()
    self.A21 = tk.StringVar()
    self.A30 = tk.StringVar()
    # self.chvarEn = tk.IntVar()
    self.A31 = tk.IntVar()
    
    self.B00 = tk.StringVar()
    self.B01 = tk.StringVar()
    self.B10 = tk.StringVar()
    self.B11 = tk.StringVar()
    self.B20 = tk.StringVar()
    self.B21 = tk.StringVar()
    self.B30 = tk.StringVar()
    self.B31 = tk.IntVar()
    
    data = assist.read_txt_data(self.data_set_path)
    # print(data)
    self.A00.set(data[0])
    self.A01.set(data[1])
    self.A10.set(data[2])
    self.A11.set(data[3])
    self.A20.set(data[4])
    self.A21.set(data[5])
    self.A30.set(data[6])
    # d = int(data[7])
    # self.A31.set(d)
    
    self.B00.set(data[8])
    self.B01.set(data[9])
    self.B10.set(data[10])
    self.B11.set(data[11])
    self.B20.set(data[12])
    self.B21.set(data[13])
    self.B30.set(data[14])
    # d = int(data[15])
    # self.B31.set(d)
    # self.B31.set(int(data[15]))
    
    row1 = tk.Frame(self)
    row1.pack(fill="x")
    
    tk.Label(row1,text='程序A',width=30 ).grid(row=0,column=0,columnspan=2)
    tk.Label(row1,text='逗号次数：',width=15).grid(row=1,column=0)
    tk.Label(row1,text='逗号停顿(秒)：',width=15).grid(row=2,column=0)
    tk.Label(row1,text='句号次数：',width=15).grid(row=3,column=0)
    tk.Label(row1,text='句号停顿(秒)',width=15).grid(row=4,column=0)
    tk.Label(row1,text='括号次数',width=15).grid(row=5,column=0)
    tk.Label(row1,text='括号停顿(秒)',width=15).grid(row=6,column=0)
    tk.Label(row1,text='全部音频次数',width=15).grid(row=7,column=0)
    tk.Label(row1,text='是否生成音频',width=15).grid(row=8,column=0)
    
    
    tk.Entry(row1, textvariable=self.A00, width=13).grid(row=1,column=1)
    tk.Entry(row1, textvariable=self.A01, width=13).grid(row=2,column=1)
    tk.Entry(row1, textvariable=self.A10, width=13).grid(row=3,column=1)
    tk.Entry(row1, textvariable=self.A11, width=13).grid(row=4,column=1)
    tk.Entry(row1, textvariable=self.A20, width=13).grid(row=5,column=1)
    tk.Entry(row1, textvariable=self.A21, width=13).grid(row=6,column=1)
    tk.Entry(row1, textvariable=self.A30, width=13).grid(row=7,column=1)
    checkA = tk.Checkbutton(row1, text="播放", variable=self.A31)
    checkA.select()
    checkA.grid(column=1, row=8, sticky=tk.W)
    
    
    tk.Label(row1,text='程序B',width=30 ).grid(row=0,column=2,columnspan=2)
    tk.Label(row1,text='逗号次数：',width=15).grid(row=1,column=2)
    tk.Label(row1,text='逗号停顿(秒)：',width=15).grid(row=2,column=2)
    tk.Label(row1,text='句号次数：',width=15).grid(row=3,column=2)
    tk.Label(row1,text='句号停顿(秒)',width=15).grid(row=4,column=2)
    tk.Label(row1,text='大段次数',width=15).grid(row=5,column=2)
    tk.Label(row1,text='大段停顿(秒)',width=15).grid(row=6,column=2)
    tk.Label(row1,text='全部音频次数',width=15).grid(row=7,column=2)
    tk.Label(row1,text='是否生成音频',width=15).grid(row=8,column=2)
    
    
    tk.Entry(row1, textvariable=self.B00, width=13).grid(row=1,column=3)
    tk.Entry(row1, textvariable=self.B01, width=13).grid(row=2,column=3)
    tk.Entry(row1, textvariable=self.B10, width=13).grid(row=3,column=3)
    tk.Entry(row1, textvariable=self.B11, width=13).grid(row=4,column=3)
    tk.Entry(row1, textvariable=self.B20, width=13).grid(row=5,column=3)
    tk.Entry(row1, textvariable=self.B21, width=13).grid(row=6,column=3)
    tk.Entry(row1, textvariable=self.B30, width=13).grid(row=7,column=3)
    checkB = tk.Checkbutton(row1, text="播放", variable=self.B31)
    checkB.select()
    checkB.grid(column=3, row=8, sticky=tk.W)
    self.A31.set(int(data[7]))
    self.B31.set(int(data[15]))
    
    tk.Button(row1, text="取消", command=self.cancel, width=13).grid(row=9,column=0)
    tk.Button(row1, text="确定", command=self.ok, width=13).grid(row=9,column=3)
    
  def ok(self):
    self.userinfo_A = [self.A00.get(), self.A01.get(),self.A10.get(),self.A11.get(),self.A20.get(),self.A21.get(),self.A30.get(),self.A31.get()] # 设置数据
    self.userinfo_B = [self.B00.get(), self.B01.get(),self.B10.get(),self.B11.get(),self.B20.get(),self.B21.get(),self.B30.get(),self.B31.get()] # 设置数据
    # print(self.userinfo)
    # print(self.userinfo_A[7])
    # print(self.userinfo_B[7])
    self.userinfo_A[7] = str(self.userinfo_A[7])
    self.userinfo_B[7] = str(self.userinfo_B[7])
    data = self.userinfo_A + self.userinfo_B
    # print(data)
    assist.write_txt_data(data,os.path.dirname(os.path.realpath(__file__))+"\\data_set.txt")
    self.destroy() # 销毁窗口
  def cancel(self):
    self.userinfo = None # 空！
    self.destroy()
    
################################
class main_code(tk.Tk):
  def __init__(self):
    super().__init__()
    self.audio_path = ""
    self.run = True
    self.vartext = tkinter.StringVar()
    
    self.txt=''
    
    
    self.mark_text = '欢迎使用！'
    self.insert_text = ''
    self.vartext.set(self.mark_text)
    self.filename = ""
    self.audio_path = ""
    self.audio_len = 0
    self.data_play = []
    self.sound = 0
    self.title('音频标记')
    self.setupUI()
    
  def setupUI(self):
    
    
    row4 = tk.Frame(self)
    row4.pack(fill="x")
    
    tk.Button(row4,text='导入文本',width=27,command=self.select).grid(row=0,column=0)
    tk.Button(row4,text='设置文本',width=27,command=self.setup_config).grid(row=0,column=1)
    tk.Button(row4,text='生成文本',width=27,command=self.lay).grid(row=0,column=2)
    
    label = tkinter.Label(row4,height=3,textvariable=self.vartext, bg='white', font=('黑体', 10), anchor='w').grid(row=1,column=0,columnspan=3)
    
    
    entry1 = tkinter.Entry(row4,width=85)
    entry1.bind('<Key>', self.insert_mark)
    entry1.grid(row=2,column=0,columnspan=3)
    self.insert_text = entry1.get()


  def select(self):   #获取audio_path和audio_len
    self.audio_path = tkinter.filedialog.askopenfilename(initialdir = 'C:\\Users\\Administrator\\Desktop',filetypes=( ("txt or docx files", "*.txt;*.docx"),("All files", "*.*")))
    if self.audio_path=="":
      self.vartext.set('未导入任何文件')
    else:
      p,f=os.path.split(self.audio_path)
      path = os.path.dirname(os.path.realpath(__file__)) + '\\data\\' + str(os.path.splitext(f)[0])
      isExists = os.path.exists(path)
      if not isExists:  #如果不存在这个目录
        os.makedirs(path)  #创建一个新的路径
        shutil.copyfile(self.audio_path,path+'\\'+f)
      if str(os.path.splitext(f)[1])=='.txt':
        file_object = open(self.audio_path,'r', encoding='UTF-8')
        try:
          file_context = file_object.read()
        finally:
          file_object.close()
        self.txt=file_context
        self.vartext.set('已导入txt文本')
      # print(1)
      elif str(os.path.splitext(f)[1])=='.docx':
        import docx
        document = docx.Document(self.audio_path)
        txt=''
        for paragraph in document.paragraphs:
          txt+=paragraph.text
        self.txt=txt
        self.vartext.set('已导入docx文本')
      self.audio_path=path+'\\'+f#更新数据路径
    
  def insert_mark(self,event):  #非主动线程
    # while self.run:
    # self.insert_text = self.insert_text + event.char
    pass
  def lay(self):
    # txt=entry1.get()
    print(self.insert_text)
    print(123)
    
    '''
    self.vartext.set('正在生成文本标记文件')
    # data_set = assist.read_txt_data(os.path.dirname(os.path.realpath(__file__))+"\\data_set.txt")
    mark_set=[1,1,0,1,1,0,1,1]
    import re
    txt_comma=re.split(u'。',self.txt[:-1])
    p,f=os.path.split(self.audio_path)
    # print(p)
    # print(f)
    for i in range(len(txt_comma)):
      txt_comma[i]=txt_comma[i].split('，')
    
    # print(txt_comma)
    if mark_set[0]==1:
      txt_A=''
      for i in txt_comma:
        txt_A=txt_A+out_comma(i,mark_set[1],mark_set[2])
      # print(txt_A)
      fobj=open(p+'\\'+str(os.path.splitext(f)[0])+'_A'+str(os.path.splitext(f)[1]),'w')
      fobj.write(txt_A)
      fobj.close()
    
    if mark_set[4]==1:
      txt_B=''
      for i in txt_comma:
        txt_B=txt_B+out_comma(i,mark_set[5],mark_set[6])
      # print(txt_B)
      fobj=open(p+'\\'+str(os.path.splitext(f)[0])+'_B'+str(os.path.splitext(f)[1]),'w')
      fobj.write(txt_B)
      fobj.close()
    '''
    
    
    
    
    
  def setup_config(self):
    res = self.ask_userinfo()
    if res is None: return
  def ask_userinfo(self):
    inputDialog = out_setconfig()
    self.wait_window(inputDialog) # 这一句很重要！！！



if __name__ == '__main__':
  app = main_code()
  app.mainloop()