Дружественные числа

def div_sum(n):
    res = 0
    for d in range(1, n):
        if n % d == 0:
            res += d
    return res
#==============================================================================
def get_amicable_numbers(n):
    res     = []
    res_inv = []
    res_ami = []
    for i in range(1, n+1):
        k = div_sum(i)
        if i != k and k <= n:
            res.append( (i, k) )
            res_inv.append( (k, i) )
            if (i,k) in res_inv:
                res_ami.append( (k, i) )
    return res_ami
#==============================================================================
n = int( input( 'n = ' ) )
print( get_amicable_numbers(n) )

n=int(input("n="))
 
for a in range(n+1):
    sa=1
    k=2
    while k*k<=a:
        if a%k==0:
            sa+=k
            m=a//k
            if m != k:
                sa+=m
        k+=1
    if sa<=n:
        ssa=1
        k=2
        while k*k<=sa:
            if sa%k==0:
                ssa+=k
                m=sa//k
                if m != k:
                    ssa+=m
            k+=1
    if a==ssa and a != sa and a < sa:
        print(a,sa)