라즈베리파이 환경에서 버튼을 눌러서 음성을 녹음하고, 파일을 저장할 수 있게 만들게 하다가 그냥 클래스화 시켰다.
(녹음 기능을 자주 우려먹어서..)
근데 이 코드 작성한 걸 다시 보니까 좀 아쉬운 점이 있다. stream 그냥 close하지 말고 with문 써서 close 안 쓸걸 그랬다. ㅜㅜ 그리고 arguments가 너무 드럽게 많은 것 같다. 솔직히 만들 때도 좀 정리하고싶었는데 이건 뭐 정리가.. 안 됨..
이거 쓰실 분들은 그대로 코드 복붙하면 오류날 수가 있다. 마이크가 있는 채널 넘버가 가끔 달라져서 생성자에 있는 channels를 조금 이리 저리 조정해야할 거다.
import pyaudio
import wave
from gpiozero import Button
class Recorder:
def __init__(self, btn_num = 21, chunk = 1024, format = pyaudio.paInt16, channels = 1, filename = 'niniz.wav') -> None:
self.p = pyaudio.PyAudio()
self.chunk = chunk
self.format = format
self.channels = channels
self.rate = int(self.p.get_device_info_by_index(1).get('defaultSampleRate'))
self.filename = filename
self.button = Button(btn_num)
self.frames = []
def recorder(self):
stream = self.p.open(
input_device_index=1,
format = self.format,
channels= self.channels,
rate = self.rate,
input = True,
frames_per_buffer=self.chunk
)
while True:
if self.button.is_pressed:
data = stream.read(self.chunk, exception_on_overflow=False)
self.frames.append(data)
print(". ", end='')
if not self.button.is_pressed:
break
print("Recording is finished.")
stream.stop_stream()
stream.close()
self.p.terminate()
self.write_file()
def write_file(self):
wf = wave.open(self.filename, 'wb')
wf.setnchannels(self.channels)
wf.setsampwidth(self.p.get_sample_size(self.format))
wf.setframerate(self.rate)
wf.writeframes(b''.join(self.frames))
wf.close()
