mirror/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2025-03-09 12:50:23 -05:00
Code Activity

Rework FileDownloader

Browse source 
Co-Authored-By: Simon Sawicki <contact@grub4k.xyz>
This commit is contained in:
sepro 2025-02-21 22:11:08 +01:00
parent fd226874f1
commit fbc6b6f026
1 changed files with 18 additions and 37 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

53
yt_dlp/downloader/bunnycdn.py
View file

@ -20,51 +20,32 @@ def real_download(self, filename, info_dict):
fd = HlsFD(self.ydl, self.params)
success = download_complete = False
ping_lock = threading.Lock()
stop_event = threading.Event()
ping_thread = threading.Thread(target=self.ping_thread, args=(stop_event,), kwargs=info_dict['_bunnycdn_ping_data'])
ping_thread.start()
timer = None
current_time = 0
try:
return fd.real_download(filename, info_dict)
finally:
stop_event.set()
ping_url = info_dict['_bunnycdn_ping_data']['url']
headers = info_dict['_bunnycdn_ping_data']['headers']
secret = info_dict['_bunnycdn_ping_data']['secret']
context_id = info_dict['_bunnycdn_ping_data']['context_id']
def ping_thread(self, stop_event, url, headers, secret, context_id):
# Site sends ping every 4 seconds, but this throttles the download. Pinging every 2 seconds seems to work.
ping_interval = 2
def send_ping():
nonlocal timer, current_time
time = current_time + round(random.random(), 6)
# Hard coded resolution as it doesn't seem to matter
res = 1080
paused = 'false'
md5_hash = hashlib.md5(f'{secret}_{context_id}_{time}_{paused}_{res}'.encode()).hexdigest()
current_time = 0
request = Request(
f'{ping_url}?hash={md5_hash}&time={time}&paused={paused}&resolution={res}',
headers=headers)
while not stop_event.wait(ping_interval):
current_time += ping_interval
time = current_time + round(random.random(), 6)
md5_hash = hashlib.md5(f'{secret}_{context_id}_{time}_{paused}_{res}'.encode()).hexdigest()
ping_url = f'{url}?hash={md5_hash}&time={time}&paused={paused}&resolution={res}'
try:
self.ydl.urlopen(request).read()
self.ydl.urlopen(Request(ping_url, headers=headers)).read()
except network_exceptions as e:
self.to_screen(f'[{self.FD_NAME}] Ping failed: {e}')
with ping_lock:
if not download_complete:
current_time += ping_interval
timer = threading.Timer(ping_interval, send_ping)
timer.start()
# Start ping loop
self.to_screen(f'[{self.FD_NAME}] Starting pings with {ping_interval} second interval...')
try:
send_ping()
success = fd.real_download(filename, info_dict)
finally:
with ping_lock:
if timer:
timer.cancel()
download_complete = True
return success