Genrátor prvočísel

import math
prvocislo = 1
buffer = 2
while True:
  if (buffer < prvocislo):
    if (prvocislo%buffer == 0):
      buffer = 2
      prvocislo = prvocislo + 1
    buffer = buffer + 1
  else:
    print(prvocislo)
    prvocislo = prvocislo+1
    buffer = 2
  if (buffer == math.sqrt(prvocislo)):
    buffer=1
    prvocislo=prvocislo+1