Розклад на прості множники

Вихідний код

def factorize(n):
    d = 2
    while d * d <= n:
        while n % d == 0:
            print(d)
            n //= d
        d += 1
    if n > 1:
        print(n)