Como testar se dois números são números amigos em Python - Escreva um programa Python que - Desafio de Programação Resolvido em Python

Informe o primeiro número: 220
Informe o segundo número: 284
Os dois números são amigos

Informe o primeiro número: 158
Informe o segundo número: 932
Os números não são amigos.

Informe o primeiro número: 1210
Informe o segundo número: 1184
Os dois números são amigos

----------------------------------------------------------------------
Precisa de ajuda? Chama no WhatsApp +55 (62) 98553-6711 (Osmar)

Este código foi útil? Paga um cafezinho pra mim :-(
PIX: osmar@arquivodecodigos.com.br 
----------------------------------------------------------------------

# função principal do programa
def main():
  # vamos ler os dois números
  n1 = int(input("Informe o primeiro número: "))
  n2 = int(input("Informe o segundo número: "))
  
  # vamos somar todos os divisores próprios do
  # primeiro número
  soma = 0
  for i in range(1, n1):
    if n1 % i == 0:
      soma = soma + i
    
  # se essa soma for igual ao valor de n2 nós já temos a
  # primeira condição
  if soma == n2:
    # zera a soma
    soma = 0
    for i in range(1, n2):
      if n2 % i == 0:
        soma = soma + i
      
    # esta soma é igual ao valor de n1?
    if soma == n1:
      print("Os dois números são amigos")
    else:
      print("Os números não são amigos.")
  else:
    print("Os números não são amigos.")
  
if __name__== "__main__":
  main()