#!/usr/bin/env python3

from PIL import Image
import random
import pprint

IMAGE_LENGTH = 32
NUM_COLOURS = 3

def generate_array(colours):
    array = [[9]*IMAGE_LENGTH]*IMAGE_LENGTH
    for i in range(IMAGE_LENGTH):
        for j in range(IMAGE_LENGTH):
            array[j][i] = random.choice(colours)
    return array

def generate_random_colours():
    array = [(0, 0, 0)]*NUM_COLOURS
    for i in range(NUM_COLOURS):
        r = random.randint(0, 255)
        g = random.randint(0, 255)
        b = random.randint(0, 255)
        array[i] = (r, g, b)

    return array

def bad_main(index):
    colours = generate_random_colours()
    array = generate_array(colours)
    im = Image.new(mode="RGB", size=(IMAGE_LENGTH, IMAGE_LENGTH))
    for i in range(im.size[0]):
        for j in range(im.size[1]):
            im.putpixel((i, j), array[i][j])
    im.save('output_{}.png'.format(index))

if __name__ == "__main__":
    for i in range(10):
        bad_main(i)