python自制录屏小工具

import wave
import pyaudio
import cv2
import moviepy
import threading
import _thread
import numpy as np
from time import sleep
from PIL import ImageGrab
from datetime import datetime
from moviepy.editor import AudioFileClip,VideoFileClip

import warnings
warnings.filterwarnings("ignore")

def record_audio(self):
        #创建一个pyaudio对象
        recordAudio = pyaudio.PyAudio()
        #创建一个输入流stream
        stream = recordAudio.open(format = pyaudio.paInt16,#使用量化位数，16位
                channels = 2,#双声道数
                rate = 44100,#采样频率
                input = True,#确定是输入流
                frames_per_buffer = 128#指定每个数据帧数
                )
        
        waveFile = wave.open(self.audio_filename,'wb')
        #音频通道数
        waveFile.setnchannels(2)
        #量化位数
        waveFile.setsampwidth(recordAudio.get_sample_size(pyaudio.paInt16))
        #采样频率
        waveFile.setframerate(44100)
        #录制，不断写入
        while self.recording:
            waveFile.writeframes(stream.read(128))
        

        waveFile.close()
        #停止输入
        stream.stop_stream()
        #关闭输入流
        stream.close()
        recordAudio.terminate()
        pass

def video_record(self):
        #获取屏幕
        screen = ImageGrab.grab()
        #转换RGB
        video = cv2.VideoWriter(self.video_record_filename,cv2.VideoWriter_fourcc(*'XVID'),20,screen.size)
        #录制
        while self.recording:
            #截取屏幕
            screen = ImageGrab.grab()
            screen = cv2.cvtColor(np.array(screen),cv2.COLOR_RGB2BGR)
            #写入到本地文件
            video.write(screen)
        print(datetime.now())
        #释放
        video.release()
        pass

def run(self):
        print("3秒后开始录制，输入quit结束")
        sleep(3)
        #文件命名为当前时间
        self.now = str(datetime.now())[:19].replace(':','_').replace('-','_').replace(' ','_')
        #音频
        self.audio_filename = "{}.wav".format(self.now)
        #视频
        self.video_record_filename = "{}_video.avi".format(self.now)
        #最终的视频
        self.video_filename = "{}.mp4".format(self.now)
        #开启两个线程
        t1 = threading.Thread(target = self.record_audio)
        t2 = threading.Thread(target = self.video_record)
        
        for t in [t1,t2]:
            t.start()
        while True:
            if(input() == 'quit'):
                break
        #两个线程同时结束
        self.recording = False
        for t in [t1,t2]:
            t.join()
        
        audio = AudioFileClip(self.audio_filename) #moviepy读取视频
        video = VideoFileClip(self.video_record_filename)#moviepy读取音频
        totalVideo = video.set_audio(audio)#音频附加到视频上
        #编码.mp4  帧率25
        totalVideo.write_videofile(self.video_filename,codec = "libx264",fps = 25)
        pass

import time
import threading

def tstart():
    time.sleep(0.5)
    print(" running....")

if __name__ == '__main__':
    #创建线程
    t1 = threading.Thread(target = tstart)
    t2 = threading.Thread(target = tstart)
    #开启线程
    for t in [t1,t2]:
        t.start()
#join的原理就是依次检验线程池中的线程是否结束
#没有结束就阻塞直到线程结束
#如果结束则跳转执行下一个线程的join函数。
    for t in [t1,t2]:
        t.join()
    print("This is main function")