Skip to content

Commit

Permalink
Check
Browse files Browse the repository at this point in the history
  • Loading branch information
@nikhil25803
nikhil25803 committed May 30, 2024
1 parent 287e163 commit 011a8ac
Show file tree
Hide file tree
Showing 3 changed files with 129 additions and 58 deletions.
13 changes: 0 additions & 13 deletions docs/modules/lastfm.md
View file

This file was deleted.

2 changes: 1 addition & 1 deletion src/scrape_up/lastfm/__init__.py
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
from .lastfm import Lastfm

__all__ = ["Lastfm"]
__all__ = ["Lastfm"]
172 changes: 128 additions & 44 deletions src/scrape_up/lastfm/lastfm.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ class Lastfm:
| Methods | Details |
| --------------------------- | ---------------------------------------------------------------------------------------------------- |
| `.user_stats()` | Returns the numbers of scrobbles, artists and loved tracks. |
| `.user_stats()` | Returns the numbers of scrobbles, artists and loved tracks. |
| `.recent_tracks()` | Returns a dictionary containing the latest tracks scrobbled by the user. |
| `.top_tracks()` | Returns a dictionary containing the top tracks of the user. |
| `.top_albums()` | Returns a dictionary containing the top albums of the user. |
Expand Down Expand Up @@ -44,19 +44,31 @@ def user_stats(self) -> dict:
req = get(self.url, self.config)
page_soup = Soup(req.content, "html.parser")

scrobbles = page_soup.find_all('a', attrs={'href':f'/user/{self.username}/library'})[1].text.replace(",", "")
artists = page_soup.find_all('a', attrs={'href':f'/user/{self.username}/library/artists'})[0].text.replace(",", "")
scrobbles = page_soup.find_all(
"a", attrs={"href": f"/user/{self.username}/library"}
)[1].text.replace(",", "")
artists = page_soup.find_all(
"a", attrs={"href": f"/user/{self.username}/library/artists"}
)[0].text.replace(",", "")
try:
loved_tracks = page_soup.find_all('a', attrs={'href':f'/user/{self.username}/loved'})[1].text
loved_tracks = page_soup.find_all(
"a", attrs={"href": f"/user/{self.username}/loved"}
)[1].text
except IndexError:
loved_tracks = 0

return {"data": {"scrobbles": int(scrobbles), "artists": int(artists), "loved_tracks": loved_tracks}, "status": "success"}


return {
"data": {
"scrobbles": int(scrobbles),
"artists": int(artists),
"loved_tracks": loved_tracks,
},
"status": "success",
}

except Exception as e:
return e


