Fix long names thumbnail

easy_thumbnails/files.py

@@ -1,3 +1,4 @@
+import hashlib
 import os
 
 from django.core.files.base import File, ContentFile
@@ -450,6 +451,18 @@ def get_thumbnail_name(self, thumbnail_options, transparent=False,
             thumbnail_options=thumbnail_options,
             prepared_options=prepared_opts,
         )
+        max_length = models.File._meta.get_field('name').max_length
+        if len(filename) > max_length:
+            length_total = len(filename) + len(self.thumbnail_highres_infix) + len(self.thumbnail_prefix)
+            additional_length = length_total + 32 - max_length
+            filename = namer_func(
+                thumbnailer=self,
+                source_filename='%s%s' % (source_filename[:-additional_length],
+                                          hashlib.md5(source_filename.encode('utf-8')).hexdigest()),
+                thumbnail_extension=extension,
+                thumbnail_options=thumbnail_options,
+                prepared_options=prepared_opts,
+            )
         if high_resolution:
             filename = self.thumbnail_highres_infix.join(
                 os.path.splitext(filename))

easy_thumbnails/tests/test_files.py

@@ -24,17 +24,31 @@ def setUp(self):
         # Save a test image in both storages.
         filename = self.create_image(self.storage, 'test.jpg')
         self.thumbnailer = files.get_thumbnailer(self.storage, filename)
+        # Save a test image with a really long name in both storages
+        long_filename = self.create_image(self.storage, '%s.jpg' % ('*' * 251))
+        self.longname_thumbnailer = files.get_thumbnailer(self.storage, long_filename)
+
+        self.longname_thumbnailer.thumbnail_storage = self.storage
         self.thumbnailer.thumbnail_storage = self.storage
 
         filename = self.create_image(self.remote_storage, 'test.jpg')
         self.remote_thumbnailer = files.get_thumbnailer(
             self.remote_storage, filename)
+        long_filename = self.create_image(self.remote_storage, '%s.jpg' % ('*' * 251))
+        self.longname_remote_thumbnailer = files.get_thumbnailer(
+            self.remote_storage, long_filename)
+
         self.remote_thumbnailer.thumbnail_storage = self.remote_storage
+        self.longname_remote_thumbnailer.thumbnail_storage = self.remote_storage
 
         # Create another thumbnailer for extension test.
         self.ext_thumbnailer = files.get_thumbnailer(self.storage, filename)
         self.ext_thumbnailer.thumbnail_storage = self.storage
 
+        # Create another thumbnailer with long filename for extension test.
+        self.longname_ext_thumbnailer = files.get_thumbnailer(self.storage, long_filename)
+        self.longname_ext_thumbnailer.thumbnail_storage = self.storage
+
         # Generate test transparent images.
         filename = self.create_image(
             self.storage, 'transparent.png', image_mode='RGBA',
@@ -92,6 +106,63 @@ def test_tag(self):
             '<img alt="" class="fish" height="75" rel="A&amp;B" '
             'src="%s" width="100" />' % local.url)
 
+    def test_tag_long_filename_multiple_same_names(self):
+        local = self.longname_thumbnailer.get_thumbnail({'size': (100, 100)})
+        long_filename_2 = self.create_image(self.storage, '%s.jpg' % ('*' * 250))
+        longname_thumbnailer_2 = files.get_thumbnailer(self.storage, long_filename_2)
+        longname_thumbnailer_2.thumbnail_storage = self.storage
+        local_2 = longname_thumbnailer_2.get_thumbnail({'size': (100, 100)})
+        self.assertEqual(
+            local.tag(), '<img alt="" height="75" src="%s" width="100" '
+                         '/>' % local.url)
+        self.assertEqual(
+            local.tag(alt='A & B'), '<img alt="A &amp; B" height="75" '
+                                    'src="%s" width="100" />' % local.url)
+        self.assertEqual(
+            local_2.tag(), '<img alt="" height="75" src="%s" width="100" '
+                           '/>' % local_2.url)
+        self.assertEqual(
+            local_2.tag(alt='A & B'), '<img alt="A &amp; B" height="75" '
+                                      'src="%s" width="100" />' % local_2.url)
+        self.assertNotEqual(local_2.url, local.url)
+
+    def test_tag_long_filename(self):
+        local = self.longname_thumbnailer.get_thumbnail({'size': (100, 100)})
+        remote = self.longname_remote_thumbnailer.get_thumbnail({'size': (100, 100)})
+
+        self.assertEqual(
+            local.tag(), '<img alt="" height="75" src="%s" width="100" '
+                         '/>' % local.url)
+        self.assertEqual(
+            local.tag(alt='A & B'), '<img alt="A &amp; B" height="75" '
+                                    'src="%s" width="100" />' % local.url)
+
+        # Can turn off dimensions.
+        self.assertEqual(
+            remote.tag(use_size=False), '<img alt="" src="%s" />' % remote.url)
+
+        # Even a remotely generated thumbnail has the dimensions cached if it
+        # was just created.
+        self.assertEqual(
+            remote.tag(),
+            '<img alt="" height="75" src="%s" width="100" />' % remote.url)
+
+        # Future requests to thumbnails on remote storage don't get
+        # dimensions...
+        remote = self.longname_remote_thumbnailer.get_thumbnail({'size': (100, 100)})
+        self.assertEqual(
+            remote.tag(), '<img alt="" src="%s" />' % remote.url)
+        # ...unless explicitly requested.
+        self.assertEqual(
+            remote.tag(use_size=True),
+            '<img alt="" height="75" src="%s" width="100" />' % remote.url)
+
+        # All other arguments are passed through as attributes.
+        self.assertEqual(
+            local.tag(**{'rel': 'A&B', 'class': 'fish'}),
+            '<img alt="" class="fish" height="75" rel="A&amp;B" '
+            'src="%s" width="100" />' % local.url)
+
     def test_tag_cached_dimensions(self):
         settings.THUMBNAIL_CACHE_DIMENSIONS = True
         self.remote_thumbnailer.get_thumbnail({'size': (100, 100)})
@@ -237,6 +308,23 @@ def test_high_resolution_force(self):
         with Image.open(hires_thumb_file) as thumb:
             self.assertEqual(thumb.size, (200, 150))
 
+    def test_high_resolution_force_off_with_longname(self):
+        self.longname_ext_thumbnailer.thumbnail_high_resolution = True
+        thumb = self.longname_ext_thumbnailer.get_thumbnail(
+            {'size': (100, 100), 'HIGH_RESOLUTION': False})
+        base, ext = path.splitext(thumb.path)
+        hires_thumb_file = ''.join([base + '@2x', ext])
+        self.assertFalse(path.exists(hires_thumb_file))
+
+    def test_high_resolution_force_with_longname(self):
+        thumb = self.longname_ext_thumbnailer.get_thumbnail(
+            {'size': (100, 100), 'HIGH_RESOLUTION': True})
+        base, ext = path.splitext(thumb.path)
+        hires_thumb_file = ''.join([base + '@2x', ext])
+        self.assertTrue(path.isfile(hires_thumb_file))
+        with Image.open(hires_thumb_file) as thumb:
+            self.assertEqual(thumb.size, (200, 150))
+
     def test_highres_infix(self):
         self.ext_thumbnailer.thumbnail_high_resolution = True
         self.ext_thumbnailer.thumbnail_highres_infix = '_2x'
@@ -247,6 +335,16 @@ def test_highres_infix(self):
         with Image.open(hires_thumb_file) as thumb:
             self.assertEqual(thumb.size, (200, 150))
 
+    def test_highres_infix_with_longname(self):
+        self.longname_ext_thumbnailer.thumbnail_high_resolution = True
+        self.longname_ext_thumbnailer.thumbnail_highres_infix = '_2x'
+        thumb = self.longname_ext_thumbnailer.get_thumbnail({'size': (100, 100)})
+        base, ext = path.splitext(thumb.path)
+        hires_thumb_file = ''.join([base + '_2x', ext])
+        self.assertTrue(path.isfile(hires_thumb_file))
+        with Image.open(hires_thumb_file) as thumb:
+            self.assertEqual(thumb.size, (200, 150))
+
     @unittest.skipIf(
         'easy_thumbnails.optimize' not in settings.INSTALLED_APPS,
         'optimize app not installed')