Add rc_context and rename rcfile to rc_file.

Matthew Emmett · memmett · commit 2045bdbd2c00 · 2012-08-19T12:15:11.000-04:00
rc_context is a context manager that optionally accepts a file name:

with mpl.rc_context():
    mpl.rc('lines', linewidth=22)
    ...

or

with mpl.rc_context(fname):
    ...
diff --git a/lib/matplotlib/__init__.py b/lib/matplotlib/__init__.py
@@ -178,7 +178,6 @@ def byte2str(b): return b
             __version__numpy__, numpy.__version__))
 del version
 
-
 def is_string_like(obj):
     if hasattr(obj, 'shape'): return 0
     try: obj + ''
@@ -904,13 +903,25 @@ def rcdefaults():
     """
     rcParams.update(rcParamsDefault)
 
-def rcfile(fname):
+
+def rc_file(fname):
     """
     Update rc params from file.
     """
     rcParams.update(rc_params_from_file(fname))
 
 
+class rc_context(object):
+    def __init__(self, fname=None):
+        self.fname = fname
+    def __enter__(self):
+        self._rcparams = rcParams.copy()
+        if self.fname:
+            rc_file(self.fname)
+    def __exit__(self, type, value, tb):
+        rcParams.update(self._rcparams)
+
+
 def rc_file_defaults():
     """
     Restore the default rc params from the original matplotlib rc that
