python SDK

This commit is contained in:
Ajay Raj 2023-02-24 10:47:17 -08:00
commit 6dc9fceeb5
18 changed files with 482 additions and 0 deletions

View file

@ -0,0 +1,15 @@
from ..models.audio_encoding import AudioEncoding
class BaseOutputDevice:
def __init__(self, sampling_rate: int, audio_encoding: AudioEncoding):
self.sampling_rate = sampling_rate
self.audio_encoding = audio_encoding
async def send_async(self, chunk):
raise NotImplemented
async def maybe_send_mark_async(self, message):
pass

View file

@ -0,0 +1,28 @@
import pyaudio
from .base_output_device import BaseOutputDevice
from ..models.audio_encoding import AudioEncoding
class SpeakerOutput(BaseOutputDevice):
DEFAULT_SAMPLING_RATE = 44100
def __init__(self, pa: pyaudio.PyAudio, device_info: dict, audio_encoding: AudioEncoding = AudioEncoding.LINEAR16):
self.device_info = device_info
sampling_rate = int(self.device_info.get('defaultSampleRate', self.DEFAULT_SAMPLING_RATE))
super().__init__(sampling_rate, audio_encoding)
self.pa = pa
self.stream = self.pa.open(
output=True,
channels=1,
rate=self.sampling_rate,
format=pyaudio.paInt16,
output_device_index=int(self.device_info['index'])
)
async def send_async(self, chunk):
self.stream.write(chunk)
def terminate(self):
self.stream.close()
self.pa.close()