sima/mpdclient.py

   1 # -*- coding: utf-8 -*-
   2 # Copyright (c) 2009-2020 kaliko <kaliko@azylum.org>
   3 #
   4 #  This file is part of sima
   5 #
   6 #  sima is free software: you can redistribute it and/or modify
   7 #  it under the terms of the GNU General Public License as published by
   8 #  the Free Software Foundation, either version 3 of the License, or
   9 #  (at your option) any later version.
  10 #
  11 #  sima is distributed in the hope that it will be useful,
  12 #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 #  GNU General Public License for more details.
  15 #
  16 #  You should have received a copy of the GNU General Public License
  17 #  along with sima.  If not, see <http://www.gnu.org/licenses/>.
  18
  19 # standard library import
  20 from difflib import get_close_matches
  21 from functools import wraps
  22 from itertools import dropwhile
  23
  24 # external module
  25 from musicpd import MPDClient, MPDError
  26
  27
  28 # local import
  29 from .lib.meta import Artist, Album
  30 from .lib.track import Track
  31 from .lib.simastr import SimaStr
  32 from .utils.leven import levenshtein_ratio
  33
  34
  35 class PlayerError(Exception):
  36     """Fatal error in poller."""
  37
  38
  39 # Some decorators
  40 def bl_artist(func):
  41     def wrapper(*args, **kwargs):
  42         cls = args[0]
  43         if not cls.database:
  44             return func(*args, **kwargs)
  45         result = func(*args, **kwargs)
  46         if not result:
  47             return
  48         names = list()
  49         for art in result.names:
  50             if cls.database.get_bl_artist(art, add_not=True):
  51                 cls.log.debug('Blacklisted "%s"', art)
  52                 continue
  53             names.append(art)
  54         if not names:
  55             return
  56         resp = Artist(name=names.pop(), mbid=result.mbid)
  57         for name in names:
  58             resp.add_alias(name)
  59         return resp
  60     return wrapper
  61
  62 def tracks_wrapper(func):
  63     @wraps(func)
  64     def wrapper(*args, **kwargs):
  65         ret = func(*args, **kwargs)
  66         if isinstance(ret, dict):
  67             return Track(**ret)
  68         elif isinstance(ret, list):
  69             return [Track(**t) for t in ret]
  70     return wrapper
  71 # / decorators
  72
  73 def blacklist(artist=False, album=False, track=False):
  74     #pylint: disable=C0111,W0212
  75     field = (album, track)
  76     def decorated(func):
  77         def wrapper(*args, **kwargs):
  78             if not args[0].database:
  79                 return func(*args, **kwargs)
  80             cls = args[0]
  81             boolgen = (bl for bl in field)
  82             bl_fun = (cls.database.get_bl_album,
  83                       cls.database.get_bl_track,)
  84             #bl_getter = next(fn for fn, bl in zip(bl_fun, boolgen) if bl is True)
  85             bl_getter = next(dropwhile(lambda _: not next(boolgen), bl_fun))
  86             #cls.log.debug('using {0} as bl filter'.format(bl_getter.__name__))
  87             results = list()
  88             for elem in func(*args, **kwargs):
  89                 if bl_getter(elem, add_not=True):
  90                     #cls.log.debug('Blacklisted "{0}"'.format(elem))
  91                     continue
  92                 if track and cls.database.get_bl_album(elem, add_not=True):
  93                     # filter album as well in track mode
  94                     # (artist have already been)
  95                     cls.log.debug('Blacklisted alb. "{0.album}"'.format(elem))
  96                     continue
  97                 results.append(elem)
  98             return results
  99         return wrapper
 100     return decorated
 101
 102
 103 class MPD(MPDClient):
 104     """
 105     Player instance inheriting from MPDClient (python-musicpd).
 106
 107     Some methods are overridden to format objects as sima.lib.Track for
 108     instance, other are calling parent class directly through super().
 109     cf. MPD.__getattr__
 110
 111     .. note::
 112
 113         * find methods are looking for exact match of the object provided attributes in MPD music library
 114         * search methods are looking for exact match + fuzzy match.
 115     """
 116     needed_cmds = ['status', 'stats', 'add', 'find',
 117                    'search', 'currentsong', 'ping']
 118     database = None
 119
 120     def __init__(self, daemon):
 121         super().__init__()
 122         self.use_mbid = True
 123         self.daemon = daemon
 124         self.log = daemon.log
 125         self.config = self.daemon.config['MPD']
 126         self._cache = None
 127
 128     # ######### Overriding MPDClient ###########
 129     def __getattr__(self, cmd):
 130         """Wrapper around MPDClient calls for abstract overriding"""
 131         track_wrapped = {
 132                          'currentsong',
 133                          'find',
 134                          'playlistinfo',
 135                          }
 136         if cmd in track_wrapped:
 137             return tracks_wrapper(super().__getattr__(cmd))
 138         return super().__getattr__(cmd)
 139
 140     def disconnect(self):
 141         """Overriding explicitly MPDClient.disconnect()"""
 142         if self._sock:
 143             super().disconnect()
 144
 145     def connect(self):
 146         """Overriding explicitly MPDClient.connect()"""
 147         # host, port, password
 148         host = self.config.get('host')
 149         port = self.config.get('port')
 150         password = self.config.get('password', fallback=None)
 151         self.disconnect()
 152         try:
 153             super().connect(host, port)
 154         # Catch socket errors
 155         except IOError as err:
 156             raise PlayerError('Could not connect to "%s:%s": %s' %
 157                               (host, port, err.strerror))
 158         # Catch all other possible errors
 159         # ConnectionError and ProtocolError are always fatal.  Others may not
 160         # be, but we don't know how to handle them here, so treat them as if
 161         # they are instead of ignoring them.
 162         except MPDError as err:
 163             raise PlayerError('Could not connect to "%s:%s": %s' %
 164                               (host, port, err))
 165         if password:
 166             try:
 167                 self.password(password)
 168             except (MPDError, IOError) as err:
 169                 raise PlayerError("Could not connect to '%s': %s", (host, err))
 170         # Controls we have sufficient rights
 171         available_cmd = self.commands()
 172         for cmd in MPD.needed_cmds:
 173             if cmd not in available_cmd:
 174                 self.disconnect()
 175                 raise PlayerError('Could connect to "%s", '
 176                                   'but command "%s" not available' %
 177                                   (host, cmd))
 178         # Controls use of MusicBrainzIdentifier
 179         # TODO: Use config instead of Artist object attibute?
 180         if self.use_mbid:
 181             tt = self.tagtypes()
 182             if 'MUSICBRAINZ_ARTISTID' not in tt:
 183                 self.log.warning('Use of MusicBrainzIdentifier is set but MPD is '
 184                                  'not providing related metadata')
 185                 self.log.info(tt)
 186                 self.log.warning('Disabling MusicBrainzIdentifier')
 187                 self.use_mbid = False
 188             else:
 189                 self.log.debug('Available metadata: %s', tt)  # pylint: disable=no-member
 190         else:
 191             self.log.warning('Use of MusicBrainzIdentifier disabled!')
 192             self.log.info('Consider using MusicBrainzIdentifier for your music library')
 193         self._reset_cache()
 194     # ######### / Overriding MPDClient #########
 195
 196     def _reset_cache(self):
 197         """
 198         Both flushes and instantiates _cache
 199         """
 200         if isinstance(self._cache, dict):
 201             self.log.info('Player: Flushing cache!')
 202         else:
 203             self.log.info('Player: Initialising cache!')
 204         self._cache = {'artists': frozenset(),
 205                        'nombid_artists': frozenset()}
 206         self._cache['artists'] = frozenset(filter(None, self.list('artist')))
 207         if Artist.use_mbid:
 208             self._cache['nombid_artists'] = frozenset(filter(None, self.list('artist', 'musicbrainz_artistid', '')))
 209
 210     def _skipped_track(self, previous):
 211         if (self.state == 'stop'
 212                 or not hasattr(previous, 'id')
 213                 or not hasattr(self.current, 'id')):
 214             return False
 215         return self.current.id != previous.id  # pylint: disable=no-member
 216
 217     def monitor(self):
 218         """OLD socket Idler
 219         Monitor player for change
 220         Returns a list a events among:
 221
 222             * database  player media library has changed
 223             * playlist  playlist modified
 224             * options   player options changed: repeat mode, etc…
 225             * player    player state changed: paused, stopped, skip track…
 226             * skipped   current track skipped
 227         """
 228         curr = self.current
 229         try:
 230             ret = self.idle('database', 'playlist', 'player', 'options')
 231         except (MPDError, IOError) as err:
 232             raise PlayerError("Couldn't init idle: %s" % err)
 233         if self._skipped_track(curr):
 234             ret.append('skipped')
 235         if 'database' in ret:
 236             self._flush_cache()
 237         return ret
 238
 239     def clean(self):
 240         """Clean blocking event (idle) and pending commands
 241         """
 242         if 'idle' in self._pending:
 243             self.noidle()
 244         elif self._pending:
 245             self.log.warning('pending commands: %s', self._pending)
 246
 247     def add(self, payload):
 248         """Overriding MPD's add method to accept Track objects"""
 249         if isinstance(payload, Track):
 250             super().__getattr__('add')(payload.file)
 251         elif isinstance(payload, list):
 252             for tr in payload:  # TODO: use send command here
 253                 self.add(tr)
 254         else:
 255             self.log.error('Cannot add %s', payload)
 256
 257     # ######### Properties #####################
 258     @property
 259     def current(self):
 260         return self.currentsong()
 261
 262     @property
 263     def playlist(self):
 264         """
 265         Override deprecated MPD playlist command
 266         """
 267         return self.playlistinfo()
 268
 269     @property
 270     def playmode(self):
 271         plm = {'repeat': None, 'single': None,
 272                'random': None, 'consume': None, }
 273         for key, val in self.status().items():
 274             if key in plm.keys():
 275                 plm.update({key: bool(int(val))})
 276         return plm
 277
 278     @property
 279     def queue(self):
 280         plst = self.playlist
 281         curr_position = int(self.current.pos)
 282         plst.reverse()
 283         return [trk for trk in plst if int(trk.pos) > curr_position]
 284
 285     @property
 286     def state(self):
 287         """Returns (play|stop|pause)"""
 288         return str(self.status().get('state'))
 289     # ######### / Properties ###################
 290
 291 # #### find_tracks ####
 292     def find_tracks(self, what):
 293         """Find tracks for a specific artist or album
 294             >>> player.find_tracks(Artist('Nirvana'))
 295             >>> player.find_tracks(Album('In Utero', artist=(Artist('Nirvana'))
 296
 297         :param Artist,Album what: Artist or Album to fetch track from
 298
 299         Returns a list of :py:obj:Track objects
 300         """
 301         if isinstance(what, Artist):
 302             return self._find_art(what)
 303         elif isinstance(what, Album):
 304             return self._find_alb(what)
 305         elif isinstance(what, str):
 306             return self.find_tracks(Artist(name=what))
 307
 308     def _find_art(self, artist):
 309         tracks = set()
 310         if self.use_mbid and artist.mbid:
 311             tracks |= set(self.find('musicbrainz_artistid', artist.mbid))
 312         for name in artist.names:
 313             tracks |= set(self.find('artist', name))
 314         return list(tracks)
 315
 316     def _find_alb(self, album):
 317         if album.mbid and self.use_mbid:
 318             filt = f'(MUSICBRAINZ_ALBUMID == {album.mbid})'
 319             return self.find(filt)
 320         # Now look for album with no MusicBrainzIdentifier
 321         if album.artist.mbid and self.use_mbid:  # Use album artist MBID if possible
 322             filt = f"((MUSICBRAINZ_ALBUMARTISTID == '{album.artist.mbid}') AND (album == '{album!s}'))"
 323             return self.find(filt)
 324         tracks = []
 325         # Falls back to albumartist/album name
 326         filt = f"((albumartist == '{album.artist!s}') AND (album == '{album!s}'))"
 327         tracks = self.find(filt)
 328         # Falls back to artist/album name
 329         if not tracks:
 330             filt = f"((artist == '{album.artist!s}') AND (album == '{album!s}'))"
 331             tracks = self.find(filt)
 332         return tracks
 333 # #### / find_tracks ##
 334
 335 # #### Search Methods #####
 336     @bl_artist
 337     def search_artist(self, artist):
 338         """
 339         Search artists based on a fuzzy search in the media library
 340             >>> art = Artist(name='the beatles', mbid=<UUID4>) # mbid optional
 341             >>> bea = player.search_artist(art)c
 342             >>> print(bea.names)
 343             >>> ['The Beatles', 'Beatles', 'the beatles']
 344
 345         Returns an Artist object
 346         """
 347         found = False
 348         if self.use_mbid and artist.mbid:
 349             # look for exact search w/ musicbrainz_artistid
 350             exact_m = self.list('artist', f"(MUSICBRAINZ_ARTISTID == '{artist.mbid}')")
 351             if exact_m:
 352                 _ = [artist.add_alias(name) for name in exact_m]
 353                 found = True
 354         # then complete with fuzzy search on artist with no musicbrainz_artistid
 355         if artist.mbid:
 356             # we already performed a lookup on artists with mbid set
 357             # search through remaining artists
 358             artists = self._cache.get('nombid_artists')
 359         else:
 360             artists = self._cache.get('artists')
 361         match = get_close_matches(artist.name, artists, 50, 0.73)
 362         if not match and not found:
 363             return None
 364         if len(match) > 1:
 365             self.log.debug('found close match for "%s": %s', artist, '/'.join(match))
 366         # Does not perform fuzzy matching on short and single word strings
 367         # Only lowercased comparison
 368         if ' ' not in artist.name and len(artist.name) < 8:
 369             for close_art in match:
 370                 # Regular lowered string comparison
 371                 if artist.name.lower() == close_art.lower():
 372                     artist.add_alias(close_art)
 373                     return artist
 374                 else:
 375                     return None
 376         for fuzz_art in match:
 377             # Regular lowered string comparison
 378             if artist.name.lower() == fuzz_art.lower():
 379                 found = True
 380                 artist.add_alias(fuzz_art)
 381                 if artist.name != fuzz_art:
 382                     self.log.debug('"%s" matches "%s".', fuzz_art, artist)
 383                 continue
 384             # SimaStr string __eq__ (not regular string comparison here)
 385             if SimaStr(artist.name) == fuzz_art:
 386                 found = True
 387                 artist.add_alias(fuzz_art)
 388                 self.log.info('"%s" quite probably matches "%s" (SimaStr)',
 389                               fuzz_art, artist)
 390         if found:
 391             if artist.aliases:
 392                 self.log.debug('Found: %s', '/'.join(list(artist.names)[:4]))
 393             return artist
 394
 395     @blacklist(track=True)
 396     def search_track(self, artist, title):
 397         """Fuzzy search of title by an artist
 398         """
 399         # Retrieve all tracks from artist
 400         all_tracks = self.find_tracks(artist)
 401         # Get all titles (filter missing titles set to 'None')
 402         all_artist_titles = frozenset([tr.title for tr in all_tracks
 403                                        if tr.title is not None])
 404         match = get_close_matches(title, all_artist_titles, 50, 0.78)
 405         if not match:
 406             return []
 407         for mtitle in match:
 408             leven = levenshtein_ratio(title.lower(), mtitle.lower())
 409             if leven == 1:
 410                 pass
 411             elif leven >= 0.79:  # PARAM
 412                 self.log.debug('title: "%s" should match "%s" (lr=%1.3f)',
 413                                mtitle, title, leven)
 414             else:
 415                 self.log.debug('title: "%s" does not match "%s" (lr=%1.3f)',
 416                                mtitle, title, leven)
 417                 return []
 418             return self.find('artist', artist, 'title', mtitle)
 419
 420     @blacklist(album=True)
 421     def search_albums(self, artist):
 422         """
 423         Fetch all albums for "AlbumArtist"  == artist
 424         Then look for albums for "artist" == artist and try to filters
 425         multi-artists albums
 426
 427         NB: Running "client.list('album', 'artist', name)" MPD returns any album
 428             containing at least a track with "artist" == name
 429         TODO: Use MusicBrainzID here cf. #30 @gitlab
 430         """
 431         albums = []
 432         for name in artist.names:
 433             if artist.aliases:
 434                 self.log.debug('Searching album for aliase: "%s"', name)
 435             kwalbart = {'albumartist': name, 'artist': name}
 436             for album in self.list('album', 'albumartist', name):
 437                 if album and album not in albums:
 438                     albums.append(Album(name=album, **kwalbart))
 439             for album in self.list('album', 'artist', name):
 440                 album_trks = self.find_tracks(Album(name=album, artist=Artist(name=name)))
 441                 if 'Various Artists' in [tr.albumartist for tr in album_trks]:
 442                     self.log.debug('Discarding %s ("Various Artists" set)', album)
 443                     continue
 444                 arts = {trk.artist for trk in album_trks}
 445                 # Avoid selecting album where artist is credited for a single
 446                 # track of the album
 447                 if len(set(arts)) < 2:  # TODO: better heuristic, use a ratio instead
 448                     if album not in albums:
 449                         albums.append(Album(name=album, **kwalbart))
 450                 elif album and album not in albums:
 451                     self.log.debug('"{0}" probably not an album of "{1}"'.format(
 452                         album, artist) + '({0})'.format('/'.join(arts)))
 453         return albums
 454 # #### / Search Methods ###
 455
 456 # VIM MODLINE
 457 # vim: ai ts=4 sw=4 sts=4 expandtab