Python求因子和

．请编写一个函数fun()，它的功能是计算并输出给定整数n的所有因子(不包括1与自身)的平方和（规定n的值不大于100）。例如：主函数从键盘给输入n的值为56，则输出为sum=1113

def fun():
    sum = 0
    n = int(input())
    for i in range(2, int(n**0.5)+1):
        if(n%i==0):
            sum += (i**2 + (n//i) ** 2) if i != n//i else i**2
    print("sum=%d" % sum)

fun()