Diamond75/zotify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Cleaned up download real time feature implementation.

Browse source
This commit is contained in:
KDalu 2025-01-22 11:32:29 -05:00
parent 8d80719f6b
commit bf45889e6a
1 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
zotify/playable.py
View file

@ -215,6 +215,8 @@ class Track(PlayableContentFeeder.LoadedStream, Playable):
""" """
if not isinstance(output, Path): if not isinstance(output, Path):
output = Path(output).expanduser() output = Path(output).expanduser()
if not real_time:
return super().write_audio_stream(output)
file = f"{output}.ogg" file = f"{output}.ogg"
time_start = time() time_start = time()
downloaded = 0 downloaded = 0
@ -223,12 +225,11 @@ class Track(PlayableContentFeeder.LoadedStream, Playable):
while chunk != b"": while chunk != b"":
chunk = self.input_stream.stream().read(1024) chunk = self.input_stream.stream().read(1024)
p_bar.update(f.write(chunk)) p_bar.update(f.write(chunk))
if real_time: downloaded += len(chunk)
downloaded += len(chunk) delta_current = time() - time_start
delta_current = time() - time_start delta_required = (downloaded / self.input_stream.size) * (self.duration/1000)
delta_required = (downloaded / self.input_stream.size) * (self.duration/1000) if delta_required > delta_current:
if delta_required > delta_current: sleep(delta_required - delta_current)
sleep(delta_required - delta_current)
return LocalFile(Path(file), AudioFormat.VORBIS) return LocalFile(Path(file), AudioFormat.VORBIS)