Fix bug in restoring logging level if exception is raised in context …

…manager
pv · Feb 7, 2015 · 044b8d6 · 044b8d6
1 parent cf95250
commit 044b8d6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/asv/console.py b/asv/console.py
@@ -308,8 +308,10 @@ def enable(self, verbose=False):
     def set_level(self, level):
         orig_level = self._logger.level
         self._logger.setLevel(level)
-        yield
-        self._logger.setLevel(orig_level)
+        try:
+            yield
+        finally:
+            self._logger.setLevel(orig_level)
 
     def is_debug_enabled(self):
         return self._logger.getEffectiveLevel() <= logging.DEBUG