Merge pull request #3 from mwek/patch-1

UpdateStatuses: Fix KeyError in mark_being_updated
DreamLab · Jan 27, 2020 · a6a25a5 · a6a25a5
2 parents bd35641 + ca094a2
commit a6a25a5
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -1,3 +1,8 @@
+1.0.3
+-----
+
+* Fixed unhandled KeyError in UpdateStatuses.
+
 1.0.2
 -----
 

diff --git a/memoize/statuses.py b/memoize/statuses.py
@@ -31,6 +31,8 @@ def mark_being_updated(self, key: CacheKey) -> None:
         self._updates_in_progress[key] = future
 
         def complete_on_timeout_passed():
+            if key not in self._updates_in_progress:
+                return
             if self._updates_in_progress[key] == future and not self._updates_in_progress[key].done():
                 self.logger.debug('Update task timed out - notifying clients awaiting for key %s', key)
                 self._updates_in_progress[key].set_result(None)

diff --git a/setup.py b/setup.py
@@ -14,7 +14,7 @@ def prepare_description():
 
 setup(
     name='py-memoize',
-    version='1.0.2',
+    version='1.0.3',
     author='Michal Zmuda',
     author_email='[email protected]',
     url='https://github.com/DreamLab/memoize',
-Original file line number
+Diff line change
@@ -1,3 +1,8 @@
+.0.3
+    -----
+    * Fixed unhandled KeyError in UpdateStatuses.
 .0.2
     -----
@@ Expand Down @@