wmclt / kaprekarsnumber Goto Github PK

from turtle import *
from types import *

myTree = ["A",["B",["C",["D","E"],"F"],"G","H"]];
s = 50;
startpos = (0,120)
def cntstrs(list):
    return len([item for item in list if type(item) is StringType])
def drawtree(tree, pos, head=0):
    c = cntstrs(tree)
    while len(tree):
        goto(pos)
        item = tree.pop(0)
        if head:
            write(item,1)
            drawtree(tree.pop(0),pos)
        else:
            if type(item) is StringType:
                newpos = (pos[0] + s*c/4 - s*cntstrs(tree), pos[1] - s)
                down()
                goto((newpos[0], newpos[1] + 15))
                up()
                goto(newpos)
                write(item,1)
            elif type(item) is ListType:
                drawtree(item,newpos)

up()
drawtree(myTree, startpos,1)