spotify/spotify.py

import config
import logging
import paho.mqtt.client as paho
import report
import subprocess
import time


try:
    from config import APP_NAME as ROOT_LOGGER_NAME
except ImportError:
    ROOT_LOGGER_NAME = 'root'
logger = logging.getLogger(ROOT_LOGGER_NAME).getChild('main')


class librespot(object):
    ON_CMD = ['play', ]
    OFF_CMD = ['pause', 'stop', ]

    def __init__(self, state_callback):
        logger.info("Starting Librespot...")
        self.__state_callback__ = state_callback
        self.__process__ = subprocess.Popen(["librespot", "-v", "--name", config.DEVICE_NAME],
                               shell=False,
                               # We pipe the output to an internal pipe
                               stderr=subprocess.PIPE)
        self.__state__ = None
        self.set_state(False)

    def run(self):
        while True:
            output = self.__process__.stderr.readline()
            # Polling returns None when the program is still running, return_code otherwise
            return_code = self.__process__.poll()
            if return_code is not None:
                self.__process__.close()
                # Program ended, get exit/return code
                raise RuntimeError("Command '{}' finished with exit code {}".format(command, return_code))
        
                # If the output is not empty, feed it to the function, strip the newline first
            if output:
                out_txt = output.decode('utf-8').strip('\n').strip()
                out_txt = out_txt[out_txt.find(']') + 2:]
                logger.debug("librespot output: %s", out_txt)
                if "command=" in out_txt:
                    command = out_txt[out_txt.find('command=') + 8:].strip().lower()
                    logger.debug("librespot command: %s", command)
                    if command.startswith("load"):
                        self.set_state(command.split(',')[2].strip() == 'true')
                    #
                    if command in self.ON_CMD:
                        self.set_state(True)
                    if command in self.OFF_CMD:
                        self.set_state(False)

    def set_state(self, target_state):
        if target_state != self.__state__:
            self.__state__ = target_state
            logger.info("spotify state changed to %s", self.__state__)
            self.__state_callback__(self.__state__)


def send_msg_mqtt(state):
    client= paho.Client("spotify")
    client.username_pw_set(config.MQTT_USER, config.MQTT_PASS)
    client.connect(config.MQTT_SERVER, 1883)
    logger.info("Sending Spotify status information to mqtt %s = %s", config.MQTT_TOPIC, str(state))
    client.publish(config.MQTT_TOPIC, "true" if state else "false")



if __name__ == '__main__': 
    report.appLoggingConfigure(config.__BASEPATH__, config.LOGTARGET, ((config.APP_NAME, config.LOGLVL), ), fmt=config.formatter, host=config.LOGHOST, port=config.LOGPORT)
    
    ls = librespot(send_msg_mqtt)
    ls.run()