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

Save accurate requested_formats

Browse Source
This commit is contained in:
tcely 2025-05-24 22:44:08 -04:00 committed by GitHub
parent 75aef2efdd
commit 68d7ced701
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tubesync/sync/hooks.py
View File

@ -3,6 +3,8 @@ import os
from common.logger import log from common.logger import log
from common.utils import remove_enclosed from common.utils import remove_enclosed
from .models import Media
progress_hook = { progress_hook = {
'status': dict(), 'status': dict(),
@ -49,7 +51,6 @@ class BaseStatus:
def update_task(self): def update_task(self):
if self.media_key is None: if self.media_key is None:
return return
from .models import Media
from .tasks import get_media_download_task from .tasks import get_media_download_task
media = task = None media = task = None
@ -224,6 +225,18 @@ def yt_dlp_postprocessor_hook(event):
status.update_task() status.update_task()
log.info(f'[{event["postprocessor"]}] {event["status"]} for: {name}') log.info(f'[{event["postprocessor"]}] {event["status"]} for: {name}')
if 'started' == event['status'] and 'Merger' == event['postprocessor']:
files_to_merge = event['info_dict'].get('__files_to_merge') or list()
log.info(f'[{event["postprocessor"]}] Files to merge: {files_to_merge}')
try:
media = Media.objects.get(pk=status.media_uuid)
media.new_metadata.value['requested_formats'] = event['info_dict'].get('requested_formats')
media.new_metadata.value['requested_subtitles'] = event['info_dict'].get('requested_subtitles')
media.new_metadata.save()
except Media.DoesNotExist:
pass
except Exception as e:
log.exception(e)
if 'finished' == event['status']: if 'finished' == event['status']:
status.cleanup() status.cleanup()