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

Use django_queryset_generator in cleanup_old_media

Browse Source
This commit is contained in:
tcely 2025-04-13 11:24:15 -04:00 committed by GitHub
parent 29e94427c9
commit 950cdbd848
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tubesync/sync/tasks.py
View File

@ -27,7 +27,8 @@ from background_task.models import Task, CompletedTask
from common.logger import log from common.logger import log
from common.errors import ( NoFormatException, NoMediaException, from common.errors import ( NoFormatException, NoMediaException,
NoMetadataException, DownloadFailedException, ) NoMetadataException, DownloadFailedException, )
from common.utils import json_serial, remove_enclosed from common.utils import ( django_queryset_generator as qa_gen,
json_serial, remove_enclosed, )
from .choices import Val, TaskQueue from .choices import Val, TaskQueue
from .models import Source, Media, MediaServer from .models import Source, Media, MediaServer
from .utils import ( get_remote_image, resize_image_to_height, delete_file, from .utils import ( get_remote_image, resize_image_to_height, delete_file,
@ -215,7 +216,7 @@ def schedule_media_servers_update():
def cleanup_old_media(): def cleanup_old_media():
with atomic(): with atomic():
for source in Source.objects.filter(delete_old_media=True, days_to_keep__gt=0): for source in qs_gen(Source.objects.filter(delete_old_media=True, days_to_keep__gt=0)):
delta = timezone.now() - timedelta(days=source.days_to_keep) delta = timezone.now() - timedelta(days=source.days_to_keep)
mqs = source.media_source.defer( mqs = source.media_source.defer(
'metadata', 'metadata',
@ -223,7 +224,7 @@ def cleanup_old_media():
downloaded=True, downloaded=True,
download_date__lt=delta, download_date__lt=delta,
) )
for media in mqs: for media in qs_gen(mqs):
log.info(f'Deleting expired media: {source} / {media} ' log.info(f'Deleting expired media: {source} / {media} '
f'(now older than {source.days_to_keep} days / ' f'(now older than {source.days_to_keep} days / '
f'download_date before {delta})') f'download_date before {delta})')