appcat
/
sonist-gtk
1
0
Fork 0
Code Issues Pull Requests Wiki Activity
sonist-gtk/usr/lib/sonist/mpd.py

881 lines
26 KiB
Python
Raw Blame History

# python-mpd2: Python MPD client library
#
# Copyright (C) 2008-2010 J. Alexander Treuman <jat@spatialrift.net>
# Copyright (C) 2012 J. Thalheim <jthalheim@gmail.com>
# Copyright (C) 2016 Robert Niederreiter <rnix@squarewave.at>
#
# python-mpd2 is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# python-mpd2 is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with python-mpd2. If not, see <http://www.gnu.org/licenses/>.
import gi
import re, socket, sys, time
from enum import Enum
gi.require_version('Gtk', '3.0')
from gi.repository import GObject
VERSION = (3, 1, 0)
HELLO_PREFIX = "OK MPD "
ERROR_PREFIX = "ACK "
ERROR_PATTERN = re.compile(r"\[(?P<errno>\d+)@(?P<offset>\d+)\]\s+{(?P<command>\w+)}\s+(?P<msg>.*)")
SUCCESS = "OK"
NEXT = "list_OK"
def escape(text):
return text.replace("\\", "\\\\").replace('"', '\\"')
# MPD Protocol errors as found in CommandError exceptions
# https://github.com/MusicPlayerDaemon/MPD/blob/master/src/protocol/Ack.hxx
class FailureResponseCode(Enum):
NOT_LIST = 1
ARG = 2
PASSWORD = 3
PERMISSION = 4
UNKNOWN = 5
NO_EXIST = 50
PLAYLIST_MAX = 51
SYSTEM = 52
PLAYLIST_LOAD = 53
UPDATE_ALREADY = 54
PLAYER_SYNC = 55
EXIST = 56
class ProtocolError(Exception):
pass
class CommandError(Exception):
def __init__(self, error):
self.errno = None
self.offset = None
self.command = None
self.msg = None
match = ERROR_PATTERN.match(error)
if match:
self.errno = FailureResponseCode(int(match.group("errno")))
self.offset = int(match.group("offset"))
self.command = match.group("command")
self.msg = match.group("msg")
class CommandListError(Exception):
pass
class IteratingError(Exception):
pass
class mpd_commands(object):
"""Decorator for registering MPD commands with it's corresponding result
callback.
"""
def __init__(self, *commands, **kwargs):
self.commands = commands
self.is_direct = kwargs.pop("is_direct", False)
self.is_binary = kwargs.pop("is_binary", False)
if kwargs:
raise AttributeError(
"mpd_commands() got unexpected keyword"
" arguments %s" % ",".join(kwargs)
)
def __call__(self, ob):
ob.mpd_commands = self.commands
ob.mpd_commands_direct = self.is_direct
ob.mpd_commands_binary = self.is_binary
return ob
def mpd_command_provider(cls):
"""Decorator hooking up registered MPD commands to concrete client
implementation.
A class using this decorator must inherit from ``MPDClientBase`` and
implement it's ``add_command`` function.
"""
def collect(cls, callbacks=dict()):
"""Collect MPD command callbacks from given class.
Searches class __dict__ on given class and all it's bases for functions
which have been decorated with @mpd_commands and returns a dict
containing callback name as keys and
(callback, callback implementing class) tuples as values.
"""
for name, ob in cls.__dict__.items():
if hasattr(ob, "mpd_commands") and name not in callbacks:
callbacks[name] = (ob, cls)
for base in cls.__bases__:
callbacks = collect(base, callbacks)
return callbacks
for name, value in collect(cls).items():
callback, from_ = value
for command in callback.mpd_commands:
cls.add_command(command, callback)
return cls
class Noop(object):
"""An instance of this class represents a MPD command callback which
does nothing.
"""
mpd_commands = None
class MPDClientBase():
"""Abstract MPD client.
This class defines a general client contract, provides MPD protocol parsers
and defines all available MPD commands and it's corresponding result
parsing callbacks. There might be the need of overriding some callbacks on
subclasses.
"""
def __init__(self):
self.iterate = False
self._reset()
@classmethod
def add_command(cls, name, callback):
raise NotImplementedError(
"Abstract ``MPDClientBase`` does not implement ``add_command``"
)
def noidle(self):
raise NotImplementedError(
"Abstract ``MPDClientBase`` does not implement ``noidle``"
)
def command_list_ok_begin(self):
raise NotImplementedError(
"Abstract ``MPDClientBase`` does not implement " "``command_list_ok_begin``"
)
def command_list_end(self):
raise NotImplementedError(
"Abstract ``MPDClientBase`` does not implement " "``command_list_end``"
)
def _reset(self):
self.mpd_version = None
self._command_list = None
def _parse_pair(self, line, separator):
if line is None:
return
pair = line.split(separator, 1)
if len(pair) < 2:
raise ProtocolError("Could not parse pair: '{}'".format(line))
return pair
def _parse_pairs(self, lines, separator=": "):
for line in lines:
yield self._parse_pair(line, separator)
def _parse_objects(self, lines, delimiters=[], lookup_delimiter=False):
obj = {}
for key, value in self._parse_pairs(lines):
key = key.lower()
if lookup_delimiter and key not in delimiters:
delimiters = delimiters + [key]
if obj:
if key in delimiters:
if lookup_delimiter:
if key in obj:
yield obj
obj = obj.copy()
while delimiters[-1] != key:
obj.pop(delimiters[-1], None)
delimiters.pop()
else:
yield obj
obj = {}
elif key in obj:
if not isinstance(obj[key], list):
obj[key] = [obj[key], value]
else:
obj[key].append(value)
continue
obj[key] = value
if obj:
yield obj
# Use this instead of _parse_objects whenever the result is returned
# immediately in a command implementation
_parse_objects_direct = _parse_objects
def _parse_raw_stickers(self, lines):
for key, sticker in self._parse_pairs(lines):
value = sticker.split("=", 1)
if len(value) < 2:
raise ProtocolError("Could not parse sticker: {}".format(repr(sticker)))
yield tuple(value)
NOOP = mpd_commands("close", "kill")(Noop())
@mpd_commands("plchangesposid", is_direct=True)
def _parse_changes(self, lines):
return self._parse_objects_direct(lines, ["cpos"])
@mpd_commands("listall", "listallinfo", "listfiles", "lsinfo", is_direct=True)
def _parse_database(self, lines):
return self._parse_objects_direct(lines, ["file", "directory", "playlist"])
@mpd_commands("idle")
def _parse_idle(self, lines):
return self._parse_list(lines)
@mpd_commands("addid", "config", "replay_gain_status", "rescan", "update")
def _parse_item(self, lines):
pairs = list(self._parse_pairs(lines))
if len(pairs) != 1:
return
return pairs[0][1]
@mpd_commands(
"channels", "commands", "listplaylist", "notcommands", "tagtypes", "urlhandlers"
)
def _parse_list(self, lines):
seen = None
for key, value in self._parse_pairs(lines):
if key != seen:
if seen is not None:
raise ProtocolError("Expected key '{}', got '{}'".format(seen, key))
seen = key
yield value
@mpd_commands("list", is_direct=True)
def _parse_list_groups(self, lines):
return self._parse_objects_direct(lines, lookup_delimiter=True)
@mpd_commands("readmessages", is_direct=True)
def _parse_messages(self, lines):
return self._parse_objects_direct(lines, ["channel"])
@mpd_commands("listmounts", is_direct=True)
def _parse_mounts(self, lines):
return self._parse_objects_direct(lines, ["mount"])
@mpd_commands("listneighbors", is_direct=True)
def _parse_neighbors(self, lines):
return self._parse_objects_direct(lines, ["neighbor"])
@mpd_commands("listpartitions", is_direct=True)
def _parse_partitions(self, lines):
return self._parse_objects_direct(lines, ["partition"])
@mpd_commands(
"add",
"addtagid",
"binarylimit",
"clear",
"clearerror",
"cleartagid",
"consume",
"crossfade",
"delete",
"deleteid",
"delpartition",
"disableoutput",
"enableoutput",
"findadd",
"load",
"mixrampdb",
"mixrampdelay",
"mount",
"move",
"moveid",
"moveoutput",
"newpartition",
"next",
"outputvolume",
"partition",
"password",
"pause",
"ping",
"play",
"playid",
"playlistadd",
"playlistclear",
"playlistdelete",
"playlistmove",
"previous",
"prio",
"prioid",
"random",
"rangeid",
"rename",
"repeat",
"replay_gain_mode",
"rm",
"save",
"searchadd",
"searchaddpl",
"seek",
"seekcur",
"seekid",
"sendmessage",
"setvol",
"shuffle",
"single",
"sticker delete",
"sticker set",
"stop",
"subscribe",
"swap",
"swapid",
"toggleoutput",
"unmount",
"unsubscribe",
"volume",
)
def _parse_nothing(self, lines):
for line in lines:
raise ProtocolError(
"Got unexpected return value: '{}'".format(", ".join(lines))
)
@mpd_commands("count", "currentsong", "readcomments", "stats", "status")
def _parse_object(self, lines):
objs = list(self._parse_objects(lines))
if not objs:
return {}
return objs[0]
@mpd_commands("outputs", is_direct=True)
def _parse_outputs(self, lines):
return self._parse_objects_direct(lines, ["outputid"])
@mpd_commands("playlist")
def _parse_playlist(self, lines):
for key, value in self._parse_pairs(lines, ":"):
yield value
@mpd_commands("listplaylists", is_direct=True)
def _parse_playlists(self, lines):
return self._parse_objects_direct(lines, ["playlist"])
@mpd_commands("decoders", is_direct=True)
def _parse_plugins(self, lines):
return self._parse_objects_direct(lines, ["plugin"])
@mpd_commands(
"find",
"listplaylistinfo",
"playlistfind",
"playlistid",
"playlistinfo",
"playlistsearch",
"plchanges",
"search",
"sticker find",
is_direct=True,
)
def _parse_songs(self, lines):
return self._parse_objects_direct(lines, ["file"])
@mpd_commands("sticker get")
def _parse_sticker(self, lines):
key, value = list(self._parse_raw_stickers(lines))[0]
return value
@mpd_commands("sticker list")
def _parse_stickers(self, lines):
return dict(self._parse_raw_stickers(lines))
@mpd_commands("albumart", "readpicture", is_binary=True)
def _parse_plain_binary(self, structure):
return structure
def _create_callback(self, function, wrap_result):
"""Create MPD command related response callback.
"""
if not callable(function):
return None
def command_callback():
# command result callback expects response from MPD as iterable lines,
# thus read available lines from socket
res = function(self, self._read_lines())
# wrap result in iterator helper if desired
if wrap_result:
res = self._wrap_iterator(res)
return res
return command_callback
def _create_command(wrapper, name, return_value, wrap_result):
"""Create MPD command related function.
"""
def mpd_command(self, *args):
callback = _create_callback(self, return_value, wrap_result)
return wrapper(self, name, args, callback)
return mpd_command
@mpd_command_provider
class MPDClient(MPDClientBase, GObject.Object):
__gsignals__ = {
'online': (GObject.SignalFlags.RUN_FIRST, None, ()),
'offline': (GObject.SignalFlags.RUN_FIRST, None, ()),
'playing': (GObject.SignalFlags.RUN_FIRST, None, (object, object)),
'song_changed': (GObject.SignalFlags.RUN_FIRST, None, (object, object)),
'state_changed': (GObject.SignalFlags.RUN_FIRST, None, (object,)),
'volume_changed': (GObject.SignalFlags.RUN_FIRST, None, (int,)),
'error': (GObject.SignalFlags.RUN_FIRST, None, (object,))
}
connected = False
try_conn_timer = None
heart_beat_timer = None
current_song_id = None
current_state = 'stop'
current_volume = 100
idletimeout = None
_timeout = None
_wrap_iterator_parsers = [
MPDClientBase._parse_list,
MPDClientBase._parse_list_groups,
MPDClientBase._parse_playlist,
MPDClientBase._parse_changes,
MPDClientBase._parse_songs,
MPDClientBase._parse_mounts,
MPDClientBase._parse_neighbors,
MPDClientBase._parse_partitions,
MPDClientBase._parse_playlists,
MPDClientBase._parse_database,
MPDClientBase._parse_messages,
MPDClientBase._parse_outputs,
MPDClientBase._parse_plugins,
]
def __init__(self, host = '127.0.0.1', port = 6600):
super(MPDClient, self).__init__()
super(GObject.Object, self).__init__()
self.host = host
self.port = port
def _reset(self):
super(MPDClient, self)._reset()
self.connected = False
self._iterating = False
self._sock = None
self._rbfile = None
self._wfile = None
def _execute(self, command, args, retval):
if self._iterating:
raise IteratingError("Cannot execute '{}' while iterating".format(command))
if self._command_list is not None:
if not callable(retval):
raise CommandListError(
"'{}' not allowed in command list".format(command)
)
self._write_command(command, args)
self._command_list.append(retval)
else:
self._write_command(command, args)
if callable(retval):
return retval()
return retval
def _write_line(self, line):
try:
if self._wfile == None:
print('MPD server is not connected!!!')
else:
self._wfile.write("{}\n".format(line))
self._wfile.flush()
except socket.error as e:
error_message = "Connection to server was reset"
self._reset()
e = ConnectionError(error_message)
raise e.with_traceback(sys.exc_info()[2])
def _write_command(self, command, args=[]):
parts = [command]
for arg in args:
if type(arg) is tuple:
if len(arg) == 0:
parts.append('":"')
elif len(arg) == 1:
parts.append('"{}:"'.format(int(arg[0])))
else:
parts.append('"{}:{}"'.format(int(arg[0]), int(arg[1])))
else:
parts.append('"{}"'.format(escape(str(arg))))
# Minimize logging cost if the logging is not activated.
cmd = " ".join(parts)
self._write_line(cmd)
def _read_line(self):
line = self._rbfile.readline().decode("utf-8")
if not line.endswith("\n"):
self.destroy()
raise ConnectionError("Connection lost while reading line")
line = line.rstrip("\n")
if line.startswith(ERROR_PREFIX):
error = line[len(ERROR_PREFIX) :].strip()
raise CommandError(error)
if self._command_list is not None:
if line == NEXT:
return
if line == SUCCESS:
raise ProtocolError("Got unexpected '{}'".format(SUCCESS))
elif line == SUCCESS:
return
return line
def _read_lines(self):
line = self._read_line()
while line is not None:
yield line
line = self._read_line()
def _read_chunk(self, amount):
chunk = bytearray()
while amount > 0:
result = self._rbfile.read(amount)
if len(result) == 0:
break
chunk.extend(result)
amount -= len(result)
return bytes(chunk)
def _read_binary(self):
"""From the data stream, read Unicode lines until one says "binary:
<number>\\n"; at that point, read binary data of the given length.
This behaves like _parse_objects (with empty set of delimiters; even
returning only a single result), but rather than feeding from a lines
iterable (which would be preprocessed too far), it reads directly off
the stream."""
obj = {}
while True:
line = self._read_line()
if line is None:
break
key, value = self._parse_pair(line, ": ")
if key == "binary":
chunk_size = int(value)
value = self._read_chunk(chunk_size)
if len(value) != chunk_size:
self.destroy()
raise ConnectionError(
"Connection lost while reading binary data: "
"expected %d bytes, got %d" % (chunk_size, len(value))
)
if self._rbfile.read(1) != b"\n":
# newline after binary content
self.destroy()
raise ConnectionError("Connection lost while reading line")
obj[key] = value
return obj
def _execute_binary(self, command, args):
"""Execute a command repeatedly with an additional offset argument,
keeping all the identical returned dictionary items and concatenating
the binary chunks following the binary item into one of exactly size.
This differs from _execute in that rather than passing the lines to the
callback which'd then call on something like _parse_objects, it builds
a parsed object on its own (as a prerequisite to the chunk driving
process) and then joins together the chunks into a single big response."""
if self._iterating or self._command_list is not None:
raise IteratingError("Cannot execute '{}' with command lists".format(command))
data = None
args = list(args)
assert len(args) == 1
args.append(0)
final_metadata = None
while True:
self._write_command(command, args)
metadata = self._read_binary()
chunk = metadata.pop('binary', None)
if final_metadata is None:
data = chunk
final_metadata = metadata
if not data:
break
try:
size = int(final_metadata['size'])
except KeyError:
size = len(chunk)
except ValueError:
raise CommandError("Size data unsuitable for binary transfer")
else:
if metadata != final_metadata:
raise CommandError("Metadata of binary data changed during transfer")
if chunk is None:
raise CommandError("Binary field vanished changed during transfer")
data += chunk
args[-1] = len(data)
if len(data) > size:
raise CommandListError("Binary data announced size exceeded")
elif len(data) == size:
break
if data is not None:
final_metadata['binary'] = data
final_metadata.pop('size', None)
return final_metadata
def _read_command_list(self):
try:
for retval in self._command_list:
yield retval()
finally:
self._command_list = None
self._parse_nothing(self._read_lines())
def _iterator_wrapper(self, iterator):
try:
for item in iterator:
yield item
finally:
self._iterating = False
def _wrap_iterator(self, iterator):
if not self.iterate:
return list(iterator)
self._iterating = True
return self._iterator_wrapper(iterator)
def _check_is_mpd_server(self, line):
if not line.endswith("\n"):
self.destroy()
return
line = line.rstrip("\n")
if not line.startswith(HELLO_PREFIX):
return
self.connected = True
self.mpd_version = line[len(HELLO_PREFIX) :].strip()
def _connect_unix(self):
if not hasattr(socket, "AF_UNIX"):
raise ConnectionError("Unix domain sockets not supported on this platform")
sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
sock.settimeout(self.timeout)
sock.connect(self.host)
return sock
def _connect_tcp(self):
try:
flags = socket.AI_ADDRCONFIG
except AttributeError:
flags = 0
for res in socket.getaddrinfo(
self.host, self.port, socket.AF_UNSPEC, socket.SOCK_STREAM, socket.IPPROTO_TCP, flags
):
af, socktype, proto, canonname, sa = res
sock = None
try:
sock = socket.socket(af, socktype, proto)
sock.setsockopt(socket.IPPROTO_TCP, socket.TCP_NODELAY, 1)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_KEEPALIVE, 1)
sock.settimeout(self.timeout)
sock.connect(sa)
return sock
except Exception as e:
if e.strerror == 'Connection refused':
self.emit('offline')
else:
self.emit('error', e)
if sock is not None:
sock.close()
@mpd_commands("idle")
def _parse_idle(self, lines):
self._sock.settimeout(self.idletimeout)
ret = self._wrap_iterator(self._parse_list(lines))
self._sock.settimeout(self._timeout)
return ret
@property
def timeout(self):
return self._timeout
@timeout.setter
def timeout(self, timeout):
self._timeout = timeout
if self._sock is not None:
self._sock.settimeout(timeout)
def _try_connect(self):
while True:
if self._sock is not None:
break
if self.host.startswith(("/", "\0")):
self._sock = self._connect_unix()
else:
self._sock = self._connect_tcp()
if self._sock is not None:
self._rbfile = self._sock.makefile("rb", newline="\n")
self._wfile = self._sock.makefile("w", encoding="utf-8", newline="\n")
try:
helloline = self._rbfile.readline().decode("utf-8")
self._check_is_mpd_server(helloline)
if self.connected:
break
except Exception as e:
self.connected = False
self.destroy()
time.sleep(2)
def heart_beat(self):
while True:
if self.connected:
try:
status = self.status()
song = self.currentsong()
state = status.get('state')
vol = int(status.get('volume') or 100)
if vol != self.current_volume:
self.emit('volume_changed', vol)
self.current_volume = vol
if state != self.current_state:
self.emit('state_changed', state)
self.current_state = state
if song.get('id') != self.current_song_id:
if self.current_song_id is not None:
self.emit('song_changed', status, song)
self.current_song_id = song.get('id')
if state == 'play':
self.emit('playing', status, song)
time.sleep(0.2)
except Exception as err:
print(err)
else:
break
self.destroy()
self.start()
def start(self):
self._try_connect()
if self.connected:
self.emit('online')
self.heart_beat()
def destroy(self):
if self._rbfile is not None:
self._rbfile.close()
if self._wfile is not None:
self._wfile.close()
if self._sock is not None:
self._sock.close()
self._reset()
def fileno(self):
if self._sock is None:
raise ConnectionError("Not connected")
return self._sock.fileno()
def command_list_ok_begin(self):
if self._command_list is not None:
raise CommandListError("Already in command list")
if self._iterating:
raise IteratingError("Cannot begin command list while iterating")
self._write_command("command_list_ok_begin")
self._command_list = []
def command_list_end(self):
if self._command_list is None:
raise CommandListError("Not in command list")
if self._iterating:
raise IteratingError("Already iterating over a command list")
self._write_command("command_list_end")
return self._wrap_iterator(self._read_command_list())
@classmethod
def add_command(cls, name, callback):
wrap_result = callback in cls._wrap_iterator_parsers
if callback.mpd_commands_binary:
method = lambda self, *args: callback(self, cls._execute_binary(self, name, args))
else:
method = _create_command(cls._execute, name, callback, wrap_result)
# create new mpd commands as function:
escaped_name = name.replace(" ", "_")
setattr(cls, escaped_name, method)
@classmethod
def remove_command(cls, name):
if not hasattr(cls, name):
raise ValueError("Can't remove not existent '{}' command".format(name))
name = name.replace(" ", "_")
delattr(cls, str(name))
# vim: set expandtab shiftwidth=4 softtabstop=4 textwidth=79:
Reference in New Issue View Git Blame Copy Permalink
python + gtk3开发的基于mpd后端的音乐播放器
Releases 2
Python 99.1%
Shell 0.9%