"""
本地录音，保存为wav格式，存储在data/speaker_wav目录下
"""
import pyaudio
import wave

def record_audio(filename, duration=5, format=pyaudio.paInt16, channels=1, rate=16000):
    """
    本地录音，保存为wav格式，存储在data/speaker_wav目录下
    """
    p = pyaudio.PyAudio()
    stream = p.open(format=format, channels=channels, rate=rate, input=True, frames_per_buffer=1024)
    
    print("按下回车键开始录音...")
    input()
    frames = []
    for i in range(0, int(rate / 1024 * duration)):
        data = stream.read(1024)
        frames.append(data)
    print("录音结束")
    stream.stop_stream()
    stream.close()
    p.terminate()
    wav_file = wave.open(filename, 'wb')
    wav_file.setnchannels(channels)
    wav_file.setsampwidth(p.get_sample_size(format))
    wav_file.setframerate(rate)
    wav_file.writeframes(b''.join(frames))
    wav_file.close()

if __name__ == "__main__":
    record_audio(
        "data/speaker_wav/test.wav",
        duration=5
    )