Feat: Add fully functional rotary guestbook

This commit is contained in:
Nick Pourazima 2022-07-22 13:24:51 -04:00
parent df309c4619
commit 9231963870
2 changed files with 105 additions and 0 deletions

1
.gitignore vendored Normal file
View File

@ -0,0 +1 @@
*.wav

104
rotaryGuestBook.py Normal file
View File

@ -0,0 +1,104 @@
#! /usr/bin/env python3
import os
import pyaudio
import time
import wave
from datetime import datetime
from gpiozero import Button
from signal import pause
hook = Button(17)
# TODO: rotary encoder: special key codes trigger certain voicemails
class AudioInterface:
def __init__(self) -> None:
self.audio = pyaudio.PyAudio() # create pyaudio instantiation
self.samp_rate = 44100 # 44.1kHz sampling rate
self.chunk = 4096 # 2^12 samples for buffer
self.chans = 1 # 1 channel
self.format = pyaudio.paInt16 # 16-bit resolution
self.frames = [] # raw data frames recorded from mic
def record(self):
# create pyaudio stream
dev_index = 1 # device index found by p.get_device_info_by_index(ii)
self.stream = self.audio.open(
format=self.format,
rate=self.samp_rate,
channels=self.chans,
input_device_index=dev_index,
input=True,
frames_per_buffer=self.chunk,
)
# loop through stream and append audio chunks to frame array
try:
while hook.is_pressed:
data = self.stream.read(self.chunk, exception_on_overflow = True) # Set to False when live
self.frames.append(data)
except KeyboardInterrupt:
print("Done recording")
except Exception as e:
print(str(e))
def play(self, file):
self.wf = wave.open(file, "rb")
self.stream = self.audio.open(
format=self.audio.get_format_from_width(self.wf.getsampwidth()),
channels=self.wf.getnchannels(),
rate=self.wf.getframerate(),
output=True,
)
""" Play entire file """
data = self.wf.readframes(self.chunk)
while data != b"":
self.stream.write(data)
data = self.wf.readframes(self.chunk)
def stop(self):
# stop the stream, close it, and terminate the pyaudio instantiation
self.stream.stop_stream()
self.stream.close()
self.audio.terminate()
def close(self):
# save the audio frames as .wav file
with wave.open(os.getcwd() + "/recordings/"
+ f"{datetime.now().isoformat()}.wav", "wb") as wavefile:
wavefile.setnchannels(self.chans)
wavefile.setsampwidth(self.audio.get_sample_size(self.format))
wavefile.setframerate(self.samp_rate)
wavefile.writeframes(b"".join(self.frames))
def offHook():
audioInterface = AudioInterface()
print("Phone off hook, ready to begin!")
# wait a second for user to place phone to ear
time.sleep(1)
# playback voice message through speaker
print("Playing voicemail message...")
audioInterface.play(os.getcwd() + "/sounds/voicemail.wav")
# start recording beep
print("Playing beep...")
audioInterface.play(os.getcwd() + "/sounds/beep.wav")
# now, while phone is not off the hook, record audio from the microphone
print("recording")
audioInterface.record()
audioInterface.stop()
audioInterface.close()
print("finished recording")
def onHook():
print("Phone on hook. Sleeping...")
def main():
hook.when_pressed = offHook
hook.when_released = onHook
pause()
if __name__ == "__main__":
main()