logged the last error message for each exception class

Erfaniaa · Apr 19, 2023 · d1259c2 · d1259c2
1 parent 2e0b0a8
commit d1259c2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/utils.py b/utils.py
@@ -6,13 +6,13 @@ def retry(max_retries, on_fail):
 	def wrapper(fn):
 		def inner(*args, **kwargs):
 			so_far = 0
-			exceptions = set()
+			exceptions = {}
 			while so_far <= max_retries:
 				try:
 					return fn(*args, **kwargs)
 				except Exception as e:
-					exceptions.add(str(e))
+					exceptions[type(e)] = str(e)
 					so_far += 1
-			return on_fail('\n'.join(exceptions))
+			return on_fail('\n'.join(exceptions.values()))
 		return inner
 	return wrapper