fix: Avoid 500 error on the website if unable to dump pickle.

pull/258/head
Raphaël Vinot 2021-09-06 16:30:43 +02:00
parent 074bb675f6
commit 7d76d96ff7
1 changed files with 4 additions and 1 deletions

View File

@ -202,7 +202,10 @@ class Lookyloo():
# is discarded in the RecursionError above. # is discarded in the RecursionError above.
default_recursion_limit = sys.getrecursionlimit() default_recursion_limit = sys.getrecursionlimit()
sys.setrecursionlimit(int(default_recursion_limit * 1.1)) sys.setrecursionlimit(int(default_recursion_limit * 1.1))
pickle.dump(ct, _p) try:
pickle.dump(ct, _p)
except RecursionError as e:
raise NoValidHarFile(f'Tree too deep, probably a recursive refresh: {e}.\n Append /export to the URL to get the files.')
sys.setrecursionlimit(default_recursion_limit) sys.setrecursionlimit(default_recursion_limit)
finally: finally:
lock_file.unlink(missing_ok=True) lock_file.unlink(missing_ok=True)