mirror of
https://github.com/KevinMidboe/spotify-downloader.git
synced 2025-10-29 18:00:15 +00:00
86 lines
3.0 KiB
Python
86 lines
3.0 KiB
Python
import tqdm
|
|
|
|
import urllib.request
|
|
import subprocess
|
|
import threading
|
|
|
|
from spotdl.encode.encoders import EncoderFFmpeg
|
|
from spotdl.metadata.embedders import EmbedderDefault
|
|
|
|
CHUNK_SIZE= 16 * 1024
|
|
|
|
class Track:
|
|
def __init__(self, metadata, cache_albumart=False):
|
|
self.metadata = metadata
|
|
self._chunksize = CHUNK_SIZE
|
|
|
|
self._cache_resources = {
|
|
"albumart": {"content": None, "threadinstance": None }
|
|
}
|
|
if cache_albumart:
|
|
self._albumart_thread = self._cache_albumart()
|
|
|
|
def _fetch_response_content_threaded(self, mutable_resource, url):
|
|
content = urllib.request.urlopen(url).read()
|
|
mutable_resource["content"] = content
|
|
|
|
def _cache_albumart(self):
|
|
# A hack to get a thread's return value
|
|
albumart_thread = threading.Thread(
|
|
target=self._fetch_response_content_threaded,
|
|
args=(self._cache_resources["albumart"],
|
|
self.metadata["album"]["images"][0]["url"]),
|
|
)
|
|
albumart_thread.start()
|
|
self._cache_resources["albumart"]["threadinstance"] = albumart_thread
|
|
|
|
def _calculate_total_chunks(self, filesize):
|
|
return (filesize // self._chunksize) + 1
|
|
|
|
def download_while_re_encoding(self, target_path, encoder=EncoderFFmpeg(), show_progress=True):
|
|
stream = self.metadata["streams"].getbest()
|
|
total_chunks = self._calculate_total_chunks(stream["filesize"])
|
|
process = encoder.re_encode_from_stdin(
|
|
stream["encoding"],
|
|
target_path
|
|
)
|
|
response = stream["connection"]
|
|
for _ in tqdm.trange(total_chunks):
|
|
chunk = response.read(self._chunksize)
|
|
process.stdin.write(chunk)
|
|
|
|
process.stdin.close()
|
|
process.wait()
|
|
|
|
def download(self, target_path, show_progress=True):
|
|
stream = self.metadata["streams"].getbest()
|
|
total_chunks = self._calculate_total_chunks(stream["filesize"])
|
|
response = stream["connection"]
|
|
with open(target_path, "wb") as fout:
|
|
for _ in tqdm.trange(total_chunks):
|
|
chunk = response.read(self._chunksize)
|
|
fout.write(chunk)
|
|
|
|
def re_encode(self, input_path, target_path, encoder=EncoderFFmpeg(), show_progress=True):
|
|
stream = self.metadata["streams"].getbest()
|
|
total_chunks = self._calculate_total_chunks(stream["filesize"])
|
|
process = encoder.re_encode_from_stdin(
|
|
stream["encoding"],
|
|
target_path
|
|
)
|
|
with open(input_path, "rb") as fin:
|
|
for _ in tqdm.trange(total_chunks):
|
|
chunk = fin.read(self._chunksize)
|
|
process.stdin.write(chunk)
|
|
|
|
process.stdin.close()
|
|
process.wait()
|
|
|
|
def apply_metadata(self, input_path, embedder=EmbedderDefault()):
|
|
albumart = self._cache_resources["albumart"]
|
|
if albumart["threadinstance"]:
|
|
albumart["threadinstance"].join()
|
|
|
|
embedder.apply_metadata(input_path, self.metadata, cached_albumart=albumart["content"])
|
|
|