import pyaudio
import wave
import keyboard

CHUNK = 8000  # 每次读取的音频数据块的大小为 CHUNK 字节
FORMAT = pyaudio.paInt16  # 音频数据的格式为 16 位整数
CHANNELS = 1  # 音频的通道数为1~单声道，2~立体声。
RATE = 16000  # 采样率
WAVE_OUTPUT_FILENAME = "Oldboy1.pcm"  # 音频保存的文件名，wav格式也可以

p = pyaudio.PyAudio()

stream = p.open(format=FORMAT,
                channels=CHANNELS,
                rate=RATE,
                input=True,
                frames_per_buffer=CHUNK)

print("按下空格键开始录音...")
keyboard.wait('space')  # 等待用户按下空格键

print("开始录音,请说话......")

frames = []

while not keyboard.is_pressed('q'):  # 当用户按下q键时停止录音
    data = stream.read(CHUNK)
    frames.append(data)

print("录音结束,请闭嘴!")

stream.stop_stream()
stream.close()
p.terminate()

wf = wave.open(WAVE_OUTPUT_FILENAME, 'wb')
wf.setnchannels(CHANNELS)
wf.setsampwidth(p.get_sample_size(FORMAT))
wf.setframerate(RATE)
wf.writeframes(b''.join(frames))
wf.close()