python循环结构的简单问题

编程实现如下功能的程序。输入正整数n，如果n为奇数， 计算并输出一的平方加五……加七的……一直加到n的平方。如果n为偶数计算并输出2的……加4的……加6的……8的平方一直加到n的平方
输入提示input n:
输出提示sum={}

……为平方的意思

a = int(input("input n:"))
sum = 0
if a%2 == 1:
    for i in range(1,a+1,2):
        sum += i**2
else:
    for i in range(2,a+1,2):
        sum += i**2
print("sum = {"+str(sum)+"}")