|
@ -7,6 +7,7 @@ import string |
|
|
import numpy as np |
|
|
import numpy as np |
|
|
from PIL import Image |
|
|
from PIL import Image |
|
|
from cat import cat, cat_says |
|
|
from cat import cat, cat_says |
|
|
|
|
|
from igel import igel |
|
|
import argparse |
|
|
import argparse |
|
|
|
|
|
|
|
|
# parser stuff |
|
|
# parser stuff |
|
@ -19,6 +20,7 @@ parser.add_argument('-i','--image') |
|
|
parser.add_argument('-r','--random', action='store_true') |
|
|
parser.add_argument('-r','--random', action='store_true') |
|
|
parser.add_argument('-c','--cat', action='store_true') |
|
|
parser.add_argument('-c','--cat', action='store_true') |
|
|
parser.add_argument('-cs','--catsays') |
|
|
parser.add_argument('-cs','--catsays') |
|
|
|
|
|
parser.add_argument('-ig','--igel', action='store_true') |
|
|
parser.add_argument('-wd','--width', default=0) |
|
|
parser.add_argument('-wd','--width', default=0) |
|
|
parser.add_argument('-ht','--height', default=0) |
|
|
parser.add_argument('-ht','--height', default=0) |
|
|
args = parser.parse_args() |
|
|
args = parser.parse_args() |
|
@ -45,6 +47,7 @@ with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock: |
|
|
# open image to be sent or query a cat image |
|
|
# open image to be sent or query a cat image |
|
|
if args.cat: im = cat(sizew,sizeh) |
|
|
if args.cat: im = cat(sizew,sizeh) |
|
|
elif args.catsays: im = cat_says(args.catsays,sizew,sizeh) |
|
|
elif args.catsays: im = cat_says(args.catsays,sizew,sizeh) |
|
|
|
|
|
elif args.igel: im = Image.open(igel()) |
|
|
elif args.image: im = Image.open(args.image) |
|
|
elif args.image: im = Image.open(args.image) |
|
|
|
|
|
|
|
|
pixels = list(im.getdata()) |
|
|
pixels = list(im.getdata()) |
|
|