[Python] Verificar se número é igual à soma dos quadrados de 4 números primos consecutivos

Preciso fazer um programa em que digito um número e o programa verifica se ele é a soma do quadrado de 4 números primos consecutivos.

Exemplo: 2020 tem que dar 17 ^ 2 + 19 ^ 2 + 23 ^ 2 + 29 ^ 2

Meu código por enquanto está assim:

n = int(input("n: "))
num = 2
cont = 0
for div in range(1, num + 1):
if num % div == 0:
cont = cont + 1
num = num + 1
if cont == 2:
nprimo = True
else:
nprimo = False
p = 0
while p < n and nprimo:
n = (num ** 2) + ((num + 1) ** 2) + ((num + 2) ** 2) + ((num + 3) ** 2)


Acho que a parte de verificar se o número é primo está correta, o problema está no final, todos os números dão falso. Alguém tem alguma ideia?

Continue reading...