Initial commit

1 files changed, 82 insertions, 0 deletions

diff --git a/beetsplug/browse.py b/beetsplug/browse.py
new file mode 100644
index 0000000..72999b9
--- /dev/null
+++ b/beetsplug/browse.py
@@ -0,0 +1,82 @@
+import subprocess
+import uuid
+import webbrowser
+
+from beets.plugins import BeetsPlugin
+from beets.ui import Subcommand, UserError
+
+MUSICBRAINZ_LOOKUP='https://musicbrainz.org/otherlookup/mbid?other-lookup.mbid='
+FIELD_NAMES = ['albumartist', 'album', 'artist', 'releasegroup', 'releasetrack', 'track', 'work']
+
+class BrowsePlugin(BeetsPlugin):
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+    def commands(self):
+        return [BrowseCommand(self.config)]
+
+class BrowseCommand(Subcommand):
+
+    explorer = 'thunar'
+
+    def __init__(self, config):
+        super().__init__('browse', parser=None,
+                help='browse items on MusicBrainz or the file system')
+
+        self.parser.add_option('-f', '--field', type='string',
+                help='which field to look up on MusicBrainz, '
+                'e.g. album, artist, track, work, ...')
+        self.parser.add_option('-o', '--open', action='store_true',
+                help='open in the file browser instead of MusicBrainz')
+        self.parser.add_album_option()
+
+        if 'explorer' in config:
+            self.explorer = config['explorer'].get()
+
+    # pylint: disable=no-self-use
+    def browse_musicbrainz(self, item, field):
+        mbid = item.get(f'mb_{field}id')
+        if not mbid:
+            raise UserError(f'\'mb_{field}id\' not available for: {item}')
+
+        try:
+            uuid.UUID(mbid)
+        except ValueError:
+            raise UserError(f'invalid UUID "{mbid}" for: {item}') from None
+
+        webbrowser.open(MUSICBRAINZ_LOOKUP + mbid)
+
+    def browse_filesystem(self, item, _):
+        try:
+            subprocess.Popen(self.explorer.split(' ') + [item.get('path')]) # pylint: disable=consider-using-with
+        except OSError as err:
+            raise UserError(err) from err
+
+    def func(self, lib, opts, args):
+        queryfun = lib.albums if opts.album else lib.items
+        browsefun = self.browse_filesystem if opts.open else self.browse_musicbrainz
+        field = opts.field or ('album' if opts.album else 'track')
+
+        if field not in FIELD_NAMES:
+            raise UserError(f'invalid field "{field}", try one of: {", ".join(FIELD_NAMES)}')
+
+        if not args:
+            raise UserError('empty query, refusing')
+
+        items = queryfun(args)
+
+        if not items:
+            return
+
+        if len(items) == 1:
+            browsefun(items[0], field)
+            return
+
+        print('Query returned multiple matches, please disambiguate:')
+        for index, item in enumerate(items):
+            if index < 5:
+                print(item)
+            else:
+                print(f'[{len(items) - index} more]')
+                return