parse arguments

banter.py

@@ -4,18 +4,13 @@ __doc__ = 'little script to make irc color art that will probably get you banned
 # whee, look at this sus shebang workaround to always cause
 # unbuffered mode lol
 
-import sys,time
+import sys,time,argparse
 from PIL import Image
 from color import closestColor
 
-ASCIIWIDTH=80
-COLORCHAR='\x03{},{}' # this will be String.format()'ted with two args
-FILLER='.'
 
-if len(sys.argv) < 2:
-    raise Exception('you must supply an image')
-
-im = Image.open(sys.argv[1], 'r')
+def main(imgPath,delay,ASCIIWIDTH,COLORCHAR,FILLER):
+    im = Image.open(imgPath, 'r')
     width, height = im.size
     pixel_values = list(im.getdata())
 
@@ -40,5 +35,23 @@ for y in range(asciiHeight):
             currentPixel+=1
 
         print(''.join(line))
-    if len(sys.argv) > 2:
-        time.sleep(float(sys.argv[2]))
+        if delay:
+            time.sleep(delay)
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser("banter")
+    parser.add_argument("file")
+    parser.add_argument("-d",metavar='delay',default=0)
+    parser.add_argument("-w",metavar='width',default=80)
+    parser.add_argument("--colorfmt",metavar='format',default='\\x03{},{}')
+    parser.add_argument("--filler",metavar='filler',default='.')
+    args = parser.parse_args()
+
+    main(
+            args.file,
+            float(args.d),
+            int(args.w),
+            args.colorfmt.encode().decode('unicode_escape'),
+            args.filler.encode().decode('unicode_escape')
+            )