130 lines
3.7 KiB
Python
Executable File
130 lines
3.7 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import json
|
|
import csv
|
|
import os.path
|
|
import requests
|
|
|
|
__author__ = 'Alexander Popov'
|
|
__version__ = '2.0.1'
|
|
__license__ = 'Unlicense'
|
|
|
|
|
|
def get_pages(username, api_key, limit=200):
|
|
""" Getting the number of pages with scrobbling data """
|
|
|
|
response = requests.get(
|
|
'https://ws.audioscrobbler.com/2.0/'
|
|
'?method=user.getrecenttracks&user='
|
|
'{0}&api_key={1}&format=json&limit={2}'
|
|
.format(username, api_key, limit))
|
|
|
|
data = json.loads(response.content.decode("utf8"))
|
|
|
|
return(int(data['recenttracks']['@attr']['totalPages']))
|
|
|
|
|
|
def get_page(username, api_key, page, limit=200):
|
|
""" Getting scrobbling data from a page """
|
|
|
|
response = requests.get(
|
|
'https://ws.audioscrobbler.com/2.0/'
|
|
'?method=user.getrecenttracks&user={0}&api_key={1}&format=json'
|
|
'&limit={2}&page={3}'.format(username, api_key, limit, page))
|
|
|
|
data = json.loads(response.content.decode("utf8"))
|
|
|
|
return(data['recenttracks']['track'])
|
|
|
|
|
|
def get_now_scrobbling(username, api_key):
|
|
""" Getting now scrobbling track """
|
|
|
|
response = requests.get(
|
|
'https://ws.audioscrobbler.com/2.0/'
|
|
'?method=user.getrecenttracks&user={0}'
|
|
'&api_key={1}&format=json&limit=1'
|
|
.format(username, api_key))
|
|
|
|
data = json.loads(response.content.decode('utf-8'))
|
|
|
|
if '@attr' in data['recenttracks']['track'][0]:
|
|
return(True)
|
|
else:
|
|
return(False)
|
|
|
|
|
|
def scrobbling_export(tracks, username, export_format='is as'):
|
|
""" Save scrobbled track via various format """
|
|
|
|
if export_format == 'is as':
|
|
with open('%s.json' % (username), 'w', encoding='utf-8') as f:
|
|
data = json.dumps(tracks, indent=4,
|
|
sort_keys=True, ensure_ascii=False)
|
|
f.write(data)
|
|
|
|
elif export_format == 'simple':
|
|
_ = []
|
|
|
|
for track in tracks:
|
|
_.append([
|
|
track['artist']['#text'], track['name'], track['date']['uts']
|
|
])
|
|
|
|
with open('%s.json' % (username), 'w', encoding='utf-8') as f:
|
|
data = json.dumps(_, indent=4,
|
|
sort_keys=True, ensure_ascii=False)
|
|
f.write(data)
|
|
|
|
elif export_format == 'csv':
|
|
_ = []
|
|
|
|
for track in tracks:
|
|
_.append([
|
|
track['artist']['#text'],
|
|
track['name'],
|
|
int(track['date']['uts'])
|
|
])
|
|
|
|
with open('%s.csv' % (username), 'w', encoding='utf-8', newline='') as f:
|
|
data = csv.writer(f, quoting=csv.QUOTE_NONNUMERIC, delimiter=',')
|
|
for row in _:
|
|
data.writerow(row)
|
|
|
|
return(1)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
_ = dict()
|
|
|
|
if os.path.exists('./config.json'):
|
|
with open('./config.json') as f:
|
|
_ = json.loads(f.read())
|
|
else:
|
|
_['api_key'] = input('API Key: ')
|
|
_['username'] = input('Username: ')
|
|
api_key, username = _['api_key'], _['username']
|
|
|
|
total_pages = get_pages(username, api_key)
|
|
current_page = 1
|
|
scrobbled = []
|
|
|
|
while current_page <= total_pages:
|
|
print('\r{0:.2f}% [{1} of {2}]'
|
|
.format((current_page * 100 / total_pages),
|
|
current_page, total_pages), end='')
|
|
|
|
response = get_page(username, api_key, current_page)
|
|
|
|
for track in response:
|
|
scrobbled.append(track)
|
|
|
|
current_page += 1
|
|
|
|
# if get_now_scrobbling(username, api_key):
|
|
# scrobbled.pop(0)
|
|
|
|
if scrobbling_export(scrobbled, username, _['export_format']):
|
|
print('\n{0} tracks saved!'.format(
|
|
len(scrobbled), username))
|