acousticbrainz: handle query args depending on request action

metabrainz · Jul 2, 2021 · 44680d2 · 44680d2
1 parent 1dee55a
commit 44680d2
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/plugins/acousticbrainz/__init__.py b/plugins/acousticbrainz/__init__.py
@@ -310,20 +310,24 @@ def _batch(self, action, recording_ids, callback, result, response=None, reply=N
             callback=partial(self._batch, action, recording_ids, callback, result))
 
     def _do_request(self, action, recording_ids, callback):
-        queryargs = {
-            'recording_ids': ';'.join(recording_ids),
-            'map_classes': 'true'
-        }
         self.album.tagger.webservice.get(
             ACOUSTICBRAINZ_HOST,
             ACOUSTICBRAINZ_PORT,
             '/api/v1/%s' % action,
             callback,
             priority=True,
             parse_response_type='json',
-            queryargs=queryargs
+            queryargs=self._get_query_args(action, recording_ids)
         )
 
+    def _get_query_args(self, action, recording_ids):
+        queryargs = {
+            'recording_ids': ';'.join(recording_ids),
+        }
+        if action == 'high-level':
+            queryargs['map_classes'] = 'true'
+        return queryargs
+
     def _merge_results(self, full, new):
         mapping = new.get('mbid_mapping', {})
         new = {mapping.get(k, k): v for (k, v) in new.items() if k != 'mbid_mapping'}