如何用python话螺旋正方形

#用循环画螺旋正方形
import turtle

#用循环制作
a=int(input('请输入螺旋正方形的圈数'))#加入一个判断语句
if a<0:
    print('请输入正确的圈数')

b=float(input('请输入正方形增长的长度'))#同上
if b<0:
    print('请输入正确的长度')
c=int(input('请输入画笔的粗细'))
turtle.pensize(c)
d=input('画笔颜色')
turtle.pencolor(d)
A=5 #这里也可以用input让用户选择初始长度
e=int(input('顺时针旋转输入1，逆时针旋转输入2'))
if e==1:
    for index in range(1,a):
        turtle.left(90)
        turtle.fd(A)
        A=A+b*index
else:
    for index in range(1,a):
        turtle.right(90)
        turtle.fd(A)
        A=A+b*index
下面画是由外到内的螺旋正方形

import turtle
a=int(input('请输入最外层的长度'))
b=int(input('请输入每次减少的长度'))
c=int(input('请输入画笔的粗细'))
d=input('请输入画笔的颜色')
S=a//b
for index in range(0,S):
    turtle.right(90)
    turtle.fd(a)
    a=a-b