Create 0005.py

ncw1992120 · web-flow · commit 008d218659ad · 2016-06-13T17:36:36.000+08:00
diff --git a/pylyria/0005/0005.py b/pylyria/0005/0005.py
@@ -0,0 +1,33 @@
+# -*- coding: utf-8 -*-
+#!/usr/bin/env python
+#第 0005 题：你有一个目录，装了很多照片，把它们的尺寸变成都不大于 iPhone5 分辨率的大小。
+import os
+from PIL import Image
+
+def get_path(root = os.curdir):
+    root += os.sep
+    for path, dirs, files in os.walk(root):
+        for file_name in files:
+            yield path, file_name
+
+def get_size(width, heigh, max_width, max_heigh):
+    if width > max_width:
+        heigh = heigh * (max_width / width)
+        width = max_width
+    if heigh > max_heigh:
+        width = width * (max_heigh / heigh)
+        heigh = max_heigh
+    return width, heigh
+
+
+if __name__ == '__main__':
+    paths = get_path()
+    image_format = ('.jpg', '.jpeg', 'png', 'bmp')
+    max_size = (640, 1136)
+    for path, file_name in paths:
+        if file_name.endswith(image_format):
+            img = Image.open(path + os.sep + file_name)
+            size = get_size(*(img.size + max_size))
+            if size != img.size:
+                img.thumbnail(size)
+                img.save(path + os.sep + 'thumbnailed_' + file_name)
