def _uploader(resize_size, final_size): img = get_image_path('mozilla.png') original_size = (82, 31) src = tempfile.NamedTemporaryFile(mode='r+w+b', suffix=".png", delete=False) # resize_icon removes the original shutil.copyfile(img, src.name) src_image = Image.open(src.name) eq_(src_image.size, original_size) if isinstance(final_size, list): for rsize, fsize in zip(resize_size, final_size): dest_name = os.path.join(settings.ADDON_ICONS_PATH, '1234') tasks.resize_icon(src.name, dest_name, resize_size, locally=True) dest_image = Image.open("%s-%s.png" % (dest_name, rsize)) eq_(dest_image.size, fsize) if os.path.exists(dest_image.filename): os.remove(dest_image.filename) assert not os.path.exists(dest_image.filename) else: dest = tempfile.NamedTemporaryFile(mode='r+w+b', suffix=".png") tasks.resize_icon(src.name, dest.name, resize_size, locally=True) dest_image = Image.open(dest.name) eq_(dest_image.size, final_size) assert not os.path.exists(src.name)
def _uploader(resize_size, final_size): img = get_image_path('mozilla.png') original_size = (82, 31) src = tempfile.NamedTemporaryFile(mode='r+w+b', suffix=".png", delete=False) # resize_icon removes the original shutil.copyfile(img, src.name) src_image = Image.open(src.name) eq_(src_image.size, original_size) if isinstance(final_size, list): for rsize, fsize in zip(resize_size, final_size): dest_name = os.path.join(settings.ADDON_ICONS_PATH, '1234') tasks.resize_icon(src.name, dest_name, resize_size) dest_image = Image.open("%s-%s.png" % (dest_name, rsize)) eq_(dest_image.size, fsize) if os.path.exists(dest_image.filename): os.remove(dest_image.filename) assert not os.path.exists(dest_image.filename) else: dest = tempfile.NamedTemporaryFile(mode='r+w+b', suffix=".png") tasks.resize_icon(src.name, dest.name, resize_size) dest_image = Image.open(dest.name) eq_(dest_image.size, final_size) assert not os.path.exists(src.name)
def _uploader(resize_size, final_size): img = get_image_path('mozilla.png') original_size = (339, 128) src = tempfile.NamedTemporaryFile(mode='r+w+b', suffix=".png", delete=False) # resize_icon removes the original shutil.copyfile(img, src.name) with storage.open(src.name) as fp: src_image = Image.open(fp) src_image.load() eq_(src_image.size, original_size) if isinstance(final_size, list): for rsize, fsize in zip(resize_size, final_size): dest_name = os.path.join(settings.ADDON_ICONS_PATH, '1234') tasks.resize_icon(src.name, dest_name, resize_size, locally=True) with storage.open("%s-%s.png" % (dest_name, rsize)) as fp: dest_image = Image.open(fp) dest_image.load() # Assert that the width is always identical. eq_(dest_image.size[0], fsize[0]) # Assert that the height can be a wee bit fuzzy. assert -1 <= dest_image.size[1] - fsize[1] <= 1, ( "Got width %d, expected %d" % (fsize[1], dest_image.size[1])) if os.path.exists(dest_image.filename): os.remove(dest_image.filename) assert not os.path.exists(dest_image.filename) else: dest = tempfile.NamedTemporaryFile(mode='r+w+b', suffix=".png") tasks.resize_icon(src.name, dest.name, resize_size, locally=True) with storage.open(dest.name) as fp: dest_image = Image.open(fp) dest_image.load() # Assert that the width is always identical. eq_(dest_image.size[0], final_size[0]) # Assert that the height can be a wee bit fuzzy. assert -1 <= dest_image.size[1] - final_size[1] <= 1, ( "Got width %d, expected %d" % (final_size[1], dest_image.size[1])) assert not os.path.exists(src.name)
def _uploader(resize_size, final_size): img = get_image_path('mozilla.png') original_size = (339, 128) for rsize, fsize in zip(resize_size, final_size): dest_name = os.path.join(settings.ADDON_ICONS_PATH, '1234') src = tempfile.NamedTemporaryFile(mode='r+w+b', suffix='.png', delete=False) # resize_icon removes the original, copy it to a tempfile and use that. shutil.copyfile(img, src.name) # Sanity check. with storage.open(src.name) as fp: src_image = Image.open(fp) src_image.load() eq_(src_image.size, original_size) val = tasks.resize_icon(src.name, dest_name, resize_size, locally=True) eq_(val, {'icon_hash': 'bb362450'}) with storage.open('%s-%s.png' % (dest_name, rsize)) as fp: dest_image = Image.open(fp) dest_image.load() # Assert that the width is always identical. eq_(dest_image.size[0], fsize[0]) # Assert that the height can be a wee bit fuzzy. assert -1 <= dest_image.size[1] - fsize[1] <= 1, ( 'Got width %d, expected %d' % ( fsize[1], dest_image.size[1])) if os.path.exists(dest_image.filename): os.remove(dest_image.filename) assert not os.path.exists(dest_image.filename) assert not os.path.exists(src.name)