From aa8b8937c03bb5e87008f756452270fed1b61260 Mon Sep 17 00:00:00 2001 From: Maya Herrscher Date: Tue, 10 Oct 2023 16:36:36 +0200 Subject: [PATCH] Add cat says stuff (untested) --- cat.py | 9 ++++++++- pixelflut.py | 5 +++-- 2 files changed, 11 insertions(+), 3 deletions(-) diff --git a/cat.py b/cat.py index 8f59d3e..4897dd5 100755 --- a/cat.py +++ b/cat.py @@ -7,7 +7,14 @@ import urllib.request from io import BytesIO def cat(w, h): - url = f'https://cataas.com/cat/says/henlo?size=100&width={w}?height={h}?text=cat' + url = f'https://cataas.com/cat/width={w}?height={h}?text=cat' + resp = urllib.request.urlopen(url) + im_data = resp.read() + im = Image.open(BytesIO(im_data)) + return im + +def cat_says(text,w, h): + url = f'https://cataas.com/cat/says/{text}?size=100&width={w}?height={h}?text=cat' resp = urllib.request.urlopen(url) im_data = resp.read() im = Image.open(BytesIO(im_data)) diff --git a/pixelflut.py b/pixelflut.py index 565c2bc..89ad5d0 100755 --- a/pixelflut.py +++ b/pixelflut.py @@ -6,7 +6,7 @@ from itertools import product import string import numpy as np from PIL import Image -from cat import cat +from cat import cat, cat_says import argparse # parser stuff @@ -18,10 +18,10 @@ 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('-cs','--catsays') parser.add_argument('-wd','--width', default=0) parser.add_argument('-ht','--height', default=0) args = parser.parse_args() - HOST, PORT = args.hostname, int(args.port) # change start point if requested @@ -44,6 +44,7 @@ with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock: # open image to be sent or query a cat image if args.cat: im = cat(sizew,sizeh) + elif args.catsays: im = cat_says(args.catsays,sizew,sizeh) elif image in args: im = Image.open(args.image) pixels = list(im.getdata())