import turtle as tle  # 小烏龜繪圖庫

def global_init(w=900, h=600, psize=15, sp=0):
    '''
    全局畫筆屬性初始化函數(shù)
    :param w: 畫布寬度
    :param h: 畫布高度
    :param psize: 畫筆尺寸
    :param sp: 繪畫速度
    :return:
    '''
    tle.setup(width=w, height=h)
    tle.color('red', 'pink')
    tle.pensize(psize)
    tle.speed(sp)

def draw_red_heart(confession='我愛你', confession_h=30,
                   sign='來自遠(yuǎn)方的仰慕者', sign_w=20, sign_h=25):
    '''
    繪制表白紅心
    :param confession: 表白語句
    :param confession_h: 表白語句尺寸大小
    :param sign: 簽名
    :param sign_w: 簽名寬度
    :param sign_h: 簽名高度
    :return:
    '''
    tle.up()
    tle.hideturtle()
    tle.goto(0, -180)
    tle.showturtle()
    tle.down()
    tle.speed(500)
    tle.begin_fill()
    tle.left(140)
    tle.forward(224)
    for n in range(200):
        tle.right(1)
        tle.forward(2)
    tle.left(120)
    for n in range(200):
        tle.right(1)
        tle.forward(2)
    tle.forward(224)
    tle.end_fill()
    tle.pensize(12)
    tle.up()
    tle.hideturtle()
    tle.goto(0, -20)
    tle.showturtle()
    tle.color('#CD5C5C', 'pink')
    tle.write(confession, font=('gungsuh', confession_h), align="center")
    tle.up()
    tle.hideturtle()
    tle.color('black', 'pink')
    tle.goto(180, -180)
    tle.showturtle()
    tle.write(sign, font=(sign_w, sign_h), align="center", move=True)

def main():
    print('按要求輸入下面的繪圖參數(shù)...')
    width = int(input('請輸入畫布寬度：\n'))
    height = int(input('請輸入畫布高度：\n'))
    psize = int(input('請輸入畫筆尺寸：\n'))
    speed = int(input('請輸入畫筆速度：\n'))
    confession = input('請輸入表白語句：\n')
    confession_h = int(input('請輸入表白語句字體大?。篭n'))
    sign = input('請輸入簽名：\n')
    sign_w = int(input('請輸入簽名寬度：\n'))
    sign_h = int(input('請輸入簽名高度：\n'))
    print('參數(shù)錄入完成，開始繪圖...')
    global_init(w=width, h=height, psize=psize, sp=speed)
    draw_red_heart(confession=confession, confession_h=confession_h,
                   sign=sign, sign_w=sign_w, sign_h=sign_h)
    print('繪圖完成！')
    # 保存eps格式的繪圖結(jié)果
    eps = tle.getscreen()
    eps.getcanvas().postscript(file=r"./表白.eps")

    # 保存JPG格式的圖片
    from PIL import Image
    im = Image.open("./表白.eps")
    im.save("表白.jpg")
    print('繪圖保存成功，默認(rèn)在當(dāng)前路徑！')

    # 繪圖完成后保持窗口不被關(guān)閉
    screen = tle.Screen()
    screen.exitonclick()

raise OSError("Unable to locate Ghostscript on paths")

main()