limit iterations for bruteforce2

This commit is contained in:
josch 2012-05-16 09:04:42 +02:00
parent 57e6af6cc8
commit 2afff50f00

View file

@ -92,10 +92,17 @@ def main():
else: else:
rot_pallet_default = False rot_pallet_default = False
if os.environ.get("iterations"):
max_iter = int(os.environ["iterations"])
else:
max_iter = -1
if try_rot_article and try_rot_pallet: if try_rot_article and try_rot_pallet:
product_it = product_varlength(4) product_it = product_varlength(4)
elif try_rot_article or try_rot_pallet: elif try_rot_article or try_rot_pallet:
product_it = product_varlength(2) product_it = product_varlength(2)
i = 0
while True: while True:
rests = list() rests = list()
layers = list() layers = list()
@ -151,5 +158,9 @@ def main():
print b2a_base64(zlib.compress(cPickle.dumps((layers, rests, pallet)))), print b2a_base64(zlib.compress(cPickle.dumps((layers, rests, pallet)))),
if not try_rot_article and not try_rot_pallet: if not try_rot_article and not try_rot_pallet:
break # only one iteration if both are deactivated break # only one iteration if both are deactivated
i+=1
if max_iter != -1 and i >= max_iter:
break
if __name__ == "__main__": if __name__ == "__main__":
main() main()