Add two new plugin events for recommendation and assign_items

beets/autotag/match.py

@@ -142,6 +142,21 @@
         for iidx, t in zip(assigned_item_idxs, tracks)
         if iidx != -1
     }
+
+    # allow plugins to modify the mapping
+    new_assign_list = plugins.send(
+        "assign_items", mapping=mapping, org_items=items, org_tracks=tracks
+    )
+
+    # get the first plugin provided mapping that is a dict and not empty
+    new_mapping = next(
+        (item for item in new_assign_list if isinstance(item, dict) and item),
+        None,
+    )
+    if new_mapping:
+        # if a plugin provided a mapping, use that instead of the default one
+        mapping = new_mapping
+
     extra_items = list(set(items) - mapping.keys())
     extra_items.sort(key=lambda i: (i.disc, i.track, i.title))
     extra_tracks = list(set(tracks) - set(mapping.values()))
@@ -328,7 +343,7 @@
         dist.add("unmatched_tracks", 1.0)
 
     # Plugins.
     dist.update(plugins.album_distance(items, album_info, mapping))
 
     return dist
 
@@ -390,6 +405,15 @@
         rec = Recommendation.low
     else:
         # No conclusion. Return immediately. Can't be downgraded any further.
+        # but allow plugins to modify the recommendation
+        new_rec_list = plugins.send(
+            "recommendation", results=results, org_rec=Recommendation.none
+        )
+        # take the first recommendation from a plugin
+        if len(new_rec_list) > 0 and isinstance(
+            new_rec_list[0], Recommendation
+        ):
+            return new_rec_list[0]
         return Recommendation.none
 
     # Downgrade to the max rec if it is lower than the current rec for an
@@ -411,6 +435,11 @@
             )
             rec = min(rec, max_rec)
 
+    # allow plugins to modify the recommendation
+    new_rec_list = plugins.send("recommendation", results=results, org_rec=rec)
+    # take the first recommendation from a plugin
+    if len(new_rec_list) > 0 and isinstance(new_rec_list[0], Recommendation):
+        rec = new_rec_list[0]
     return rec
 
 
@@ -555,9 +584,9 @@
 
         # Get the results from the data sources.
         for matched_candidate in hooks.album_candidates(
             items, search_artist, search_album, va_likely, extra_tags
         ):
             _add_candidate(items, candidates, matched_candidate)
 
     log.debug("Evaluating {0} candidates.", len(candidates))
     # Sort and get the recommendation.
@@ -616,9 +645,9 @@
     log.debug("Item search terms: {0} - {1}", search_artist, search_title)
 
     # Get and evaluate candidate metadata.
     for track_info in hooks.item_candidates(item, search_artist, search_title):
         dist = track_distance(item, track_info, incl_artist=True)
         candidates[track_info.track_id] = hooks.TrackMatch(dist, track_info)
 
     # Sort by distance and return with recommendation.
     log.debug("Found {0} candidates.", len(candidates))