squeezebox-notify.py

#! /usr/bin/python
# -*- coding: utf-8 -*-

from telnetlib import Telnet
import getopt, notifications, pynotify, re, sys, urllib

DEFAULT_ICON_LOCATION = '/home/mmm/kode/squeezebox-notify/resources/squeezebox.jpg'

players = {}
notification = pynotify.Notification('Squuezebox Notification')

# TODO: Move the Player class to a separate file.
class Player():
    id = None
    name = None
    fetcher = None

    def __init__(self, id, name, fetcher):
        self.id = id
        self.name = urllib.unquote(name)
        self.fetcher = fetcher

    def get_track_id(self):
        self.fetcher.write('%s status - 1 tags:\n' % player.id)
        return [urllib.unquote(x) for x in self.fetcher.read_until('\n').split(' ') if urllib.unquote(x).startswith('id:')][0].lstrip('id:')

    def get_current_title(self):
        self.fetcher.write('%s current_title ?\n' % player.id)
        return urllib.unquote(self.fetcher
            .read_until('\n')
            .replace('%s current_title ' % player.id, '')
            .rstrip('\n')
        )

def get_player_info(fetcher, player_mac):
    if not player_mac in players.keys():
        fetcher.write('player name %s ?\n' % player_mac)
        name = fetcher.read_until('\n').replace('player name %s ' % player_mac, '').rstrip('\n')
        players[player_mac] = Player(player_mac, name, fetcher)
    return players[player_mac]

def notify(player, cmd, server, port, www_port, username=None, password=None):
    # Only show notifications that we know how to handle.
    known_notifications = {
        'playlist pause' : notifications.pause,
        'playlist newsong' : notifications.new_song,
    }
    for notif in known_notifications.keys():
        if cmd.startswith(notif):
            title, body, icon = known_notifications[notif](player, cmd, server, port, www_port, username, password)
            notification.update(title, body, icon if icon else DEFAULT_ICON_LOCATION)
            notification.show()

def print_help():
    print """
squeezebox-notify

Usage: squeezebox-notify [options] <server>

<server> is IP address or domain name of machine running Logitech Music Server.

Options:

  -s      Server hostname.

  -p      Port number of telnet interface (default is 9090)

  -u      Username for authentication

  -P      Password for authentication

  -w      Port number of web interface (default is 9000)
"""

if __name__ == '__main__':
    # Get arguments and options.
    try:
        opts, args = getopt.getopt(sys.argv[1:], 's:w:p:u:P:')
    except getopt.GetoptError:
        print_help()
        sys.exit(2)

    # Set server telnet port.
    server = None
    port = 9090
    username = None
    password = None
    for opt, arg in opts:
        if opt == '-s':
            if not arg:
                print_help()
                sys.exit(2)
            server = arg
        if opt == '-p':
            try:
                port = int(arg)
            except ValueError:
                print_help()
        if opt == '-u':
            username = arg
        if opt == '-P':
            password = arg

    # Set server web port.
    www_port = 9000
    for opt, arg in opts:
        if opt == '-w':
            try:
                www_port = int(arg)
            except ValueError:
                print_help()

    # Initialise notifications.
    pynotify.init('squeezebox-notify')
    notification = pynotify.Notification('Squeezebox Notify',
        'Connecting to %s' % (server if port == 9090 else ('%s:%i' % (server, port,))),
        '/home/mmm/kode/squeezebox-notify/resources/squeezebox.jpg')
    notification.show()

    # Compile a regex for player related notifications, which are identified by a MAC address and then some.
    player_pattern = re.compile('^(([0-9a-f]{2}%3A){5}[0-9a-f]{2}) (.+)')

    # Start telnet clients (one a as listener, another to fetch info).
    listener = Telnet()
    listener.open(server, port)
    fetcher = Telnet()
    fetcher.open(server, port)
    # Authenticate, if necessary.
    if username and password:
        listener.write('login %s %s\n' % (
            urllib.quote(username),
            urllib.quote(password),
        ))
        listener.read_until('\n')
        fetcher.write('login %s %s\n' % (
            urllib.quote(username),
            urllib.quote(password),
        ))
        fetcher.read_until('\n')
    listener.write('listen 1\n')

    # Start listening for server notifications and process them.
    while True:
        # Get notification data.
        data = listener.read_until('\n')
        # Analyse notification.
        m = player_pattern.match(data)
        # If it looks like a player related notification, break it apart and find out what it's about.
        if m:
            # First, get info about the player.
            player = get_player_info(fetcher, m.group(1))
            # Then, notify.
            notify(player, m.group(3).replace('\n', ''), server, port, www_port)