Escreva um programa Python que pede um número inteiro e informa se o número informado é um número de Armstrong - Desafio de Programação Resolvido em Python

Informe um número inteiro: 153
O número informado é um número de Armstrong

----------------------------------------------------------------------
Se precisar de ajuda com o código abaixo, pode me chamar
no WhatsApp +55 (62) 98553-6711 (Osmar)
----------------------------------------------------------------------

# vamos importar o módulo Math
import math

# método principal
def main():
  quant_digitos = 0
  soma = 0
	
  # vamos pedir para o usuário informar um número
  numero = int(input("Informe um número inteiro: "))
    
  # o primeiro passo é obter a quantidade de dígitos que o número informado
  # possui
  temp = numero;
  while temp != 0:
    quant_digitos = quant_digitos + 1
    temp = temp // 10
    
  # agora efetuamos o cálculo que verifica se o número é um número
  # de Armstrong ou não
  temp = numero;
  while temp > 0:
    resto = temp % 10
    soma = soma + (int)(math.pow(resto, quant_digitos))
    temp = temp // 10

  if soma == numero:
    print("O número informado é um número de Armstrong")
  else:
    print("O número informado não é um número de Armstrong")
  
if __name__== "__main__":
  main()