Python-实现录音小程序

import wave

from pyaudio import PyAudio, paInt16

# 参数设定

framerate = 8000

NUM_SAMPLES = 2000

channels = 1

sampwidth = 2

TIME = 2

# 写 Wav 文件

def save_wave_file(filename, data):

    wf = wave.open(filename, 'wb')

    wf.setnchannels(channels)

    wf.setsampwidth(sampwidth)

    wf.setframerate(framerate)

    wf.writeframes(b''.join(data))

    wf.close()

# 录音

def my_record():

    pa = PyAudio()

    stream = pa.open(format=paInt16,

                     channels=1,

                     rate=framerate,

                     input=True,

                     frames_per_buffer=NUM_SAMPLES)

    my_buf = []

    count = 0

    # 控制录音时间

    while count < TIME * 20:

        string_audio_data = stream.read(NUM_SAMPLES)

        my_buf.append(string_audio_data)

    print(my_buf)

    save_wave_file('01.wav', my_buf)

    stream.close()

# 播放音频

chunk = 2014

def play():

    wf = wave.open(r'01.wav', 'rb')

    p = PyAudio()

    stream = p.open(format=p.get_format_from_width(wf.getsampwidth()),

                    channels=wf.getnchannels(),

                    rate=wf.getframerate(),

                    output=True)

    while True:

        data = wf.readframes(chunk)

        if data == '':

            break

        stream.write(data)

    stream.close()

    p.terminate()

if __name__ == '__main__':

    my_record()

    print('over')

    play()