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

Reduce the memory usage

Browse Source
This commit is contained in:
tcely 2025-04-12 09:22:25 -04:00 committed by GitHub
parent ba63cece20
commit ef1dd0eba8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tubesync/sync/tasks.py
View File

@ -674,7 +674,10 @@ def save_all_media_for_source(source_id):
raise InvalidTaskError(_('no such source')) from e raise InvalidTaskError(_('no such source')) from e
saved_later = set() saved_later = set()
mqs = Media.objects.filter(source=source) mqs = Media.objects.all().defer(
'metadata',
'thumb',
).filter(source=source)
task = get_source_check_task(source_id) task = get_source_check_task(source_id)
refresh_qs = mqs.filter( refresh_qs = mqs.filter(
can_download=False, can_download=False,
@ -701,11 +704,17 @@ def save_all_media_for_source(source_id):
# Trigger the post_save signal for each media item linked to this source as various # Trigger the post_save signal for each media item linked to this source as various
# flags may need to be recalculated # flags may need to be recalculated
tvn_format = '2/{:,}' + f'/{mqs.count():,}' tvn_format = '2/{:,}' + f'/{mqs.count():,}'
for mn, media in enumerate(mqs, start=1): for mn, media_uuid in enumerate(mqs.values_list('uuid', flat=True), start=1):
if media.uuid not in saved_later: if media_uuid not in saved_later:
update_task_status(task, tvn_format.format(mn)) update_task_status(task, tvn_format.format(mn))
with atomic(): try:
media.save() media = Media.objects.get(pk=str(media_uuid))
except Media.DoesNotExist as e:
log.exception(str(e))
pass
else:
with atomic():
media.save()
# Reset task.verbose_name to the saved value # Reset task.verbose_name to the saved value
update_task_status(task, None) update_task_status(task, None)