55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
|
import os
|
||
|
|
|
||
|
|
import pyaudio
|
||
|
|
import wave
|
||
|
|
|
||
|
|
def record(dirname, device_id=0, max_recording_time_s=3):
|
||
|
|
print("###################")
|
||
|
|
fname = os.path.join(dirname, "recorded.wav")
|
||
|
|
|
||
|
|
form_1 = pyaudio.paInt16 # 16-bit resolution
|
||
|
|
chans = 1 # 1 channel
|
||
|
|
samp_rate = 44100 # 44.1kHz sampling rate
|
||
|
|
chunk = 4096 # 2^12 samples for buffer
|
||
|
|
record_secs = max_recording_time_s # seconds to record
|
||
|
|
|
||
|
|
audio = pyaudio.PyAudio() # create pyaudio instantiation
|
||
|
|
|
||
|
|
# create pyaudio stream
|
||
|
|
stream = audio.open(format = form_1,rate = samp_rate,channels = chans, \
|
||
|
|
input_device_index = device_id,input = True, \
|
||
|
|
frames_per_buffer=chunk)
|
||
|
|
print(f"Recording via microphone for {max_recording_time_s} seconds")
|
||
|
|
frames = []
|
||
|
|
|
||
|
|
# loop through stream and append audio chunks to frame array
|
||
|
|
for ii in range(0,int((samp_rate/chunk)*record_secs)):
|
||
|
|
data = stream.read(chunk)
|
||
|
|
frames.append(data)
|
||
|
|
|
||
|
|
print("Finished recording")
|
||
|
|
|
||
|
|
# stop the stream, close it, and terminate the pyaudio instantiation
|
||
|
|
stream.stop_stream()
|
||
|
|
stream.close()
|
||
|
|
audio.terminate()
|
||
|
|
|
||
|
|
# save the audio frames as .wav file
|
||
|
|
wavefile = wave.open(fname,'wb')
|
||
|
|
wavefile.setnchannels(chans)
|
||
|
|
wavefile.setsampwidth(audio.get_sample_size(form_1))
|
||
|
|
wavefile.setframerate(samp_rate)
|
||
|
|
wavefile.writeframes(b''.join(frames))
|
||
|
|
wavefile.close()
|
||
|
|
|
||
|
|
def get_mics():
|
||
|
|
p = pyaudio.PyAudio()
|
||
|
|
for ii in range(p.get_device_count()):
|
||
|
|
device_name = p.get_device_info_by_index(ii).get('name')
|
||
|
|
print(f"{ii}: {device_name}")
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
get_mics()
|
||
|
|
dirname = os.path.dirname(__file__)
|
||
|
|
record(dirname, device_id=0, max_recording_time_s=5)
|