69 lines
2.6 KiB
Python
Executable File
69 lines
2.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
from time import sleep
|
|
import json
|
|
import pylistenbrainz
|
|
import sys
|
|
|
|
|
|
def connect_client(auth_token):
|
|
client = pylistenbrainz.ListenBrainz()
|
|
client.set_auth_token(auth_token)
|
|
return client
|
|
|
|
|
|
def parse_listen(raw):
|
|
listen = pylistenbrainz.Listen(
|
|
track_name=raw['track_metadata']['track_name'],
|
|
artist_name=raw['track_metadata']['artist_name']
|
|
)
|
|
if 'listened_at' in raw:
|
|
listen.listened_at = raw['listened_at']
|
|
if 'user_name' in raw:
|
|
listen.username = raw['user_name']
|
|
if 'release_name' in raw['track_metadata']:
|
|
listen.release_name = raw['track_metadata']['release_name']
|
|
if 'additional_info' in raw['track_metadata']:
|
|
listen.additional_info = raw['track_metadata']['additional_info']
|
|
if 'tracknumber' in raw['track_metadata']['additional_info']:
|
|
listen.tracknumber = raw['track_metadata']['additional_info']['tracknumber']
|
|
if 'spotify_id' in raw['track_metadata']['additional_info']:
|
|
listen.spotify_id = raw['track_metadata']['additional_info']['spotify_id']
|
|
if 'isrc' in raw['track_metadata']['additional_info']:
|
|
listen.spotify_id = raw['track_metadata']['additional_info']['isrc']
|
|
if 'listening_from' in raw['track_metadata']['additional_info']:
|
|
listen.listening_from = raw['track_metadata']['additional_info']['listening_from']
|
|
if 'mbid_mapping' in raw['track_metadata']:
|
|
if 'recording_mbid' in raw['track_metadata']['mbid_mapping']:
|
|
listen.recording_mbid = raw['track_metadata']['mbid_mapping']['recording_mbid']
|
|
if 'artist_mbids' in raw['track_metadata']['mbid_mapping']:
|
|
listen.artist_mbids = raw['track_metadata']['mbid_mapping']['artist_mbids']
|
|
if 'release_mbid' in raw['track_metadata']['mbid_mapping']:
|
|
listen.release_mbid = raw['track_metadata']['mbid_mapping']['release_mbid']
|
|
return listen
|
|
|
|
|
|
def main():
|
|
client = connect_client(sys.argv[1])
|
|
with open(sys.argv[2], 'r') as read_file:
|
|
exported = json.load(read_file)
|
|
print("Opened file, starting import...")
|
|
for raw_listen in exported:
|
|
listen = parse_listen(raw_listen)
|
|
for x in range(0, 5):
|
|
try:
|
|
client.submit_single_listen(listen)
|
|
submit_error = None
|
|
except BaseException:
|
|
submit_error = True
|
|
|
|
if submit_error:
|
|
print("Error occurred, retrying in 5 seconds...")
|
|
sleep(5)
|
|
else:
|
|
break
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|