mirror of
https://github.com/yt-dlp/yt-dlp.git
synced 2025-03-09 12:50:23 -05:00
65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
|
import hashlib
|
||
|
|
import random
|
||
|
|
import threading
|
||
|
|
|
||
|
|
from .common import FileDownloader
|
||
|
|
from . import HlsFD
|
||
|
|
from ..networking import Request
|
||
|
|
|
||
|
|
|
||
|
|
class BunnyCdnFD(FileDownloader):
|
||
|
|
"""Downloads from BunnyCDN with required pings"""
|
||
|
|
|
||
|
|
def real_download(self, filename, info_dict):
|
||
|
|
self.to_screen(f'[{self.FD_NAME}] Downloading from BunnyCDN')
|
||
|
|
|
||
|
|
fd = HlsFD(self.ydl, self.params)
|
||
|
|
|
||
|
|
success = download_complete = False
|
||
|
|
timer = [None]
|
||
|
|
ping_lock = threading.Lock()
|
||
|
|
current_time = [0]
|
||
|
|
|
||
|
|
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']
|
||
|
|
# Site sends ping every 4 seconds, but this throttles the download. Pinging every 2 seconds seems to work.
|
||
|
|
ping_interval = 2
|
||
|
|
|
||
|
|
def send_ping():
|
||
|
|
time = current_time[0] + 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()
|
||
|
|
|
||
|
|
request = Request(
|
||
|
|
f'{ping_url}?hash={md5_hash}&time={time}&paused={paused}&resolution={res}',
|
||
|
|
headers=headers,
|
||
|
|
)
|
||
|
|
|
||
|
|
try:
|
||
|
|
self.ydl.urlopen(request).read()
|
||
|
|
except Exception:
|
||
|
|
self.to_screen(f'[{self.FD_NAME}] Ping failed')
|
||
|
|
|
||
|
|
with ping_lock:
|
||
|
|
if not download_complete:
|
||
|
|
current_time[0] += ping_interval
|
||
|
|
timer[0] = threading.Timer(ping_interval, send_ping)
|
||
|
|
timer[0].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[0]:
|
||
|
|
timer[0].cancel()
|
||
|
|
download_complete = True
|
||
|
|
|
||
|
|
return success
|