Turtle Art

#awesome christmas tree
from turtle import *
from math import *
levels=[]
speed=0
def triangle(s):
    seth(0)
    right(60)
    for side in range(3):
        pendown()
        forward(s)
        right(120)
        if side == 0:
            rp = pos()
        if side == 1:
            lp = pos()
    p=[rp,lp]
    penup()
    right (30)
    forward(s*sqrt(3)/8)
    return p

def ball(p):
    colors=["red","gold","blue","silver","purple"]
    c=random.randint(0,len(colors)-1)
    penup()
    goto(p[0],p[1])
    fillcolor(colors[c])
    pencolor(colors[c])

    seth(180)
    pendown()

    begin_fill()
    circle(3)
    end_fill()
def stump(p):
    penup()
    goto(p[0]+10,p[1])
    fillcolor("brown")
    pencolor("brown")
    seth(180)
    pendown()
    begin_fill()
    for i in range(2):

        forward(20)
        left(90)
        forward(50)
        left(90)
    end_fill()


import random
goto(0,200)
fillcolor("green")
pencolor("green")
begin_fill()
for i in range(1,10):
    levels.append(triangle(i*20))
end_fill()
for i in range(40):
    r=random.randint(0,len(levels)-1)
    xr=levels[r][0][0]
    xl=levels[r][1][0]
    x=random.randint(int(xl),int(xr))
    ball([x,levels[r][0][1]])
stump([0,levels[-1][1][1]])