mirror of
https://github.com/beetbox/beets.git
synced 2026-01-01 13:33:02 +01:00
acousticbrainz: Initial cleanup
This commit is contained in:
nath@home
parent
6b0249df08
commit
1963b52903
1 changed files with 104 additions and 104 deletions
|
|
@ -42,7 +42,7 @@ class AcousticPlugin(plugins.BeetsPlugin):
|
|||
|
||||
def func(lib, opts, args):
|
||||
items = lib.items(ui.decargs(args))
|
||||
fetch_info(self._log, items, ui.should_write())
|
||||
self._fetch_info(items, ui.should_write())
|
||||
|
||||
cmd.func = func
|
||||
return [cmd]
|
||||
|
|
@ -50,119 +50,119 @@ class AcousticPlugin(plugins.BeetsPlugin):
|
|||
def import_task_files(self, session, task):
|
||||
"""Function is called upon beet import.
|
||||
"""
|
||||
self._fetch_info(task.imported_items(), False)
|
||||
|
||||
items = task.imported_items()
|
||||
fetch_info(self._log, items, False)
|
||||
def _fetch_info(self, items, write):
|
||||
"""Get data from AcousticBrainz for the items.
|
||||
"""
|
||||
|
||||
|
||||
def fetch_info(log, items, write):
|
||||
"""Get data from AcousticBrainz for the items.
|
||||
"""
|
||||
|
||||
def get_value(*map_path):
|
||||
try:
|
||||
return reduce(operator.getitem, map_path, data)
|
||||
except KeyError:
|
||||
log.debug(u'Invalid Path: {}', map_path)
|
||||
|
||||
for item in items:
|
||||
if item.mb_trackid:
|
||||
log.info(u'getting data for: {}', item)
|
||||
|
||||
# Fetch the data from the AB API.
|
||||
urls = [generate_url(item.mb_trackid, path) for path in LEVELS]
|
||||
log.debug(u'fetching URLs: {}', ' '.join(urls))
|
||||
def get_value(*map_path):
|
||||
try:
|
||||
res = [requests.get(url) for url in urls]
|
||||
return reduce(operator.getitem, map_path, data)
|
||||
except KeyError:
|
||||
log.debug(u'Invalid Path: {}', map_path)
|
||||
|
||||
for item in (item for item in items if item.mb_trackid):
|
||||
self._log.info(u'getting data for: {}', item)
|
||||
data = self._get_data(item.mb_trackid)
|
||||
if data:
|
||||
# Get each field and assign it on the item.
|
||||
item.danceable = get_value(
|
||||
"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_aggressive", "all", "aggressive"
|
||||
)
|
||||
item.mood_electronic = get_value(
|
||||
"highlevel", "mood_electronic", "all", "electronic"
|
||||
)
|
||||
item.mood_happy = get_value(
|
||||
"highlevel", "mood_happy", "all", "happy"
|
||||
)
|
||||
item.mood_party = get_value(
|
||||
"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.bpm = get_value(
|
||||
"rhythm", "bpm"
|
||||
)
|
||||
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_strength"
|
||||
)
|
||||
|
||||
# Store the data.
|
||||
item.store()
|
||||
if write:
|
||||
item.try_write()
|
||||
|
||||
def _get_data(self, mbid):
|
||||
data = {}
|
||||
for url in _generate_urls(mbid):
|
||||
self._log.debug(u'fetching URL: {}', url)
|
||||
|
||||
try:
|
||||
res = requests.get(url)
|
||||
except requests.RequestException as exc:
|
||||
log.info(u'request error: {}', exc)
|
||||
continue
|
||||
self._log.info(u'request error: {}', exc)
|
||||
return {}
|
||||
|
||||
# Check for missing tracks.
|
||||
if any(r.status_code == 404 for r in res):
|
||||
log.info(u'recording ID {} not found', item.mb_trackid)
|
||||
continue
|
||||
if res.status_code == 404:
|
||||
self._log.info(u'recording ID {} not found', mbid)
|
||||
return {}
|
||||
|
||||
# Parse the JSON response.
|
||||
try:
|
||||
data = res[0].json()
|
||||
data.update(res[1].json())
|
||||
data.update(res.json())
|
||||
except ValueError:
|
||||
log.debug(u'Invalid Response: {} & {}', [r.text for r in res])
|
||||
self._log.debug(u'Invalid Response: {}', res.text)
|
||||
return {}
|
||||
|
||||
# Get each field and assign it on the item.
|
||||
item.danceable = get_value(
|
||||
"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_aggressive", "all", "aggressive"
|
||||
)
|
||||
item.mood_electronic = get_value(
|
||||
"highlevel", "mood_electronic", "all", "electronic"
|
||||
)
|
||||
item.mood_happy = get_value(
|
||||
"highlevel", "mood_happy", "all", "happy"
|
||||
)
|
||||
item.mood_party = get_value(
|
||||
"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.bpm = get_value(
|
||||
"rhythm", "bpm"
|
||||
)
|
||||
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_strength"
|
||||
)
|
||||
|
||||
# Store the data.
|
||||
item.store()
|
||||
if write:
|
||||
item.try_write()
|
||||
return data
|
||||
|
||||
|
||||
def generate_url(mbid, level):
|
||||
def _generate_urls(mbid):
|
||||
"""Generates AcousticBrainz end point url for given MBID.
|
||||
"""
|
||||
return ACOUSTIC_BASE + mbid + level
|
||||
for level in LEVELS:
|
||||
yield ACOUSTIC_BASE + mbid + level
|
||||
|
|
|
|||
Loading…
Reference in a new issue