Add tests for encoders

and some refactoring
This commit is contained in:
Ritiek Malhotra
2020-03-17 17:58:44 +05:30
parent 29005f24ed
commit dae76a0abb
12 changed files with 210 additions and 21 deletions

View File

@@ -8,13 +8,13 @@ BASE_URL = "https://genius.com"
class Genius(LyricBase):
def __init__(self, artist, song):
def __init__(self, artist, track):
self.artist = artist
self.song = song
self.track = track
self.base_url = BASE_URL
def _guess_lyric_url(self):
query = "/{} {} lyrics".format(self.artist, self.song)
query = "/{} {} lyrics".format(self.artist, self.track)
query = query.replace(" ", "-")
encoded_query = urllib.request.quote(query)
lyric_url = self.base_url + encoded_query
@@ -28,7 +28,7 @@ class Genius(LyricBase):
except urllib.request.HTTPError:
raise LyricsNotFoundError(
"Could not find lyrics for {} - {} at URL: {}".format(
self.artist, self.song, url
self.artist, self.track, url
)
)
else: