Fix #15, add support for base64encoded strings as source

t.py

@@ -1,4 +1,5 @@
 # -*- coding: utf-8 -*-
+from tests import data
 import thumbnails
 from thumbnails.conf import settings
 
@@ -7,9 +8,10 @@
 
 print(settings.THUMBNAIL_PATH)
 
-print(thumbnails.get_thumbnail(URL, '800x800'))
-print(thumbnails.get_thumbnail(URL, '800'))
-print(thumbnails.get_thumbnail(URL, 'x800'))
-print(thumbnails.get_thumbnail(URL, '400x400', 'center'))
-print(thumbnails.get_thumbnail(URL, '400x400', 'top'))
-print(thumbnails.get_thumbnail(URL, '400x400', 'left'))
+print(thumbnails.get_thumbnail(URL, '800x800').url)
+print(thumbnails.get_thumbnail(URL, '800').url)
+print(thumbnails.get_thumbnail(URL, 'x800').url)
+print(thumbnails.get_thumbnail(URL, '400x400', crop='center').url)
+print(thumbnails.get_thumbnail(URL, '400x400', crop='top').url)
+print(thumbnails.get_thumbnail(URL, '400x400', crop='left').url)
+print(thumbnails.get_thumbnail(data.BASE64_STRING_OF_IMAGE, '200x200', crop='center').url)

tests/test_images.py

@@ -1,10 +1,14 @@
 # -*- coding: utf-8 -*-
+import hashlib
 import os
 import unittest
+from PIL import Image
+from thumbnails.compat import BytesIO
 
 from thumbnails.conf import settings
 from thumbnails.images import SourceFile, Thumbnail
 
+from . import data
 from .compat import mock
 from .utils import has_installed
 
@@ -64,3 +68,12 @@ def test_django_image_files(self):
         self.assertEqual(f.file, self.FILE_PATH)
         f = SourceFile(files.ImageFieldFile(field=field, instance=None, name=self.FILE_PATH))
         self.assertEqual(f.file, self.FILE_PATH)
+
+    def test_base64_encoded_string(self):
+        file = SourceFile(data.BASE64_STRING_OF_IMAGE)
+        self.assertEqual(
+            hashlib.sha1(file.open().getvalue()).hexdigest(),
+            '6666212f5302426c845ecb2a2901fae021735f24'
+        )
+        image = Image.open(BytesIO(file.open().read()))
+        self.assertIsNotNone(image.load())

thumbnails/__init__.py

@@ -10,7 +10,8 @@ def get_thumbnail(original, size, **options):
     Creates or gets an already created thumbnail for the given image with the given size and
     options.
 
-    :param original: File-path or url to the image that you want an thumbnail of.
+    :param original: File-path, url or base64-encoded string of the image that you want an
+                     thumbnail.
     :param size: String with the wanted thumbnail size. On the form: ``200x200``, ``200`` or
                  ``x200``.
 

thumbnails/images.py

@@ -1,7 +1,9 @@
 # -*- coding: utf-8 -*-
+import base64
 import os
 
 import requests
+from thumbnails.compat import BytesIO
 
 from thumbnails.conf import settings
 from thumbnails.helpers import get_engine, get_storage_backend
@@ -80,4 +82,6 @@ def __init__(self, source_file):
     def open(self):
         if self.file.startswith('http'):
             return requests.get(self.file, stream=True).raw
+        elif self.file.startswith(r'data:image/'):
+            return BytesIO(base64.b64decode(self.file.replace('data:image/jpeg;base64,', '')))
         return get_storage_backend().open(self.file)