Fix MultiContext exception handling

mailpile · Oct 23, 2014 · cbe7b88 · cbe7b88
1 parent c8ec2ae
commit cbe7b88
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/mailpile/util.py b/mailpile/util.py
@@ -179,8 +179,14 @@ def __enter__(self, *args, **kwargs):
         return self
 
     def __exit__(self, *args, **kwargs):
+        raised = []
         for ctx in reversed(self.contexts):
-            ctx.__exit__(*args, **kwargs)
+            try:
+                ctx.__exit__(*args, **kwargs)
+            except Exception as e:
+                raised.append(e)
+        if raised:
+            raise raised[0]
 
 
 def FixupForWith(obj):