added image processing

Author: Jackywathy

back_end/image.py

@@ -0,0 +1,29 @@
+from PIL import Image
+import base64
+from io import BytesIO
+import imghdr
+
+
+iconsize = 256
+
+def open_image(data, b64=False):
+    """Returns an image object, given image binary data (gotten from request.get_file) """
+    if b64:
+        data = base64.b64decode(data)
+
+    io = BytesIO(data)
+
+    image_type = imghdr.what(io)
+    if image_type:
+        return (Image.open(io), image_type)
+    else:
+        return None
+
+
+def to_icon(img):
+    """Takes an Image object, and mutates it into a thumbname with the size iconsize"""
+    reducerate = 1 # the ratio to times the img height and width
+    if max(img.height, img.width) > iconsize:
+        reducerate = 256 / max(img.height, img.width)
+
+    return img.resize((int(img.width*reducerate), int(img.height*reducerate)), Image.ANTIALIAS)

setup.py

@@ -1,5 +1,7 @@
 import pip
 import sys
+import os
+
 try:
     argument = sys.argv[1].lower()
 except:
@@ -8,5 +10,15 @@
     def install():
         pip.main(['install', '-r', 'requirements.txt'])
 
+        # make the database
+        if not os.path.exists('db.db'):
+            from subprocess import call
+            try:
+                call(["python", "db/db_create.py"])
+            except:
+                call(["python3", "db/db_create.py"])
+
+
+
     if __name__ == "__main__":
         install()

unittests/image/correct1.jpg

@@ -0,0 +1,19 @@
+import back_end.image as img
+import unittest
+
+class TestImageMethods(unittest.TestCase):
+    def test_image1(self):
+        x = img.open_image(open('correct1.jpg', 'rb').read())
+        self.assertFalse(isinstance(x, type(None)))
+        self.assertEqual(x[1], 'jpeg')
+
+    def test_image2(self):
+        x = img.open_image(open('correct1.jpg', 'rb').read())[0]
+        x.save("out1.jpg")
+        x = img.to_icon(x)
+        x.save("out2.jpg")
+        self.assertFalse(False)
+
+
+if __name__ == "__main__":
+    unittest.main()