use black library for code

This commit is contained in:
Alexander Popov 2020-10-05 02:32:15 +03:00
parent b63e1f1536
commit f6f3a653df
4 changed files with 94 additions and 80 deletions

View File

@ -5,118 +5,124 @@ import csv
import os.path
import requests
__author__ = 'Alexander Popov'
__version__ = '2.1.0'
__license__ = 'Unlicense'
__author__ = "Alexander Popov"
__version__ = "2.1.0"
__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))
"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']))
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))
"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'])
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))
"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'))
data = json.loads(response.content.decode("utf-8"))
if '@attr' in data['recenttracks']['track'][0]:
return(True)
if "@attr" in data["recenttracks"]["track"][0]:
return True
else:
return(False)
return False
def scrobbling_export(tracks, username, export_format='dump'):
def scrobbling_export(tracks, username, export_format="dump"):
""" Save scrobbled track via various format """
if export_format == 'dump':
with open('%s.json' % (username), 'w', encoding='utf-8') as f:
data = json.dumps(tracks, indent=4,
sort_keys=True, ensure_ascii=False)
if export_format == "dump":
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':
elif export_format == "simple":
_ = []
for track in tracks:
_.append({
'artist': track['artist']['#text'],
'name': track['name'],
'album': track['album']['#text'],
'date': int(track['date']['uts'])
})
_.append(
{
"artist": track["artist"]["#text"],
"name": track["name"],
"album": track["album"]["#text"],
"date": int(track["date"]["uts"]),
}
)
with open('%s.json' % (username), 'w', encoding='utf-8') as f:
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':
elif export_format == "csv":
_ = []
for track in tracks:
_.append([
track['artist']['#text'],
track['name'],
track['album']['#text'],
int(track['date']['uts'])
])
_.append(
[
track["artist"]["#text"],
track["name"],
track["album"]["#text"],
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=',')
data.writerow(['artist', 'track', 'album', 'date'])
with open("%s.csv" % (username), "w", encoding="utf-8", newline="") as f:
data = csv.writer(f, quoting=csv.QUOTE_NONNUMERIC, delimiter=",")
data.writerow(["artist", "track", "album", "date"])
data.writerows(_)
return(1)
return 1
if __name__ == '__main__':
if __name__ == "__main__":
_ = dict()
if os.path.exists('./config.json'):
with open('./config.json') as f:
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']
_["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='')
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)
@ -128,6 +134,5 @@ if __name__ == '__main__':
# 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))
if scrobbling_export(scrobbled, username, _["export_format"]):
print("\n{0} tracks saved!".format(len(scrobbled), username))

1
requirements.txt Normal file
View File

@ -0,0 +1 @@
requests==2.24.0

View File

@ -2,18 +2,24 @@
import lastfm_backup as lfm
API_KEY = '0'
USERNAME = 'admin'
API_KEY = "0"
USERNAME = "admin"
if __name__ == '__main__':
pages = lfm.get_pages(USERNAME, API_KEY)
count = 0
if __name__ == "__main__":
pages = lfm.get_pages(USERNAME, API_KEY)
count = 0
for page in range(1, pages + 1):
tracks = lfm.get_scrobbles(USERNAME, API_KEY, page)
for page in range(1, pages + 1):
tracks = lfm.get_scrobbles(USERNAME, API_KEY, page)
for track in tracks:
print('%s - %s' % (track['artist']['#text'], track['name'],))
count = count + 1
for track in tracks:
print(
"%s - %s"
% (
track["artist"]["#text"],
track["name"],
)
)
count = count + 1
print('\nTotal scrobbling tracks: %d' % count)
print("\nTotal scrobbling tracks: %d" % count)

View File

@ -3,19 +3,21 @@ import setuptools
import lastfm_backup
setuptools.setup(
name='lastfm-backup',
name="lastfm-backup",
version=lastfm_backup.__version__,
description='Last.fm scrobbling backup',
description="Last.fm scrobbling backup",
author=lastfm_backup.__author__,
author_email='iiiypuk@fastmail.fm',
url='https://github.com/iiiypuk/lastfm-backup',
py_modules=['lastfm_backup'],
scripts=['lastfm_backup.py'],
author_email="iiiypuk@fastmail.fm",
url="https://github.com/iiiypuk/lastfm-backup",
py_modules=["lastfm_backup"],
scripts=["lastfm_backup.py"],
license=lastfm_backup.__license__,
platforms='any',
keywords=['last.fm', 'lastfm', 'backup'],
classifiers=['License :: Public Domain',
'Programming Language :: Python :: 3',
'Operating System :: OS Independent'],
python_requires='>=3.2'
platforms="any",
keywords=["last.fm", "lastfm", "backup"],
classifiers=[
"License :: Public Domain",
"Programming Language :: Python :: 3",
"Operating System :: OS Independent",
],
python_requires=">=3.2",
)