用Python解相親數,請各位來幫助

對使用者所輸入的正整數進行判斷:

  1. 如果它有「相親數」,則印出它的相親數。
  2. 如果它沒有「相親數」,則印出 None

[範例]
輸入-1
220
輸出-1
284

輸入-2
123
輸出-2
None

def judge(n):
    s = 0
    for i in range(1, n):
        if n % i == 0:
            s += i
            
    return s


n = int(input(">>>"))

a1 = judge(n)
a2 = judge(a1)

if n == a2:
    print(a1)
else:
    print(None)