7段数码管绘制

import turtle, datetime
def drawGap():  # 绘制数码管间隔
    turtle.penup()
    turtle.fd(5)

def drawLine(draw):  # 绘制单段数码管
    drawGap()
    turtle.pendown() if draw else turtle.penup()
    turtle.fd(40)
    drawGap()
    turtle.right(90)

def drawDigit(d):  # 根据数字绘制七段数码管
    drawLine(True) if d in [2, 3, 4, 5, 6, 8, 9] else drawLine(False)#日中间那一横
    drawLine(True) if d in [0, 1, 3, 4, 5, 6, 7, 8, 9] else drawLine(False)#日右下角那一竖
    drawLine(True) if d in [0, 2, 3, 5, 6, 8, 9] else drawLine(False)#日下面那一横
    drawLine(True) if d in [0, 2, 6, 8] else drawLine(False)#日左下角那一竖
    turtle.left(90)
    drawLine(True) if d in [0, 4, 5, 6, 8, 9] else drawLine(False)#日左上角那一竖
    drawLine(True) if d in [0, 2, 3, 5, 6, 7, 8, 9] else drawLine(False)#日上面那一横
    drawLine(True) if d in [0, 1, 2, 3, 4, 7, 8, 9] else drawLine(False)#日右上角那一竖
    turtle.left(180)
    turtle.penup()
    turtle.fd(20)


def drawDate(date):
    turtle.pencolor("red")
    for i in date:
        if i == '-':
            turtle.write('年', font=("Arial ", 36, "normal"))
            turtle.pencolor("green")
            turtle.fd(50)
        elif i == '=':
            turtle.write('月', font=("Arial", 36, "normal"))
            turtle.pencolor("blue")
            turtle.fd(50)
        elif i == '+':
            turtle.write('日', font=("Arial", 36, "normal"))
            turtle.pencolor("yellow")
            turtle.fd(50)
        elif i == '*':
            turtle.write('时', font=("Arial", 36, "normal"))
            turtle.pencolor("purple")
            turtle.fd(50)
        elif i == '/':
            turtle.write('分', font=("Arial", 36, "normal"))
            turtle.pencolor("gold")
            turtle.fd(50)
        elif i == '$':
            turtle.write('秒', font=("Arial", 36, "normal"))
        else:
            drawDigit(eval(i))


def main():
    turtle.speed(30) #海龟移动速度
    turtle.setup(1400, 350, 200, 200)#设置窗口大小
    turtle.penup()
    turtle.fd(-650)
    turtle.pensize(5)
    drawDate(datetime.datetime.now().strftime('%Y-%m=%d+%H*'))
    turtle.hideturtle()
    turtle.done()

main()