python实现屏幕录制。(声音内外录制都可)

实现屏幕录制

import wave
import os
import pyaudio
from PIL import ImageGrab
import numpy as np
import cv2
from moviepy.editor import *
from moviepy.audio.fx import all
from pynput import keyboard
import time
CHUNK = 1024
FORMAT = pyaudio.paInt16
CHANNELS = 2
RATE = 44100
WAVE_OUTPUT_FILENAME = "output.wav"
p = pyaudio.PyAudio()
wf = wave.open(WAVE_OUTPUT_FILENAME, 'wb')
wf.setnchannels(CHANNELS)
wf.setsampwidth(p.get_sample_size(FORMAT))
wf.setframerate(RATE)
audio_record_flag = True
key=0
def callback(in_data, frame_count, time_info, status):
    global key
    if(key==1):
        wf.writeframes(in_data)
    if audio_record_flag:
        return (in_data, pyaudio.paContinue)
    else:
        return (in_data, pyaudio.paComplete)
target = '立体声混音'
for i in range(p.get_device_count()):
        devInfo = p.get_device_info_by_index(i)
        if devInfo['name'].find(target) >= 0 and devInfo['hostApi'] == 0:
#改变hostApi的数值采用内/外录制的方法。可以通过增加函数输出设备名称等信息。
            break
stream = p.open(input_device_index=i,
                format=p.get_format_from_width(wf.getsampwidth()),
                channels=wf.getnchannels(),
                rate=wf.getframerate(),
                input=True,
                stream_callback=callback)
image = ImageGrab.grab()#获得当前屏幕
width = image.size[0]
height = image.size[1]
print("width:", width, "height:", height)
print("image mode:",image.mode)
k=np.zeros((width,height),np.uint8)
fourcc = cv2.VideoWriter_fourcc(*'mp4v')#编码格式
video = cv2.VideoWriter('test.mp4', fourcc, 11.8, (width, height))
#若设置帧率与实际帧率不一致，会导致视频时间与音频时间不一致,调试最好的帧率。
print("video recording!!!!!")
begain_time=time.time()
stream.start_stream()
print("audio recording!!!!!")
record_count = 0
time.sleep(3)
while True:
    img_rgb = ImageGrab.grab()
    img_bgr=cv2.cvtColor(np.array(img_rgb), cv2.COLOR_RGB2BGR)#转为opencv的BGR格式
    key=1
    video.write(img_bgr)
    record_count += 1
    if(record_count > 200):
        audio_record_flag = False
        video.release()
        break
    ti=time.time()-begain_time
    print(record_count)
cv2.destroyAllWindows()
stream.stop_stream()
stream.close()
wf.close()
p.terminate()
audioclip = AudioFileClip("output.wav")
videoclip = VideoFileClip("test.mp4")
videoclip2 = videoclip.set_audio(audioclip)
video = CompositeVideoClip([videoclip2])
video.write_videofile("test2.mp4",codec='libx264')

具体时间等参数自己通过对代码的修改来实现，也可以直接设计一个GUI来封装成一个软件实现随意修改。GUI设计我之前过一个类似的。。。