mercurial-scm/hg: comparison mercurial/ui.py

equal deleted inserted replaced

-:798bcb1274dd
+:b0a8337ba9af
 # This software may be used and distributed according to the terms of the
 # GNU General Public License version 2 or any later version.
 from __future__ import absolute_import
+import contextlib
 import errno
 import getpass
 import inspect
 import os
 import re
 return
 msg += ("\n(compatibility will be dropped after Mercurial-%s,"
 " update your code.)") % version
 self.develwarn(msg, stacklevel=2, config='deprec-warn')
+@contextlib.contextmanager
+def configoverride(self, overrides, source=""):
+"""Context manager for temporary config overrides
+`overrides` must be a dict of the following structure:
+{(section, name) : value}"""
+backups = {}
+for (section, name), value in overrides.items():
+backups[(section, name)] = self.backupconfig(section, name)
+self.setconfig(section, name, value, source)
+yield
+for __, backup in backups.items():
+self.restoreconfig(backup)
+# just restoring ui.quiet config to the previous value is not enough
+# as it does not update ui.quiet class member
+if ('ui', 'quiet') in overrides:
+self.fixconfig(section='ui')
 class paths(dict):
 """Represents a collection of paths and their configs.
 Data is initially derived from ui instances and the config files they have
 loaded.