From 92319638704d5d162e3faebbfbeedd434b12d7f2 Mon Sep 17 00:00:00 2001
From: Nick Pourazima <nick.pourazima@gmail.com>
Date: Fri, 22 Jul 2022 13:24:51 -0400
Subject: [PATCH] Feat: Add fully functional rotary guestbook

---
 .gitignore         |   1 +
 rotaryGuestBook.py | 104 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 105 insertions(+)
 create mode 100644 .gitignore
 create mode 100644 rotaryGuestBook.py

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..d8dd753
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+*.wav
diff --git a/rotaryGuestBook.py b/rotaryGuestBook.py
new file mode 100644
index 0000000..09d068a
--- /dev/null
+++ b/rotaryGuestBook.py
@@ -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()