send_locally_feed.py: use profile images

This commit is contained in:
2020-02-19 13:07:23 -08:00
parent 908793d077
commit dc01f894f7
2 changed files with 7 additions and 5 deletions

View File

@ -92,9 +92,9 @@ class ProductImage:
# return path
@staticmethod
def get_all_image_paths(sap, image_format):
def get_all_image_paths(sap, image_format, view_types=VIEW_TYPES):
paths = []
for view in ProductImage.VIEW_TYPES:
for view in view_types:
path = os.path.join(ProductImage.THUMB_DIR,
'{}_{}-{}'.format(sap, view, image_format.value))
if os.path.isfile(path):
@ -108,8 +108,8 @@ class ProductImage:
# return path.replace(ProductImage.THUMB_DIR, ProductImage.THUMB_URL)
@staticmethod
def get_all_image_urls(sap, image_format, full_url=False):
paths = ProductImage.get_all_image_paths(sap, image_format)
def get_all_image_urls(sap, image_format, view_types=VIEW_TYPES, full_url=False):
paths = ProductImage.get_all_image_paths(sap, image_format, view_types)
url_prefix = ProductImage.THUMB_URL
if full_url:
url_prefix = f'https://{settings.PUBLIC_WEB_HOST}{url_prefix}'