Delete keys after iterating to prevent mutation during iter

mailgun · Aug 2, 2019 · cab2131 · cab2131
1 parent 4f4b2df
commit cab2131
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/expiringdict/__init__.py b/expiringdict/__init__.py
@@ -58,14 +58,17 @@ def __init__(self, max_len, max_age_seconds, items=None):
     def __len__(self):
         with self.lock:
             current_key = iter(self)
+            keys_to_del = []
             for k in current_key:
                 item = OrderedDict.__getitem__(self, k)
                 time_added = item[1]
                 item_age = time.time() - time_added
                 if item_age > self.max_age:
-                    del self[k]
+                    keys_to_del.append(k)
                 else:
                     break
+            for k in keys_to_del:
+                del self[k]
 
         return super(ExpiringDict, self).__len__()