SwitchAudioSource.py

from subprocess import check_output, call
from json import dumps
from sys import stdout
from os import environ


PATH_TO_SWITCH_AUDIO_OUTPUT = environ['SWITCH_AUDIO_SOURCE_PATH']
LOOKUP_WARNING = "Error: Could not find SwitchAudioSource"


class AudioSource:
    def __init__(self, description):
        words = description.split(' ')
        output = words.pop(-1)
        title = ' '.join(words)

        self.uid = title
        self.title = title
        self.arg = title
        self.autocomplete = title

        self.output = output.find('output') > -1
        self.input = not self.output

    def __str__(self):
        return str(self.__dict__)


def get_sources():
    command_output = check_output([
        PATH_TO_SWITCH_AUDIO_OUTPUT, '-a'
    ])

    return map(lambda line: AudioSource(line), command_output.splitlines())


def set_output(device):
    call([
        PATH_TO_SWITCH_AUDIO_OUTPUT, '-s', device
    ])


def no_path_provided():
    stdout.write(dumps({
        "items": [{
            "title": LOOKUP_WARNING
        }]
    }))
    exit()