ant

ant.py

from kandinsky import *
from random import *

def rcolor():
    return color(randint(0, 255),
                 randint(0, 255),
                 randint(0, 255))

rule = [1, -1, -1, 1]

nr = len(rule)
moves = [(-1, 0), (0, -1), (1, 0), (0, 1)]
colors = [(0, 0, 0)] + [rcolor() for _ in range(nr-1)]

p = (160, 111)
d = 0

fill_rect(0, 0, 320, 222, colors[0])
while True:
    idx = colors.index(get_pixel(*p))
    set_pixel(p[0], p[1], colors[(idx + 1) % nr])
    d = (d + rule[idx]) % len(moves)
    p = ((p[0] + moves[d][0]) % 320, (p[1] + moves[d][1]) % 222)