clearcmd.py
46 lines
| 1.3 KiB
| text/x-python
|
PythonLexer
vivainio
|
r245 | # -*- coding: utf-8 -*- | ||
""" IPython extension: add %clear magic """ | ||||
import IPython.ipapi | ||||
import gc | ||||
ip = IPython.ipapi.get() | ||||
def clear_f(self,arg): | ||||
""" Clear various data (e.g. stored history data) | ||||
%clear out - clear output history | ||||
%clear in - clear input history | ||||
""" | ||||
api = self.getapi() | ||||
for target in arg.split(): | ||||
if target == 'out': | ||||
print "Flushing output cache (%d entries)" % len(api.user_ns()['_oh']) | ||||
self.outputcache.flush() | ||||
elif target == 'in': | ||||
print "Flushing input history" | ||||
from IPython import iplib | ||||
del self.input_hist[:] | ||||
del self.input_hist_raw[:] | ||||
for n in range(1,self.outputcache.prompt_count + 1): | ||||
key = '_i'+`n` | ||||
try: | ||||
del self.user_ns[key] | ||||
except: pass | ||||
elif target == 'array': | ||||
try: | ||||
pylab=ip.IP.pylab | ||||
for x in self.user_ns.keys(): | ||||
if isinstance(self.user_ns[x],pylab.arraytype): | ||||
del self.user_ns[x] | ||||
except AttributeError: | ||||
print "Clear array only available in -pylab mode" | ||||
gc.collect() | ||||
ip.expose_magic("clear",clear_f) | ||||