163 lines
4.7 KiB
Python
163 lines
4.7 KiB
Python
"""
|
|
Python utilities to use it from ein.el
|
|
|
|
Copyright (C) 2012- Takafumi Arakaki
|
|
|
|
Author: Takafumi Arakaki <aka.tkf at gmail.com>
|
|
|
|
ein.py is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
ein.py is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with ein.py. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
"""
|
|
|
|
__ein_pytools_version = "1.1.0"
|
|
|
|
try:
|
|
from matplotlib import rc as __ein_rc
|
|
from matplotlib import rcParams as __ein_rcParams
|
|
__ein_matplotlib_available = True
|
|
except ImportError:
|
|
__ein_matplotlib_available = False
|
|
|
|
def __ein_export_nb(nb_json, format):
|
|
import IPython.nbconvert as nbconvert
|
|
import IPython.nbformat as nbformat
|
|
nb = nbformat.reads(nb_json, nbformat.NO_CONVERT)
|
|
output = nbconvert.export_by_name(format, nb)
|
|
print(output[0])
|
|
|
|
|
|
def __ein_find_edit_target_012(*args, **kwds):
|
|
from IPython.core.interactiveshell import InteractiveShell
|
|
inst = InteractiveShell.instance()
|
|
return inst._find_edit_target(*args, **kwds)
|
|
|
|
|
|
def __ein_find_edit_target_013(*args, **kwds):
|
|
from IPython.core.interactiveshell import InteractiveShell
|
|
inst = InteractiveShell.instance()
|
|
return CodeMagics._find_edit_target(inst, *args, **kwds)
|
|
|
|
|
|
def __ein_find_edit_target_python(name):
|
|
from inspect import getsourcefile, getsourcelines
|
|
try:
|
|
obj = eval(name)
|
|
except NameError:
|
|
return False
|
|
else:
|
|
sfile = getsourcefile(obj)
|
|
sline = getsourcelines(obj)[-1]
|
|
if sfile and sline:
|
|
return(sfile, sline, False)
|
|
else:
|
|
return False
|
|
|
|
try:
|
|
from IPython.core.magics import CodeMagics
|
|
__ein_find_edit_target = __ein_find_edit_target_013
|
|
except ImportError:
|
|
__ein_find_edit_target = __ein_find_edit_target_012
|
|
|
|
|
|
def __ein_set_matplotlib_param(family, setting, value):
|
|
settings = {}
|
|
if __ein_matplotlib_available:
|
|
settings[setting] = eval(value)
|
|
__ein_rc(family, **settings)
|
|
else:
|
|
raise RuntimeError("Matplotlib not installed in this instance of python!")
|
|
|
|
|
|
def __ein_set_figure_size(dim):
|
|
__ein_set_matplotlib_param('figure', 'figsize', dim)
|
|
|
|
|
|
def __ein_set_figure_dpi(dpi):
|
|
__ein_set_matplotlib_param('figure', 'dpi', dpi)
|
|
|
|
|
|
def __ein_get_matplotlib_params():
|
|
if __ein_matplotlib_available:
|
|
import json
|
|
print(json.dumps([k for k in __ein_rcParams.keys()]))
|
|
else:
|
|
raise RuntimeError("Matplotlib not installed in this instance of python!")
|
|
|
|
|
|
def __ein_find_source(name):
|
|
"""Given an object as string, `name`, print its place in source code."""
|
|
# FIXME: use JSON display object instead of stdout
|
|
ret = __ein_find_edit_target_python(name) or __ein_find_edit_target(name, {}, [])
|
|
if ret:
|
|
(filename, lineno, use_temp) = ret
|
|
if not use_temp:
|
|
print(filename)
|
|
print(lineno)
|
|
return
|
|
raise RuntimeError("Source code for {0} cannot be found".format(name))
|
|
|
|
|
|
def __ein_run_docstring_examples(obj, verbose=True):
|
|
from IPython.core.interactiveshell import InteractiveShell
|
|
import doctest
|
|
inst = InteractiveShell.instance()
|
|
globs = inst.user_ns
|
|
return doctest.run_docstring_examples(obj, globs, verbose=verbose)
|
|
|
|
|
|
def __ein_generate_oinfo_data(ostrings, locals=None):
|
|
import json
|
|
|
|
defined_objects = [__ein_maybe_undefined_object(obj, locals) for obj in ostrings
|
|
if __ein_maybe_undefined_object(obj, locals) is not None]
|
|
odata = [__ein_object_info_for(obj) for obj in defined_objects]
|
|
|
|
print (json.dumps(odata))
|
|
return odata
|
|
|
|
def __ein_maybe_undefined_object(obj, locals=None):
|
|
try:
|
|
return eval(obj, None, locals)
|
|
except Exception:
|
|
return None
|
|
except SyntaxError:
|
|
return None
|
|
|
|
def __ein_object_info_for(obj):
|
|
import IPython.core.oinspect
|
|
|
|
inspector = IPython.core.oinspect.Inspector()
|
|
|
|
try:
|
|
return inspector.info(obj)
|
|
except Exception:
|
|
return inspector.info(None)
|
|
|
|
|
|
def __ein_print_object_info_for(obj):
|
|
import json
|
|
|
|
oinfo = __ein_object_info_for(obj)
|
|
print (json.dumps(oinfo))
|
|
|
|
def __ein_eval_hy_string(obj):
|
|
try:
|
|
import hy
|
|
except ImportError:
|
|
print("Hy not supported in this kernel. Execute `pip install hy` if you want this support.")
|
|
|
|
expr = hy.read_str(obj)
|
|
ret = hy.eval(expr)
|
|
|
|
return ret
|