def recent_tracks(self) -> dict:
"""
Class - `Lastfm`
Expand Down Expand Up @@ -88,18 +100,38 @@ def recent_tracks(self) -> dict:
try:
req = get(f"{self.url}", self.config)
page_soup = Soup(req.content, "html.parser")

song_names = [i.text.strip('\n') for i in page_soup.find_all("td", attrs={"class": "chartlist-name"})]
artist_names = [i.text.strip('\n') for i in page_soup.find_all("td", attrs={"class": "chartlist-artist"})]
time_stamps = [unicodedata.normalize("NFKD", i.get_text(strip=True)) for i in page_soup.find_all("td", attrs={"class": "chartlist-timestamp"})]

song_names = [
i.text.strip("\n")
for i in page_soup.find_all("td", attrs={"class": "chartlist-name"})
]
artist_names = [
i.text.strip("\n")
for i in page_soup.find_all("td", attrs={"class": "chartlist-artist"})
]
time_stamps = [
unicodedata.normalize("NFKD", i.get_text(strip=True))
for i in page_soup.find_all(
"td", attrs={"class": "chartlist-timestamp"}
)
]

min_len = min(len(song_names), len(time_stamps), len(artist_names))
return {"data": [{"song": song_names[i], "artist": artist_names[i], "time": time_stamps[i]} for i in range(min_len)], "status": "success"}
return {
"data": [
{
"song": song_names[i],
"artist": artist_names[i],
"time": time_stamps[i],
}
for i in range(min_len)
],
"status": "success",
}

except Exception as e:
return e


def top_artists(self) -> dict:
"""
Class - `Lastfm`
Expand Down Expand Up @@ -131,18 +163,29 @@ def top_artists(self) -> dict:
}
```
"""
try:
try:
req = get(f"{self.url}", self.config)
page_soup = Soup(req.content, "html.parser")
data = page_soup.find("section", attrs={"id": "top-artists"}).text.split("All time")[1]
data = [i.strip() for i in data.replace("\n", "").strip().split(" ") if i not in ['']][:-1]

return {"data": [{"artist": data[i], "plays": data[i+1]} for i in range(0, len(data), 2)], "status": "success"}

data = page_soup.find("section", attrs={"id": "top-artists"}).text.split(
"All time"
)[1]
data = [
i.strip()
for i in data.replace("\n", "").strip().split(" ")
if i not in [""]
][:-1]

return {
"data": [
{"artist": data[i], "plays": data[i + 1]}
for i in range(0, len(data), 2)
],
"status": "success",
}

except Exception as e:
return e


def top_albums(self) -> dict:
"""
Class - `Lastfm`
Expand Down Expand Up @@ -174,16 +217,28 @@ def top_albums(self) -> dict:
"""
try:
req = get(f"{self.url}", self.config)
page_soup = Soup(req.content, "html.parser").find("section", attrs={"id": "top-albums"})
data = [j.get_text(strip=True) for i in page_soup.find_all("div", attrs={"class": "grid-items-item-details"}) for j in i.find_all("a")]
page_soup = Soup(req.content, "html.parser").find(
"section", attrs={"id": "top-albums"}
)
data = [
j.get_text(strip=True)
for i in page_soup.find_all(
"div", attrs={"class": "grid-items-item-details"}
)
for j in i.find_all("a")
]

return {
"data": [
{"album": data[i], "artist": data[i + 1], "plays": data[i + 2]}
for i in range(0, len(data), 3)
],
"status": "success",
}

return {"data": [{"album": data[i], "artist": data[i+1], "plays": data[i+2]} for i in range(0, len(data), 3)], "status": "success"}

except Exception as e:
return e



def top_tracks(self) -> dict:
"""
Class - `Lastfm`
Expand All @@ -199,19 +254,38 @@ def top_tracks(self) -> dict:
{"data": [{"song": "Pain", "artist": "PinkPantheress", "plays": 10}, {"song": "Attracted to You", "artist": "PinkPantheress", "plays": 5}, {"song": "Boy's a liar Pt. 2", "artist": "PinkPantheress", "plays": 4}, {"song": "I must apologise", "artist": "PinkPantheress", "plays": 4}], "status": "success"}
```
"""

try:
req = get(f"{self.url}", self.config)
page_soup = Soup(req.content, "html.parser").find("section", attrs={"id": "top-tracks"})
song = [i.get_text(strip=True) for i in page_soup.find_all("td", attrs={"class": "chartlist-name"})]
artist = [i.get_text(strip=True) for i in page_soup.find_all("td", attrs={"class": "chartlist-artist"})]
plays = [int(i.get_text(strip=True).split("scrobbles")[0]) for i in page_soup.find_all("span", attrs={"class": "chartlist-count-bar-value"})]

return {"data": [{"song": song[i], "artist": artist[i], "plays": plays[i]} for i in range(min(len(song), len(artist), len(plays)))], "status": "success"}

page_soup = Soup(req.content, "html.parser").find(
"section", attrs={"id": "top-tracks"}
)
song = [
i.get_text(strip=True)
for i in page_soup.find_all("td", attrs={"class": "chartlist-name"})
]
artist = [
i.get_text(strip=True)
for i in page_soup.find_all("td", attrs={"class": "chartlist-artist"})
]
plays = [
int(i.get_text(strip=True).split("scrobbles")[0])
for i in page_soup.find_all(
"span", attrs={"class": "chartlist-count-bar-value"}
)
]

return {
"data": [
{"song": song[i], "artist": artist[i], "plays": plays[i]}
for i in range(min(len(song), len(artist), len(plays)))
],
"status": "success",
}

except Exception as e:
return e

def get_following(self) -> dict:
"""
Class - `Lastfm`
Expand All @@ -222,17 +296,22 @@ def get_following(self) -> dict:
lastfm_user.get_following()
```
Returns the total number of users followed by the user.:
Returns the total number of users followed by the user.:
```python
{'data': {'following': 1}, 'status': 'success'}
```
"""
try:
req = get(f"{self.url}/following", self.config)
page_soup = Soup(req.content, "html.parser")
following = page_soup.find("h1", attrs={"class": "content-top-header"}).get_text(strip=True).split('\n')[-1].strip()
following = following.replace(')', '').replace('(', '')

following = (
page_soup.find("h1", attrs={"class": "content-top-header"})
.get_text(strip=True)
.split("\n")[-1]
.strip()
)
following = following.replace(")", "").replace("(", "")

return {"data": {"following": int(following)}, "status": "success"}

except Exception as e:
Expand All @@ -256,10 +335,15 @@ def get_followers(self) -> dict:
try:
req = get(f"{self.url}/followers", self.config)
page_soup = Soup(req.content, "html.parser")
followers = page_soup.find("h1", attrs={"class": "content-top-header"}).get_text(strip=True).split('\n')[-1].strip()
followers = followers.replace(')', '').replace('(', '')

followers = (
page_soup.find("h1", attrs={"class": "content-top-header"})
.get_text(strip=True)
.split("\n")[-1]
.strip()
)
followers = followers.replace(")", "").replace("(", "")

return {"data": {"followers": int(followers)}, "status": "success"}

except Exception as e:
return e
return e

0 comments on commit 011a8ac

Please sign in to comment.