From 51617f6ea56da7f0043fd0f46bf350081854cdc3 Mon Sep 17 00:00:00 2001
From: Ethan Dalool <git@voussoir.net>
Date: Tue, 30 Jun 2020 20:27:35 -0700
Subject: [PATCH] Use argparse for resize.py.

---
 resize.py | 55 +++++++++++++++++++++++++++++++++++++++++--------------
 1 file changed, 41 insertions(+), 14 deletions(-)

diff --git a/resize.py b/resize.py
index 6efff3b..794d6fb 100644
--- a/resize.py
+++ b/resize.py
@@ -1,3 +1,4 @@
+import argparse
 import os
 from PIL import Image
 import sys
@@ -7,31 +8,57 @@ from voussoirkit import winglob
 
 filenames = sys.argv[1]
 
-filenames = winglob.glob(filenames)
-for filename in filenames:
+def resize(filename, new_x=None, new_y=None, scale=None, nearest_neighbor=False):
     i = Image.open(filename)
-    if all(x.isdigit() for x in sys.argv[2:3]):
-        new_x = int(sys.argv[2])
-        new_y = int(sys.argv[3])
-    else:
-        try:
-            ratio = float(sys.argv[2])
-            new_x = int(i.size[0] * ratio)
-            new_y = int(i.size[1] * ratio)
-        except ValueError:
-            print('you did it wrong')
-            quit()
 
     (image_width, image_height) = i.size
 
+    if new_x is not None and new_y is not None:
+        pass
+    elif scale:
+        new_x = int(image_width * scale)
+        new_y = int(image_height * scale)
+    print(new_x, new_y)
+
     if new_x == 0:
         (new_x, new_y) = imagetools.fit_into_bounds(image_width, image_height, 10000000, new_y)
     if new_y == 0:
         (new_x, new_y) = imagetools.fit_into_bounds(image_width, image_height, new_x, 10000000)
 
     print(i.size, new_x, new_y)
-    i = i.resize( (new_x, new_y), Image.ANTIALIAS)
+    if nearest_neighbor:
+        i = i.resize( (new_x, new_y), Image.NEAREST)
+    else:
+        i = i.resize( (new_x, new_y), Image.ANTIALIAS)
     suffix = '_{width}x{height}'.format(width=new_x, height=new_y)
     (base, extension) = os.path.splitext(filename)
     newname = base + suffix + extension
     i.save(newname, quality=100)
+
+
+def resize_argparse(args):
+    filenames = winglob.glob(args.pattern)
+    for filename in filenames:
+        resize(
+            filename,
+            args.new_x,
+            args.new_y,
+            scale=args.scale,
+            nearest_neighbor=args.nearest_neighbor,
+        )
+
+def main(argv):
+    parser = argparse.ArgumentParser(description=__doc__)
+
+    parser.add_argument('pattern')
+    parser.add_argument('new_x', nargs='?', type=int, default=None)
+    parser.add_argument('new_y', nargs='?', type=int, default=None)
+    parser.add_argument('--scale', dest='scale', type=float, default=None)
+    parser.add_argument('--nearest', dest='nearest_neighbor', action='store_true')
+    parser.set_defaults(func=resize_argparse)
+
+    args = parser.parse_args(argv)
+    return args.func(args)
+
+if __name__ == '__main__':
+    raise SystemExit(main(sys.argv[1:]))