hoelee/tubesync
1
0
Fork 0
mirror of https://github.com/meeb/tubesync.git synced 2025-06-23 13:36:35 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Switch from key to filename

Browse Source 
When downloading subtitles, key isn't available.
This commit is contained in:
tcely 2025-02-07 13:04:23 -05:00 committed by GitHub
parent 034110e6fe
commit 074230b737
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tubesync/sync/hooks.py
View File

@ -58,9 +58,6 @@ class ProgressHookStatus(BaseStatus):
self.status = status self.status = status
self.download_progress = 0 self.download_progress = 0
def __str__(self):
return f'{self!r}: {self.status} ({self.download_progress}) file: {self.filename}'
def next_progress(self): def next_progress(self):
if 0 == self.download_progress: if 0 == self.download_progress:
return 0 return 0
@ -87,7 +84,7 @@ def yt_dlp_progress_hook(event):
log.warn(f'[youtube-dl] unknown progress event: {str(event)}') log.warn(f'[youtube-dl] unknown progress event: {str(event)}')
return None return None
key = 'Unknown' key = None
if 'display_id' in event['info_dict']: if 'display_id' in event['info_dict']:
key = event['info_dict']['display_id'] key = event['info_dict']['display_id']
elif 'id' in event['info_dict']: elif 'id' in event['info_dict']:
@ -97,12 +94,11 @@ def yt_dlp_progress_hook(event):
if 'error' == event['status']: if 'error' == event['status']:
log.error(f'[youtube-dl] error occured downloading: {filename}') log.error(f'[youtube-dl] error occured downloading: {filename}')
elif 'downloading' == event['status']: elif 'downloading' == event['status']:
# get or create the status for key # get or create the status for filename
status = ProgressHookStatus.get(key) status = ProgressHookStatus.get(filename)
if status is None: if status is None:
status = ProgressHookStatus(**event) status = ProgressHookStatus(**event)
status.register(key, filename, status.filename) status.register(key, filename, status.filename)
log.info(str(ProgressHookStatus.status_dict))
downloaded_bytes = event.get('downloaded_bytes', 0) or 0 downloaded_bytes = event.get('downloaded_bytes', 0) or 0
total_bytes_estimate = event.get('total_bytes_estimate', 0) or 0 total_bytes_estimate = event.get('total_bytes_estimate', 0) or 0
@ -123,12 +119,11 @@ def yt_dlp_progress_hook(event):
log.info(f'[youtube-dl] downloading: {filename} - {percent_str} ' log.info(f'[youtube-dl] downloading: {filename} - {percent_str} '
f'of {total} at {speed}, {eta} remaining') f'of {total} at {speed}, {eta} remaining')
elif 'finished' == event['status']: elif 'finished' == event['status']:
# update the status for key to the finished value # update the status for filename to the finished value
status = ProgressHookStatus.get(key) status = ProgressHookStatus.get(filename)
if status is None: if status is None:
status = ProgressHookStatus(**event) status = ProgressHookStatus(**event)
status.register(key, filename, status.filename) status.register(key, filename, status.filename)
log.info(str(ProgressHookStatus.status_dict))
status.download_progress = 100 status.download_progress = 100
total_size_str = event.get('_total_bytes_str', '?').strip() total_size_str = event.get('_total_bytes_str', '?').strip()