recalculate media skip flag when source download cap is updated, resolves #77

This commit is contained in:
meeb 2021-03-07 14:03:19 +11:00
parent 972c184c70
commit 2e0d0385b0
1 changed files with 33 additions and 4 deletions

View File

@ -93,18 +93,47 @@ def task_task_failed(sender, task_id, completed_task, **kwargs):
@receiver(post_save, sender=Media) @receiver(post_save, sender=Media)
def media_post_save(sender, instance, created, **kwargs): def media_post_save(sender, instance, created, **kwargs):
# Triggered after media is saved, Recalculate the "can_download" flag, this may # Triggered after media is saved
cap_changed = False
can_download_changed = False
# Reset the skip flag if the download cap has changed if the media has not
# already been downloaded
if not instance.downloaded:
max_cap_age = instance.source.download_cap_date
if max_cap_age:
if instance.published > max_cap_age and instance.skip:
# Media was published after the cap date but is set to be skipped
log.info(f'Media: {instance.source} / {instance} has a valid '
f'publishing date, marking to be unskipped')
instance.skip = False
cap_changed = True
elif instance.published <= max_cap_age and not instance.skip:
log.info(f'Media: {instance.source} / {instance} is too old for the '
f'download cap date, marking to be skipped')
instance.skip = True
cap_changed = True
else:
if instance.skip:
# Media marked to be skipped but source download cap removed
log.info(f'Media: {instance.source} / {instance} has a valid '
f'publishing date, marking to be unskipped')
instance.skip = False
cap_changed = True
# Recalculate the "can_download" flag, this may
# need to change if the source specifications have been changed # need to change if the source specifications have been changed
if instance.metadata: if instance.metadata:
post_save.disconnect(media_post_save, sender=Media)
if instance.get_format_str(): if instance.get_format_str():
if not instance.can_download: if not instance.can_download:
instance.can_download = True instance.can_download = True
instance.save() can_download_changed = True
else: else:
if instance.can_download: if instance.can_download:
instance.can_download = False instance.can_download = False
instance.save() can_download_changed = True
# Save the instance if any changes were required
if cap_changed or can_download_changed:
post_save.disconnect(media_post_save, sender=Media)
instance.save()
post_save.connect(media_post_save, sender=Media) post_save.connect(media_post_save, sender=Media)
# If the media is missing metadata schedule it to be downloaded # If the media is missing metadata schedule it to be downloaded
if not instance.metadata: if not instance.metadata: