Python prímkereső példakód