mirror of
https://code.eliotberriot.com/funkwhale/funkwhale.git
synced 2025-10-05 18:24:21 +02:00
Blacked the code
This commit is contained in:
parent
b6fc0051fa
commit
62ca3bd736
279 changed files with 8861 additions and 9527 deletions
|
@ -1,42 +1,43 @@
|
|||
|
||||
|
||||
def load(model, *args, **kwargs):
|
||||
importer = registry[model.__name__](model=model)
|
||||
return importer.load(*args, **kwargs)
|
||||
|
||||
|
||||
class Importer(object):
|
||||
def __init__(self, model):
|
||||
self.model = model
|
||||
|
||||
def load(self, cleaned_data, raw_data, import_hooks):
|
||||
mbid = cleaned_data.pop('mbid')
|
||||
mbid = cleaned_data.pop("mbid")
|
||||
m = self.model.objects.update_or_create(mbid=mbid, defaults=cleaned_data)[0]
|
||||
for hook in import_hooks:
|
||||
hook(m, cleaned_data, raw_data)
|
||||
return m
|
||||
|
||||
|
||||
class Mapping(object):
|
||||
"""Cast musicbrainz data to funkwhale data and vice-versa"""
|
||||
|
||||
def __init__(self, musicbrainz_mapping):
|
||||
self.musicbrainz_mapping = musicbrainz_mapping
|
||||
|
||||
self._from_musicbrainz = {}
|
||||
self._to_musicbrainz = {}
|
||||
for field_name, conf in self.musicbrainz_mapping.items():
|
||||
self._from_musicbrainz[conf['musicbrainz_field_name']] = {
|
||||
'field_name': field_name,
|
||||
'converter': conf.get('converter', lambda v: v)
|
||||
self._from_musicbrainz[conf["musicbrainz_field_name"]] = {
|
||||
"field_name": field_name,
|
||||
"converter": conf.get("converter", lambda v: v),
|
||||
}
|
||||
self._to_musicbrainz[field_name] = {
|
||||
'field_name': conf['musicbrainz_field_name'],
|
||||
'converter': conf.get('converter', lambda v: v)
|
||||
"field_name": conf["musicbrainz_field_name"],
|
||||
"converter": conf.get("converter", lambda v: v),
|
||||
}
|
||||
def from_musicbrainz(self, key, value):
|
||||
return self._from_musicbrainz[key]['field_name'], self._from_musicbrainz[key]['converter'](value)
|
||||
|
||||
registry = {
|
||||
'Artist': Importer,
|
||||
'Track': Importer,
|
||||
'Album': Importer,
|
||||
'Work': Importer,
|
||||
}
|
||||
def from_musicbrainz(self, key, value):
|
||||
return (
|
||||
self._from_musicbrainz[key]["field_name"],
|
||||
self._from_musicbrainz[key]["converter"](value),
|
||||
)
|
||||
|
||||
|
||||
registry = {"Artist": Importer, "Track": Importer, "Album": Importer, "Work": Importer}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue