python record audio from microphone

1import pyaudio
2import wave
3
4CHUNK = 1024
5FORMAT = pyaudio.paInt16
6CHANNELS = 2
7RATE = 44100
8RECORD_SECONDS = 5
9WAVE_OUTPUT_FILENAME = "output.wav"
10
11p = pyaudio.PyAudio()
12
13stream = p.open(format=FORMAT,
14                channels=CHANNELS,
15                rate=RATE,
16                input=True,
17                frames_per_buffer=CHUNK)
18
19print("* recording")
20
21frames = []
22
23for i in range(0, int(RATE / CHUNK * RECORD_SECONDS)):
24    data = stream.read(CHUNK)
25    frames.append(data)
26
27print("* done recording")
28
29stream.stop_stream()
30stream.close()
31p.terminate()
32
33wf = wave.open(WAVE_OUTPUT_FILENAME, 'wb')
34wf.setnchannels(CHANNELS)
35wf.setsampwidth(p.get_sample_size(FORMAT))
36wf.setframerate(RATE)
37wf.writeframes(b''.join(frames))
38wf.close()