Reado parsing with argparse

.gitignore

@@ -0,0 +1 @@
+__pycache__

cat.py

@@ -6,13 +6,13 @@ from PIL import Image
 import urllib.request
 from io import BytesIO
 
-def cat(h, w):
-    url = f'https://cataas.com/cat?width={w}&height={h}'
+def cat(w, h):
+    url = f'https://cataas.com/cat/says/DOCHKATZE?size=100&width={w}?height={h}?text=cat'
     resp = urllib.request.urlopen(url)
     im_data = resp.read()
     im = Image.open(BytesIO(im_data))
-    im.show()
     return im
 
 if __name__ == "__main__":
-    cat(1920,1080)
+    cat = cat(1920,1080)
+    cat.show()

pixelflut.py

@@ -7,23 +7,28 @@ import string
 import numpy as np
 from PIL import Image
 from cat import cat
+import argparse
 
-# open image to be sent
-image = sys.argv[1]
-im = Image.open(image)
+# parser stuff
+parser = argparse.ArgumentParser(
+                    prog='FluffyPixelfluter',
+                    description='used to paint Pixelflut canvas',)
+parser.add_argument('hostname')
+parser.add_argument('port')
+parser.add_argument('-i','--image')
+parser.add_argument('-r','--random', action='store_true')
+parser.add_argument('-c','--cat', action='store_true')
+parser.add_argument('-wd','--width', default=0)
+parser.add_argument('-ht','--height', default=0)
+args = parser.parse_args()
 
-pixels = list(im.getdata())
-width, height = im.size
-pixels = [pixels[i * width:(i + 1) * width] for i in range(0,height)]
-
-HOST, PORT = "px.oeinf.de", 1234
+HOST, PORT = args.hostname, int(args.port)
 
 # change start point if requested
-starth, startw = 0, 0
-if len(sys.argv) > 2: starth = int(sys.argv[2])
-if len(sys.argv) > 3: startw = int(sys.argv[3])
-rnd = False
-if len(sys.argv) > 4: rnd = sys.argv[4] == "rnd"
+starth = int(args.height)
+startw = int(args.width)
+
+rnd = args.random
 
 # Create a socket (SOCK_STREAM means a TCP socket)
     # Connect to server and send data
@@ -36,6 +41,14 @@ with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock:
     received = str(sock.recv(1024), "utf-8")
     sizew = int(received.split(" ")[1])
     sizeh = int(received.split(" ")[2])
+
+    # open image to be sent or query a cat image
+    if  args.cat: im = cat(sizew,sizeh) 
+    elif image in args: im = Image.open(args.image)
+
+    pixels = list(im.getdata())
+    width, height = im.size
+    pixels = [pixels[i * width:(i + 1) * width] for i in range(0,height)]
     
     # create list of stuff and shuffle if requested
     widthlist = list(range(0 + startw,min(int(sizew),(width+startw))))