diff --git a/apps/establishment/tasks.py b/apps/establishment/tasks.py index 2df4711d..332a3d7a 100644 --- a/apps/establishment/tasks.py +++ b/apps/establishment/tasks.py @@ -56,10 +56,14 @@ def update_establishment_image_urls(part_number: int, summary_tasks: int, bucket queryset = Establishment.objects.filter(id__in=bucket_ids) for establishment in queryset: - for data in [ + live_link = None + + image_urls = [ ('image_url', establishment.image_url), ('preview_image_url', establishment.preview_image_url) - ]: + ] + + for data in image_urls: attr, url = data if establishment.image_url is not None: @@ -69,11 +73,21 @@ def update_establishment_image_urls(part_number: int, summary_tasks: int, bucket if response.status_code != 200: setattr(establishment, attr, None) + else: + live_link = url + except ( requests.exceptions.ConnectionError, requests.exceptions.ConnectTimeout ): setattr(establishment, attr, None) + if live_link is not None: + if establishment.image_url is None: + establishment.image_url = live_link + + elif establishment.preview_image_url is None: + establishment.preview_image_url = live_link + logger.info(f'The {part_number}th part of the image update ' f'from {summary_tasks} parts was completed')