Repository navigation
Expand file tree
/
Copy paththinkpython.py
More file actions
93 lines (73 loc) · 2.66 KB
/
Copy paththinkpython.py
File metadata and controls
93 lines (73 loc) · 2.66 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
import contextlib
import io
import re
def extract_function_name(text):
"""Find a function definition and return its name.
text: String
returns: String or None
"""
pattern = r"def\s+(\w+)\s*\("
match = re.search(pattern, text)
if match:
func_name = match.group(1)
return func_name
else:
return None
# the functions that define cell magic commands are only defined
# if we're running in Jupyter.
try:
from IPython.core.magic import register_cell_magic
from IPython.core.magic_arguments import argument, magic_arguments, parse_argstring
@register_cell_magic
def add_method_to(args, cell):
# get the name of the function defined in this cell
func_name = extract_function_name(cell)
if func_name is None:
return f"This cell doesn't define any new functions."
# get the class we're adding it to
namespace = get_ipython().user_ns
class_name = args
cls = namespace.get(class_name, None)
if cls is None:
return f"Class '{class_name}' not found."
# save the old version of the function if it was already defined
old_func = namespace.get(func_name, None)
if old_func is not None:
del namespace[func_name]
# Execute the cell to define the function
get_ipython().run_cell(cell)
# get the newly defined function
new_func = namespace.get(func_name, None)
if new_func is None:
return f"This cell didn't define {func_name}."
# add the function to the class and remove it from the namespace
setattr(cls, func_name, new_func)
del namespace[func_name]
# restore the old function to the namespace
if old_func is not None:
namespace[func_name] = old_func
@register_cell_magic
def expect_error(line, cell):
try:
get_ipython().run_cell(cell)
except Exception as e:
get_ipython().run_cell("%tb")
@magic_arguments()
@argument("exception", help="Type of exception to catch")
@register_cell_magic
def expect(line, cell):
args = parse_argstring(expect, line)
exception = eval(args.exception)
try:
get_ipython().run_cell(cell)
except exception as e:
get_ipython().run_cell("%tb")
def traceback(mode):
"""Set the traceback mode.
mode: string
"""
with contextlib.redirect_stdout(io.StringIO()):
get_ipython().run_cell(f"%xmode {mode}")
traceback("Minimal")
except (ImportError, NameError):
print("Warning: IPython is not available, cell magic not defined.")