From 325ae6f0eea3c69a8f7a0cd394375eeb8a7d4928 Mon Sep 17 00:00:00 2001
From: Owen Parry <oparry@twitter.com>
Date: Mon, 27 Nov 2017 07:55:38 -0800
Subject: [PATCH] Add tags configuration to acousticbrainz plugin

---
 beetsplug/acousticbrainz.py     | 18 +++++++++++++-----
 docs/plugins/acousticbrainz.rst |  4 +++-
 2 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/beetsplug/acousticbrainz.py b/beetsplug/acousticbrainz.py
index 4291d9117..cee765bbb 100644
--- a/beetsplug/acousticbrainz.py
+++ b/beetsplug/acousticbrainz.py
@@ -110,6 +110,7 @@ class AcousticPlugin(plugins.BeetsPlugin):
         self.config.add({
             'auto': True,
             'force': False,
+            'tags': []
         })
 
         if self.config['auto']:
@@ -164,6 +165,7 @@ class AcousticPlugin(plugins.BeetsPlugin):
     def _fetch_info(self, items, write, force):
         """Fetch additional information from AcousticBrainz for the `item`s.
         """
+        tags = self.config['tags'].get()
         for item in items:
             # If we're not forcing re-downloading for all tracks, check
             # whether the data is already present. We use one
@@ -183,11 +185,17 @@ class AcousticPlugin(plugins.BeetsPlugin):
             data = self._get_data(item.mb_trackid)
             if data:
                 for attr, val in self._map_data_to_scheme(data, ABSCHEME):
-                    self._log.debug(u'attribute {} of {} set to {}',
-                                    attr,
-                                    item,
-                                    val)
-                    setattr(item, attr, val)
+                    if not tags or attr in tags:
+                        self._log.debug(u'attribute {} of {} set to {}',
+                                        attr,
+                                        item,
+                                        val)
+                        setattr(item, attr, val)
+                    else:
+                        self._log.debug(u'skipping attribute {} of {} (value {}) due to config',
+                                        attr,
+                                        item,
+                                        val)
                 item.store()
                 if write:
                     item.try_write()
diff --git a/docs/plugins/acousticbrainz.rst b/docs/plugins/acousticbrainz.rst
index bf2102790..5bd514c64 100644
--- a/docs/plugins/acousticbrainz.rst
+++ b/docs/plugins/acousticbrainz.rst
@@ -54,10 +54,12 @@ Configuration
 -------------
 
 To configure the plugin, make a ``acousticbrainz:`` section in your
-configuration file. There is one option:
+configuration file. There are three options:
 
 - **auto**: Enable AcousticBrainz during ``beet import``.
   Default: ``yes``.
 - **force**: Download AcousticBrainz data even for tracks that already have
   it.
   Default: ``no``.
+- **tags**: Which tags from the list above to set on your files.
+  Default: [] (all)