Euler #92

def squaresum(n):
    total = 0
    for x in str(n):
        total += (int(x))**2
    return total

answer = 0
for x in range(1, 10**7+1):
    print(x)
    while True:
        if squaresum(x) == 89:
            answer += 1
            break
        if squaresum(x) == 1:
            break
        else:
            x = squaresum(x)
print(answer)