parse arguments

banter.py

@@ -4,28 +4,23 @@ __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')
+def main(imgPath,delay,ASCIIWIDTH,COLORCHAR,FILLER):
+    im = Image.open(imgPath, 'r')
+    width, height = im.size
+    pixel_values = list(im.getdata())
 
-im = Image.open(sys.argv[1], 'r')
-width, height = im.size
-pixel_values = list(im.getdata())
+    ipix = width // ASCIIWIDTH # // instead of / to devide with a round number
 
-ipix = width // ASCIIWIDTH # // instead of / to devide with a round number
+    asciiHeight = height // ipix // 2
 
-asciiHeight = height // ipix // 2
+    currentPixel = 0
 
-currentPixel = 0
-
-for y in range(asciiHeight):
+    for y in range(asciiHeight):
         line = []
         lastColor=69420
 
@@ -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')
+            )