mirror of
https://github.com/beetbox/beets.git
synced 2026-01-05 23:43:31 +01:00
Merge pull request #1787 from opatel99/AcousticBrainzP2
AcousticBrainz: More features (autotag)
This commit is contained in:
Adrian Sampson
commit
b44601176a
3 changed files with 126 additions and 36 deletions
|
|
@ -186,6 +186,7 @@ class MusicalKey(types.String):
|
|||
for flat, sharp in self.ENHARMONIC.items():
|
||||
key = re.sub(flat, sharp, key)
|
||||
key = re.sub(r'[\W\s]+minor', 'm', key)
|
||||
key = re.sub(r'[\W\s]+major', '', key)
|
||||
return key.capitalize()
|
||||
|
||||
def normalize(self, key):
|
||||
|
|
|
|||
|
|
@ -13,23 +13,29 @@
|
|||
# The above copyright notice and this permission notice shall be
|
||||
# included in all copies or substantial portions of the Software.
|
||||
|
||||
""" Fetch various AcousticBrainz metadata using MBID
|
||||
"""Fetch various AcousticBrainz metadata using MBID.
|
||||
"""
|
||||
from __future__ import (division, absolute_import, print_function,
|
||||
unicode_literals)
|
||||
|
||||
import requests
|
||||
import operator
|
||||
|
||||
from beets import plugins, ui
|
||||
|
||||
ACOUSTIC_URL = "http://acousticbrainz.org/"
|
||||
LEVEL = "/high-level"
|
||||
ACOUSTIC_BASE = "http://acousticbrainz.org/"
|
||||
LEVELS = ["/low-level", "/high-level"]
|
||||
|
||||
|
||||
class AcousticPlugin(plugins.BeetsPlugin):
|
||||
def __init__(self):
|
||||
super(AcousticPlugin, self).__init__()
|
||||
|
||||
self.config.add({'auto': True})
|
||||
if self.config['auto']:
|
||||
self.register_listener('import_task_files',
|
||||
self.import_task_files)
|
||||
|
||||
def commands(self):
|
||||
cmd = ui.Subcommand('acousticbrainz',
|
||||
help="fetch metadata from AcousticBrainz")
|
||||
|
|
@ -41,49 +47,110 @@ class AcousticPlugin(plugins.BeetsPlugin):
|
|||
cmd.func = func
|
||||
return [cmd]
|
||||
|
||||
def import_task_files(self, session, task):
|
||||
"""Function is called upon beet import.
|
||||
"""
|
||||
|
||||
items = task.imported_items()
|
||||
fetch_info(self._log, items)
|
||||
|
||||
|
||||
def fetch_info(log, items):
|
||||
"""Currently outputs MBID and corresponding request status code
|
||||
"""Currently outputs MBID and corresponding request status code.
|
||||
"""
|
||||
|
||||
def get_value(*map_path):
|
||||
try:
|
||||
return reduce(operator.getitem, map_path, data)
|
||||
except KeyError:
|
||||
log.debug('Invalid Path: {}', map_path)
|
||||
|
||||
for item in items:
|
||||
if item.mb_trackid:
|
||||
log.info('getting data for: {}', item)
|
||||
|
||||
# Fetch the data from the AB API.
|
||||
url = generate_url(item.mb_trackid)
|
||||
log.debug('fetching URL: {}', url)
|
||||
urls = [generate_url(item.mb_trackid, path) for path in LEVELS]
|
||||
log.debug('fetching URLs: {}', urls)
|
||||
try:
|
||||
rs = requests.get(url)
|
||||
res = [requests.get(url) for url in urls]
|
||||
except requests.RequestException as exc:
|
||||
log.info('request error: {}', exc)
|
||||
continue
|
||||
|
||||
# Check for missing tracks.
|
||||
if rs.status_code == 404:
|
||||
if any(r.status_code == 404 for r in res):
|
||||
log.info('recording ID {} not found', item.mb_trackid)
|
||||
continue
|
||||
|
||||
# Parse the JSON response.
|
||||
try:
|
||||
data = rs.json()
|
||||
data = res[0].json()
|
||||
data.update(res[1].json())
|
||||
except ValueError:
|
||||
log.debug('Invalid Response: {}', rs.text)
|
||||
log.debug('Invalid Response: {} & {}', [r.text for r in res])
|
||||
|
||||
# Get each field and assign it on the item.
|
||||
item.danceable = get_value(
|
||||
log,
|
||||
data,
|
||||
["highlevel", "danceability", "all", "danceable"],
|
||||
"highlevel", "danceability", "all", "danceable",
|
||||
)
|
||||
item.gender = get_value(
|
||||
"highlevel", "gender", "value",
|
||||
)
|
||||
item.genre_rosamerica = get_value(
|
||||
"highlevel", "genre_rosamerica", "value"
|
||||
)
|
||||
item.mood_acoustic = get_value(
|
||||
"highlevel", "mood_acoustic", "all", "acoustic"
|
||||
)
|
||||
item.mood_aggressive = get_value(
|
||||
"highlevel", "mood_aggresive", "all", "aggresive"
|
||||
)
|
||||
item.mood_electronic = get_value(
|
||||
"highlevel", "mood_electronic", "all", "electronic"
|
||||
)
|
||||
item.mood_happy = get_value(
|
||||
log,
|
||||
data,
|
||||
["highlevel", "mood_happy", "all", "happy"],
|
||||
"highlevel", "mood_happy", "all", "happy"
|
||||
)
|
||||
item.mood_party = get_value(
|
||||
log,
|
||||
data,
|
||||
["highlevel", "mood_party", "all", "party"],
|
||||
"highlevel", "mood_party", "all", "party"
|
||||
)
|
||||
item.mood_relaxed = get_value(
|
||||
"highlevel", "mood_relaxed", "all", "relaxed"
|
||||
)
|
||||
item.mood_sad = get_value(
|
||||
"highlevel", "mood_sad", "all", "sad"
|
||||
)
|
||||
item.rhythm = get_value(
|
||||
"highlevel", "ismir04_rhythm", "value"
|
||||
)
|
||||
item.tonal = get_value(
|
||||
"highlevel", "tonal_atonal", "all", "tonal"
|
||||
)
|
||||
item.voice_instrumental = get_value(
|
||||
"highlevel", "voice_instrumental", "value"
|
||||
)
|
||||
item.average_loudness = get_value(
|
||||
"lowlevel", "average_loudness"
|
||||
)
|
||||
item.chords_changes_rate = get_value(
|
||||
"tonal", "chords_changes_rate"
|
||||
)
|
||||
item.chords_key = get_value(
|
||||
"tonal", "chords_key"
|
||||
)
|
||||
item.chords_number_rate = get_value(
|
||||
"tonal", "chords_number_rate"
|
||||
)
|
||||
item.chords_scale = get_value(
|
||||
"tonal", "chords_scale"
|
||||
)
|
||||
item.initial_key = '{} {}'.format(
|
||||
get_value("tonal", "key_key"),
|
||||
get_value("tonal", "key_scale")
|
||||
)
|
||||
item.key_strength = get_value(
|
||||
"tonal", "key_stength"
|
||||
)
|
||||
|
||||
# Store the data. We only update flexible attributes, so we
|
||||
|
|
@ -91,16 +158,7 @@ def fetch_info(log, items):
|
|||
item.store()
|
||||
|
||||
|
||||
def generate_url(mbid):
|
||||
"""Generates url of AcousticBrainz end point for given MBID
|
||||
def generate_url(mbid, level):
|
||||
"""Generates AcousticBrainz end point url for given MBID.
|
||||
"""
|
||||
return ACOUSTIC_URL + mbid + LEVEL
|
||||
|
||||
|
||||
def get_value(log, data, map_path):
|
||||
"""Allows traversal of dictionary with cleaner formatting
|
||||
"""
|
||||
try:
|
||||
return reduce(lambda d, k: d[k], map_path, data)
|
||||
except KeyError:
|
||||
log.debug('Invalid Path: {}', map_path)
|
||||
return ACOUSTIC_BASE + mbid + level
|
||||
|
|
|
|||
|
|
@ -14,9 +14,40 @@ Enable the ``acousticbrainz`` plugin in your configuration (see :ref:`using-plug
|
|||
For all tracks with a MusicBrainz recording ID, the plugin currently sets
|
||||
these fields:
|
||||
|
||||
* ``danceable``: Predicts how easy the track is to dance to.
|
||||
* ``mood_happy``: Predicts the probability this track will evoke happiness.
|
||||
* ``mood_party``: Predicts the probability this track should be played at a
|
||||
party.
|
||||
* ``average_loudness``
|
||||
* ``chords_changes_rate``
|
||||
* ``chords_key``
|
||||
* ``chords_number_rate``
|
||||
* ``chords_scale``
|
||||
* ``danceable``
|
||||
* ``gender``
|
||||
* ``genre_rosamerica``
|
||||
* ``initial_key``
|
||||
* ``key_strength``
|
||||
* ``mood_acoustic``
|
||||
* ``mood_aggressive``
|
||||
* ``mood_electronic``
|
||||
* ``mood_happy``
|
||||
* ``mood_party``
|
||||
* ``mood_relaxed``
|
||||
* ``mood_sad``
|
||||
* ``rhythm``
|
||||
* ``tonal``
|
||||
* ``voice_instrumental``
|
||||
|
||||
These three fields are all numbers between 0.0 and 1.0.
|
||||
Automatic Tagging
|
||||
-----------------
|
||||
|
||||
To automatically tag files using AcousticBrainz data during import, just
|
||||
enable the ``acousticbrainz`` plugin (see :ref:`using-plugins`). When importing
|
||||
new files, beets will query the AcousticBrainz API using MBID and
|
||||
set the appropriate metadata.
|
||||
|
||||
Configuration
|
||||
-------------
|
||||
|
||||
To configure the plugin, make a ``acousticbrainz:`` section in your
|
||||
configuration file. There is one option:
|
||||
|
||||
- **auto**: Enable AcousticBrainz during ``beet import``.
|
||||
Default: ``yes``.
|
||||
|
|
|
|||
Loading…
Reference in a new issue