Backup_Repos/beets
1
0
Fork 0
mirror of https://github.com/beetbox/beets.git synced 2026-01-06 07:53:40 +01:00
Code Issues Projects Releases Wiki Activity

Update listenbrainz.py

Browse source
This commit is contained in:
Alok Saboo 2023-12-19 15:18:27 -05:00
parent 619fb17383
commit 7d6c123436
1 changed files with 33 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
beetsplug/listenbrainz.py
View file

@ -36,14 +36,16 @@ class ListenBrainzPlugin(BeetsPlugin):
def _lbupdate(self, items, write):
"""Obtain view count from Listenbrainz."""
ls = self.get_listens()
tracks = self.get_tracks_from_listens(ls)
self._log.info(f"Found {len(ls)} listens")
def _make_request(self, url):
def _make_request(self, url, params=None):
try:
response = requests.get(
url=url,
headers=self.AUTH_HEADER,
timeout=10,
params=params,
)
response.raise_for_status()
return response.json()
@ -72,13 +74,42 @@ class ListenBrainzPlugin(BeetsPlugin):
An IndexError if the JSON is not structured as expected.
"""
url = f"{self.ROOT}/user/{self.username}/listens"
response = self._make_request(url)
params={
"min_ts": min_ts,
"max_ts": max_ts,
"count": count,
},
response = self._make_request(url, params)
if response is not None:
return response["payload"]["listens"]
else:
return None
# write a function to return all the listens in the followign JSON format:
"""JSON format:
[
{
"mbid": "...",
"artist": "...",
"title": "...",
"playcount": "..."
}
]
"""
def get_tracks_from_listens(self, listens):
tracks = []
for listen in listens:
tracks.append(
{
"artist": listen.get("artist_name"),
"identifier": listen.get("identifier"),
"title": listen.get("track_name"),
}
)
self._log.debug(f"track: {tracks[-1]}")
return self.get_track_info(tracks)
def get_playlists_createdfor(self, username):
"""Returns a list of playlists created by a user."""
url = f"{self.ROOT}/user/{username}/playlists/createdfor"