added image optimizer

mjhea0 · mjhea0 · commit 75f1eb491516 · 2014-05-18T15:36:36.000-06:00
diff --git a/09_basic_link_web_crawler.py b/09_basic_link_web_crawler.py
@@ -26,7 +26,8 @@ def crawl(url):
         link = urlparse.urljoin(url, link)
 
         print link
-
+        
 
 if __name__ == '__main__':
     crawl('http://www.realpython.com')
+
diff --git a/11_optimize_images_with_wand.py b/11_optimize_images_with_wand.py
@@ -0,0 +1,52 @@
+import fnmatch
+import os
+
+# sudo pip install Wand
+from wand.image import Image
+# sudo pip install http://pypi.python.org/packages/source/h/hurry.filesize/hurry.filesize-0.9.tar.gz
+from hurry.filesize import size
+
+
+# constants
+PATH = '/../../../..'
+PATTERN = '*.jpg'
+
+
+def get_image_file_names(filepath, pattern):
+    matches = []
+    if os.path.exists(filepath):
+        for root, dirnames, filenames in os.walk(filepath):
+            for filename in fnmatch.filter(filenames, pattern):
+                matches.append(os.path.join(root, filename))  # full path
+        if matches:
+            print "Found {} files, with a total file size of {}.".format(len(matches), get_total_size(matches))
+            return matches
+        else:
+            print "No files found."
+    else:
+        print "Sorry that path does not exist. Try again."
+
+
+def get_total_size(list_of_image_names):
+    total_size = 0
+    for image_name in list_of_image_names:
+        total_size += os.path.getsize(image_name)
+    return size(total_size)
+
+
+def resize_images(list_of_image_names):
+    print "Optimizing ... "
+    for index, image_name in enumerate(list_of_image_names):
+        with open(image_name) as f:
+            image_binary = f.read()
+        with Image(blob=image_binary) as img:
+            if img.height >= 600:
+                img.transform(resize='x600')
+                img.save(filename=image_name)
+    print "Optimization complete."
+
+
+if __name__ == '__main__':
+    all_images = get_image_file_names(PATH, PATTERN)
+    resize_images(all_images)
+    get_image_file_names(PATH, PATTERN)
