mirrors/streamrip
1
0
Fork 0
mirror of https://github.com/nathom/streamrip.git synced 2024-09-19 19:28:46 -04:00
Code Issues Projects Releases Packages Wiki Activity

Fix docstrings, remove Version class

Browse source 
Signed-off-by: nathom <nathanthomas707@gmail.com>
This commit is contained in:
nathom 2021-06-21 10:48:42 -07:00
parent f92002d3aa
commit b9eca77c12
2 changed files with 12 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
streamrip/core.py
View file

@ -366,7 +366,6 @@ class MusicDL(list):
:param urls: :param urls:
""" """
# Available keys: ['artist', 'title'] # Available keys: ['artist', 'title']
QUERY_FORMAT: Dict[str, str] = { QUERY_FORMAT: Dict[str, str] = {
"tidal": "{title}", "tidal": "{title}",
@ -637,7 +636,6 @@ class MusicDL(list):
:type url: str :type url: str
:rtype: Tuple[str, list] :rtype: Tuple[str, list]
""" """
logger.debug("Fetching lastfm playlist") logger.debug("Fetching lastfm playlist")
info = [] info = []

33
streamrip/utils.py
View file

@ -112,6 +112,12 @@ def get_quality_id(bit_depth: Optional[int], sampling_rate: Optional[int]):
def get_stats_from_quality( def get_stats_from_quality(
quality_id: int, quality_id: int,
) -> Tuple[Optional[int], Optional[int]]: ) -> Tuple[Optional[int], Optional[int]]:
"""Get bit depth and sampling rate based on the quality id.
:param quality_id:
:type quality_id: int
:rtype: Tuple[Optional[int], Optional[int]]
"""
if quality_id <= 1: if quality_id <= 1:
return (None, None) return (None, None)
elif quality_id == 2: elif quality_id == 2:
@ -343,6 +349,12 @@ deezer_id_link_regex = re.compile(
def extract_deezer_dynamic_link(url: str) -> Tuple[str, str]: def extract_deezer_dynamic_link(url: str) -> Tuple[str, str]:
"""Extract a deezer url that includes an ID from a deezer.page.link url.
:param url:
:type url: str
:rtype: Tuple[str, str]
"""
session = gen_threadsafe_session({"User-Agent": AGENT}) session = gen_threadsafe_session({"User-Agent": AGENT})
r = session.get(url) r = session.get(url)
match = deezer_id_link_regex.search(r.text) match = deezer_id_link_regex.search(r.text)
@ -370,24 +382,3 @@ def get_container(quality: int, source: str) -> str:
return "AAC" return "AAC"
return "MP3" return "MP3"
class Version:
def __init__(self, version: str):
self.value = tuple(map(int, version.split(".")))
def __gt__(self, other) -> bool:
assert isinstance(other, Version)
for v1, v2 in zip(self.value, other.value):
if v1 > v2:
return True
elif v1 < v2:
return False
return False
def __lt__(self, other) -> bool:
return not self.__gt__(other) and not self.__eq__(other)
def __eq__(self, other) -> bool:
assert isinstance(other, Version)
return all(v1 == v2 for v1, v2 in zip(self.value, other.value))