python分形树树枝上色

啦啦啦，思考题做出来了。留个纪念
我土味情话学的这么好。都没人夸我

#self_product分形树上色

import turtle

def draw_branch(branch_length):
    #绘制右侧树枝
    if branch_length <= 5:
        turtle.color('green')

    if branch_length > 5:
        turtle.forward(branch_length)
        #print('向前',branch_length)
        turtle.right(20)
        #print('向右转20')
        draw_branch(branch_length - 15)

        #绘制左侧树枝
        turtle.left(40)
        #print('向左转40')
        draw_branch(branch_length - 15)

        #返回之前的树枝
        turtle.right(20)
        #print('向右转20')
        turtle.backward(branch_length)
        #print('向后',branch_length)
        turtle.color('pink')

def main():
    turtle.left(90)
    turtle.penup()
    turtle.backward(180) #调整落笔中心点的位置
    turtle.pendown()
    turtle.screensize(800, 600, "black")
    turtle.pensize(2)
    turtle.color('pink')
    draw_branch(110)
    turtle.penup()
    turtle.goto(-170,-10)
    turtle.write("不要抱怨了", font=("Arial", 20, "normal"))
    turtle.pendown()
    turtle.penup()
    turtle.goto(30,-100)
    turtle.write("抱我呀", font=("Arial", 20, "normal"))
    turtle.pendown()
    turtle.exitonclick()

if __name__ == '__main__':
    main()