Skip to content

HTTPS clone URL

Subversion checkout URL

You can clone with HTTPS or Subversion.

Download ZIP
Browse files

Add drop_by_id method to shell, to remove variables added by extensions.

  • Loading branch information...
commit ea8ae4e94c17f102bb15fea16bbbda1d06a9a3c1 1 parent 71ae352
@takluyver authored
View
18 IPython/core/interactiveshell.py
@@ -1264,6 +1264,24 @@ def push(self, variables, interactive=True):
else:
for name,val in vdict.iteritems():
config_ns[name] = val
+
+ def drop_by_id(self, variables):
+ """Remove a dict of variables from the user namespace, if they are the
+ same as the values in the dictionary.
+
+ This is intended for use by extensions: variables that they've added can
+ be taken back out if they are unloaded, without removing any that the
+ user has overwritten.
+
+ Parameters
+ ----------
+ variables : dict
+ A dictionary mapping object names (as strings) to the objects.
+ """
+ for name, obj in variables.iteritems():
+ if name in self.user_ns and self.user_ns[name] is obj:
+ del self.user_ns[name]
+ self.user_ns_hidden.pop(name, None)
#-------------------------------------------------------------------------
# Things related to object introspection
View
15 IPython/core/tests/test_interactiveshell.py
@@ -179,4 +179,17 @@ def test_bad_custom_tb_return(self):
finally:
io.stderr = save_stderr
-
+ def test_drop_by_id(self):
+ ip = get_ipython()
+ myvars = {"a":object(), "b":object(), "c": object()}
+ ip.push(myvars, interactive=False)
+ for name in myvars:
+ assert name in ip.user_ns, name
+ assert name in ip.user_ns_hidden, name
+ ip.user_ns['b'] = 12
+ ip.drop_by_id(myvars)
+ for name in ["a", "c"]:
+ assert name not in ip.user_ns, name
+ assert name not in ip.user_ns_hidden, name
+ assert ip.user_ns['b'] == 12
+ ip.reset()
Please sign in to comment.
Something went wrong with that request. Please try again.