def test_resize_no_upscaling(self):
"""
Ensure upscaling is forbidden in resize action.
"""
renderer = LazyThumbRenderer()
mock_img = MockImg()
img = renderer.resize(width=2000, height=2000, img=mock_img)
self.assertEqual(img.size[0], 1000)
self.assertEqual(img.size[1], 1000)
self.assertEqual(len(mock_img.called), 0)
def test_resize(self):
"""
Test behavior of resize action.
"""
renderer = LazyThumbRenderer()
mock_img = MockImg()
img = renderer.resize(width=48, height=50, img=mock_img)
self.assertEqual(img.size[1], 50)
self.assertEqual(len(mock_img.called), 2)
self.assertTrue('crop' in mock_img.called)
self.assertTrue('resize' in mock_img.called)
def test_thumbnail_noop(self):
"""
Test that no image operations occur if the desired w/h match image's
existing w/h
"""
renderer = LazyThumbRenderer()
mock_img = MockImg()
mock_img.size = (100, 100)
img = renderer.thumbnail(width=100, img=mock_img)
self.assertEqual(img.size[0], 100)
self.assertEqual(img.size[1], 100)
self.assertEqual(len(mock_img.called), 0)
def test_thumbnail_no_upscaling(self):
"""
Ensure that upscaling is forbidden in thumbnail action.
"""
renderer = LazyThumbRenderer()
mock_img = MockImg()
mock_img.size = (100, 100)
img = renderer.thumbnail(width=20000, img=mock_img)
self.assertEqual(img.size[0], 100)
self.assertEqual(img.size[1], 100)
self.assertEqual(len(mock_img.called), 0)
def test_thumbnail_square(self):
"""
Test behavior of thumbnail action when no width == height
"""
renderer = LazyThumbRenderer()
mock_img = MockImg()
mock_img.size = (100, 100)
img = renderer.thumbnail(width=50, img=mock_img)
self.assertEqual(img.size[0], 50)
self.assertEqual(img.size[1], 50)
self.assertEqual(len(mock_img.called), 1)
self.assertTrue('resize' in mock_img.called)
def test_aresize_height(self):
"""
Ensure landscape aresize action thumbnails to height and center-crops.
"""
renderer = LazyThumbRenderer()
mock_img = MockImg(width=2000, height=1000)
mock_Image = Mock()
# aresize 2000x1000 => 1000x800 should thumbnail to 1600x800 and then
# paste into the center of a new image, losing some left/right content.
with patch('lazythumbs.views.Image', mock_Image):
img = renderer.aresize(width=1000, height=800, img=mock_img)
self.assertEqual(mock_img.called, ['resize'])
self.assertEqual(img, mock_Image.new.return_value)
self.assertEqual(mock_Image.new.call_args[1]['size'], (1000, 800))
img.paste.assert_called_once_with(mock_img, (-300, 0))
def test_aresize_width(self):
"""
Ensure landscape aresize action thumbnails to width and center-crops.
"""
renderer = LazyThumbRenderer()
mock_img = MockImg(width=1500, height=1000)
mock_Image = Mock()
# aresize 1500x1000 => 750x400 should thumbnail to 750x500 and then
# paste into the center of a new image, losing some top/bottom content.
with patch('lazythumbs.views.Image', mock_Image):
img = renderer.aresize(width=750, height=400, img=mock_img)
self.assertEqual(mock_img.called, ['resize'])
self.assertEqual(img, mock_Image.new.return_value)
self.assertEqual(mock_Image.new.call_args[1]['size'], (750, 400))
img.paste.assert_called_once_with(mock_img, (0, -50))
def test_aresize_no_crop_small(self):
"""
Ensure aresize_no_crop action does not grow a small image.
"""
renderer = LazyThumbRenderer()
mock_img = MockImg(width=100, height=80)
mock_Image = Mock()
# aresize_no_crop 100x80 => 300x200 should not resize, but should paste
# into the center of a new image, leaving matte background content
# on the sides.
with patch('lazythumbs.views.Image', mock_Image):
img = renderer.aresize(width=300, height=200, img=mock_img)
self.assertEqual(mock_img.called, [])
self.assertEqual(img, mock_Image.new.return_value)
self.assertEqual(mock_Image.new.call_args[1]['size'], (300, 200))
img.paste.assert_called_once_with(mock_img, (100, 60))
def test_aresize_portrait(self):
"""
Ensure aresize action from portrait to landscape shrinks image and
mattes sides.
"""
renderer = LazyThumbRenderer()
mock_img = MockImg(width=1000, height=2000)
mock_Image = Mock()
# aresize 1000x2000 => 600x500 should thumbnail to 250x500 and then
# paste into the center of a new image, leaving matte background
# content on the sides.
with patch('lazythumbs.views.Image', mock_Image):
img = renderer.aresize(width=600, height=500, img=mock_img)
self.assertEqual(mock_img.called, ['resize'])
self.assertEqual(img, mock_Image.new.return_value)
self.assertEqual(mock_Image.new.call_args[1]['size'], (600, 500))
img.paste.assert_called_once_with(mock_img, (175, 0))
def test_aresize_no_crop_opposite_orientation_landscape(self):
"""
Ensure aresize_no_crop action when the source and target are the
opposite orientation, and the source is landscape, that it scales
according to width and mattes the top and bottom.
"""
renderer = LazyThumbRenderer()
mock_img = MockImg(width=2000, height=1000)
mock_Image = Mock()
# aresize_no_crop 2000x1000 => 500x600 should thumbnail to 500x250 and
# then paste into the center of a new image, leaving matte background
# content on the top and bottom.
with patch('lazythumbs.views.Image', mock_Image):
img = renderer.aresize_no_crop(width=500, height=600, img=mock_img)
self.assertEqual(mock_img.called, ['resize'])
self.assertEqual(img, mock_Image.new.return_value)
self.assertEqual(mock_Image.new.call_args[1]['size'], (500, 600))
img.paste.assert_called_once_with(mock_img, (0, 175))
def test_aresize_no_crop_opposite_orientation_portrait(self):
"""
Ensure aresize_no_crop action when the source and target are the
opposite orientation, and the source is portrait, that it scales
according to height and mattes sides.
"""
renderer = LazyThumbRenderer()
mock_img = MockImg(width=1000, height=2000)
mock_Image = Mock()
# aresize_no_crop 1000x2000 => 600x500 should thumbnail to 250x500 and
# then paste into the center of a new image, leaving matte background
# content on the sides.
with patch('lazythumbs.views.Image', mock_Image):
img = renderer.aresize_no_crop(width=600, height=500, img=mock_img)
self.assertEqual(mock_img.called, ['resize'])
self.assertEqual(img, mock_Image.new.return_value)
self.assertEqual(mock_Image.new.call_args[1]['size'], (600, 500))
img.paste.assert_called_once_with(mock_img, (175, 0))
def test_extensionless_gif(self):
"""If the image file is a GIF without an extension, we can produce
a valid thumbnail for it."""
# Note: this test image file was breaking thumbnails if lazythumbs didn't
# see the .gif extension. I tried creating a gif on the fly using
# PIL but didn't hit the same problem, so it might be something about
# this image that's special, maybe that it has a transparent background.
# (The error was "cannot write mode P as JPEG"; the symptom was a 404
# response.)
MEDIA_ROOT = tempfile.gettempdir()
# Need to override MEDIA_ROOT in both django file storage and lazythumbs views
# and Django doesn't provide override_settings until 1.4
with patch('django.core.files.storage.settings') as settings1:
settings1.MEDIA_ROOT = MEDIA_ROOT
with patch('lazythumbs.views.settings') as settings2:
settings2.MEDIA_ROOT = MEDIA_ROOT
testfile = TEST_IMG_GIF
filename = None
try:
filename = os.path.join(MEDIA_ROOT, "gif_without_extension")
shutil.copy(testfile, filename)
# Now we have a gif file in a filename that doesn't end in .gif
renderer = LazyThumbRenderer()
source_path = os.path.relpath(filename, MEDIA_ROOT)
rsp = renderer.get(
request=Mock(path="/thumbnail/x50/" + source_path),
action="thumbnail",
geometry="x50",
source_path=source_path
)
# if you get 404, jpeg encoder is probably missing for Pillow
self.assertEqual(200, rsp.status_code)
finally:
if filename:
os.remove(filename)
"""
{% lazythumb image.url thumbnail '48' as img_tag %}
<img src="{{img_tag.src}}" width="{{img_tag.width}}" height="{{img_tag.height}} />
{% endlazythumb %}
{% lazythumb image.url resize '150x200' %}
<img src="{{img_tag.src}}" width="{{img_tag.width}}" height="{{img_tag.height}} />
{% endlazythumb %}
"""
import logging
from django.template import TemplateSyntaxError, Library, Node, Variable
from lazythumbs.util import compute_img, get_attr_string
from lazythumbs.views import LazyThumbRenderer
SUPPORTED_ACTIONS = LazyThumbRenderer().allowed_actions
register = Library()
logger = logging.getLogger()
register.tag('lazythumb', lambda p, t: LazythumbNode(p, t))
class LazythumbNode(Node):
usage = 'Expected invocation is {% lazythumb url|ImageFile|Object action geometry [**kwargs] as variable %}'
def __init__(self, parser, token):
# simple alias
tse = lambda m: TemplateSyntaxError('lazythumb: %s' % m)
bits = token.contents.split()
try:
def test_thumbnail_no_img(self):
renderer = LazyThumbRenderer()
self.assertRaises(ValueError, renderer.thumbnail, 200, 200)
def test_thumbnail_no_width_and_no_height_specified(self):
renderer = LazyThumbRenderer()
mock_img = MockImg()
mock_img.size = (100, 100)
self.assertRaises(ValueError, renderer.thumbnail, img=mock_img)
def test_no_img(self):
renderer = LazyThumbRenderer()
self.assertRaises(ValueError, renderer.scale, 200, 200)
def test_maximum_width_and_height(self):
renderer = LazyThumbRenderer()
new_img = renderer.scale(1000, 1000, img_path=TEST_IMG_GIF)
self.assertEqual(new_img.size, (399, 499))
def test_new_img(self):
renderer = LazyThumbRenderer()
new_img = renderer.matte(200, 200, img_path=TEST_IMG_GIF)
self.assertEqual(new_img.size, (200, 200))
def setUp(self):
self.renderer = LazyThumbRenderer()
self.mock_Image = Mock()
self.mock_img = Mock()
self.mock_Image.open = Mock(return_value=self.mock_img)
self.mock_img.size = [1,1]
