Make sure the context manager teardown is done even if there an excep…

…tion.
zopefoundation · Feb 1, 2017 · 497a9ba · 497a9ba
1 parent 215ffa1
commit 497a9ba
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/src/AccessControl/SimpleObjectPolicies.py b/src/AccessControl/SimpleObjectPolicies.py
@@ -128,8 +128,10 @@ def override_containers(type_, assertions):
     """Temporarily override the container assertions."""
     orig_container = Containers(type_, _marker)
     ContainerAssertions[type_] = assertions
-    yield
-    if orig_container is _marker:
-        del ContainerAssertions[type_]
-    else:
-        ContainerAssertions[type_] = orig_container
+    try:
+        yield
+    finally:
+        if orig_container is _marker:
+            del ContainerAssertions[type_]
+        else:
+            ContainerAssertions[type_] = orig_container