mirror/maloja
1
0
Fork 0
mirror of https://github.com/krateng/maloja.git synced 2023-08-10 21:12:55 +03:00
Code Issues Projects Releases Wiki Activity

Implemented import from own export

Browse Source
This commit is contained in:
krateng 2022-04-04 17:50:46 +02:00
parent 1b087e92db
commit c647a57983
1 changed files with 31 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
maloja/proccontrol/tasks/import_scrobbles.py
View File

@ -32,24 +32,28 @@ def import_scrobbles(inputf):
filename = os.path.basename(inputf) filename = os.path.basename(inputf)
if re.match(".*\.csv",filename): if re.match(".*\.csv",filename):
type = "Last.fm" typeid,typedesc = "lastfm","Last.fm"
importfunc = parse_lastfm importfunc = parse_lastfm
elif re.match("endsong_[0-9]+\.json",filename): elif re.match("endsong_[0-9]+\.json",filename):
type = "Spotify" typeid,typedesc = "spotify","Spotify"
importfunc = parse_spotify_full importfunc = parse_spotify_full
elif re.match("StreamingHistory[0-9]+\.json",filename): elif re.match("StreamingHistory[0-9]+\.json",filename):
type = "Spotify" typeid,typedesc = "spotify","Spotify"
importfunc = parse_spotify_lite importfunc = parse_spotify_lite
elif re.match("maloja_export_[0-9]+\.json",filename):
typeid,typedesc = "maloja","Maloja"
importfunc = parse_maloja
else: else:
print("File",inputf,"could not be identified as a valid import source.") print("File",inputf,"could not be identified as a valid import source.")
return result return result
print(f"Parsing {col['yellow'](inputf)} as {col['cyan'](type)} export") print(f"Parsing {col['yellow'](inputf)} as {col['cyan'](typedesc)} export")
print("This could take a while...")
timestamps = set() timestamps = set()
scrobblebuffer = [] scrobblebuffer = []
@ -75,7 +79,7 @@ def import_scrobbles(inputf):
"length":None "length":None
}, },
"duration":scrobble['duration'], "duration":scrobble['duration'],
"origin":"import:" + import_type, "origin":"import:" + typeid,
"extra":{ "extra":{
"album":scrobble['album'] "album":scrobble['album']
# saving this in the scrobble instead of the track because for now it's not meant # saving this in the scrobble instead of the track because for now it's not meant
@ -281,3 +285,24 @@ def parse_lastfm(inputf):
except Exception as e: except Exception as e:
yield ('FAIL',None,f"{entry} could not be parsed. Scrobble not imported. ({repr(e)})") yield ('FAIL',None,f"{entry} could not be parsed. Scrobble not imported. ({repr(e)})")
continue continue
def parse_maloja(inputf):
with open(inputf,'r') as inputfd:
data = json.load(inputfd)
scrobbles = data['scrobbles']
for s in scrobbles:
try:
yield ('CONFIDENT_IMPORT',{
'title': s['track']['title'],
'artists': s['track']['artists'],
'album': s['track'].get('album',{}).get('name',''),
'timestamp': s['time'],
'duration': s['duration']
},'')
except Exception as e:
yield ('FAIL',None,f"{s} could not be parsed. Scrobble not imported. ({repr(e)})")
continue