Show More
@@ -1,191 +1,199 b'' | |||
|
1 | 1 | .. _defining_magics: |
|
2 | 2 | |
|
3 | 3 | Defining custom magics |
|
4 | 4 | ====================== |
|
5 | 5 | |
|
6 | 6 | There are two main ways to define your own magic functions: from standalone |
|
7 | 7 | functions and by inheriting from a base class provided by IPython: |
|
8 | 8 | :class:`IPython.core.magic.Magics`. Below we show code you can place in a file |
|
9 | 9 | that you load from your configuration, such as any file in the ``startup`` |
|
10 | 10 | subdirectory of your default IPython profile. |
|
11 | 11 | |
|
12 | 12 | First, let us see the simplest case. The following shows how to create a line |
|
13 | 13 | magic, a cell one and one that works in both modes, using just plain functions: |
|
14 | 14 | |
|
15 | 15 | .. sourcecode:: python |
|
16 | 16 | |
|
17 | 17 | from IPython.core.magic import (register_line_magic, register_cell_magic, |
|
18 | 18 | register_line_cell_magic) |
|
19 | 19 | |
|
20 | 20 | @register_line_magic |
|
21 | 21 | def lmagic(line): |
|
22 | 22 | "my line magic" |
|
23 | 23 | return line |
|
24 | 24 | |
|
25 | 25 | @register_cell_magic |
|
26 | 26 | def cmagic(line, cell): |
|
27 | 27 | "my cell magic" |
|
28 | 28 | return line, cell |
|
29 | 29 | |
|
30 | 30 | @register_line_cell_magic |
|
31 | 31 | def lcmagic(line, cell=None): |
|
32 | 32 | "Magic that works both as %lcmagic and as %%lcmagic" |
|
33 | 33 | if cell is None: |
|
34 | 34 | print("Called as line magic") |
|
35 | 35 | return line |
|
36 | 36 | else: |
|
37 | 37 | print("Called as cell magic") |
|
38 | 38 | return line, cell |
|
39 | 39 | |
|
40 | 40 | # In an interactive session, we need to delete these to avoid |
|
41 | 41 | # name conflicts for automagic to work on line magics. |
|
42 | 42 | del lmagic, lcmagic |
|
43 | 43 | |
|
44 | 44 | |
|
45 | 45 | You can also create magics of all three kinds by inheriting from the |
|
46 | 46 | :class:`IPython.core.magic.Magics` class. This lets you create magics that can |
|
47 | 47 | potentially hold state in between calls, and that have full access to the main |
|
48 | 48 | IPython object: |
|
49 | 49 | |
|
50 | 50 | .. sourcecode:: python |
|
51 | 51 | |
|
52 | 52 | # This code can be put in any Python module, it does not require IPython |
|
53 | 53 | # itself to be running already. It only creates the magics subclass but |
|
54 | 54 | # doesn't instantiate it yet. |
|
55 | 55 | from __future__ import print_function |
|
56 | 56 | from IPython.core.magic import (Magics, magics_class, line_magic, |
|
57 | 57 | cell_magic, line_cell_magic) |
|
58 | 58 | |
|
59 | 59 | # The class MUST call this class decorator at creation time |
|
60 | 60 | @magics_class |
|
61 | 61 | class MyMagics(Magics): |
|
62 | 62 | |
|
63 | 63 | @line_magic |
|
64 | 64 | def lmagic(self, line): |
|
65 | 65 | "my line magic" |
|
66 | 66 | print("Full access to the main IPython object:", self.shell) |
|
67 | 67 | print("Variables in the user namespace:", list(self.shell.user_ns.keys())) |
|
68 | 68 | return line |
|
69 | 69 | |
|
70 | 70 | @cell_magic |
|
71 | 71 | def cmagic(self, line, cell): |
|
72 | 72 | "my cell magic" |
|
73 | 73 | return line, cell |
|
74 | 74 | |
|
75 | 75 | @line_cell_magic |
|
76 | 76 | def lcmagic(self, line, cell=None): |
|
77 | 77 | "Magic that works both as %lcmagic and as %%lcmagic" |
|
78 | 78 | if cell is None: |
|
79 | 79 | print("Called as line magic") |
|
80 | 80 | return line |
|
81 | 81 | else: |
|
82 | 82 | print("Called as cell magic") |
|
83 | 83 | return line, cell |
|
84 | 84 | |
|
85 | 85 | |
|
86 | 86 | # In order to actually use these magics, you must register them with a |
|
87 | 87 | # running IPython. |
|
88 | 88 | |
|
89 | 89 | def load_ipython_extension(ipython): |
|
90 | 90 | """ |
|
91 | 91 | Any module file that define a function named `load_ipython_extension` |
|
92 | 92 | can be loaded via `%load_ext module.path` or be configured to be |
|
93 | 93 | autoloaded by IPython at startup time. |
|
94 | 94 | """ |
|
95 | 95 | # You can register the class itself without instantiating it. IPython will |
|
96 | 96 | # call the default constructor on it. |
|
97 | 97 | ipython.register_magics(MyMagics) |
|
98 | 98 | |
|
99 | 99 | If you want to create a class with a different constructor that holds |
|
100 | 100 | additional state, then you should always call the parent constructor and |
|
101 | 101 | instantiate the class yourself before registration: |
|
102 | 102 | |
|
103 | 103 | .. sourcecode:: python |
|
104 | 104 | |
|
105 | 105 | @magics_class |
|
106 | 106 | class StatefulMagics(Magics): |
|
107 | 107 | "Magics that hold additional state" |
|
108 | 108 | |
|
109 | 109 | def __init__(self, shell, data): |
|
110 | 110 | # You must call the parent constructor |
|
111 | 111 | super(StatefulMagics, self).__init__(shell) |
|
112 | 112 | self.data = data |
|
113 | 113 | |
|
114 | 114 | # etc... |
|
115 | 115 | |
|
116 | 116 | def load_ipython_extension(ipython): |
|
117 | 117 | """ |
|
118 | 118 | Any module file that define a function named `load_ipython_extension` |
|
119 | 119 | can be loaded via `%load_ext module.path` or be configured to be |
|
120 | 120 | autoloaded by IPython at startup time. |
|
121 | 121 | """ |
|
122 | 122 | # This class must then be registered with a manually created instance, |
|
123 | 123 | # since its constructor has different arguments from the default: |
|
124 | 124 | magics = StatefulMagics(ipython, some_data) |
|
125 | 125 | ipython.register_magics(magics) |
|
126 | 126 | |
|
127 | 127 | |
|
128 | 128 | .. note:: |
|
129 | 129 | |
|
130 | 130 | In early IPython versions 0.12 and before the line magics were |
|
131 | 131 | created using a :func:`define_magic` API function. This API has been |
|
132 | 132 | replaced with the above in IPython 0.13 and then completely removed |
|
133 | 133 | in IPython 5. Maintainers of IPython extensions that still use the |
|
134 | 134 | :func:`define_magic` function are advised to adjust their code |
|
135 | 135 | for the current API. |
|
136 | 136 | |
|
137 | 137 | |
|
138 | 138 | Accessing user namespace and local scope |
|
139 | 139 | ======================================== |
|
140 | 140 | |
|
141 | 141 | When creating line magics, you may need to access surrounding scope to get user |
|
142 | 142 | variables (e.g when called inside functions). IPython provide the |
|
143 | 143 | ``@needs_local_scope`` decorator that can be imported from |
|
144 | 144 | ``IPython.core.magics``. When decorated with ``@needs_local_scope`` a magic will |
|
145 | 145 | be passed ``local_ns`` as an argument. As a convenience ``@needs_local_scope`` |
|
146 | 146 | can also be applied to cell magics even if cell magics cannot appear at local |
|
147 | 147 | scope context. |
|
148 | 148 | |
|
149 | 149 | Complete Example |
|
150 | 150 | ================ |
|
151 | 151 | |
|
152 | 152 | Here is a full example of a magic package. You can distribute magics using |
|
153 | 153 | setuptools, distutils, or any other distribution tools like `flit |
|
154 | 154 | <http://flit.readthedocs.io>`_ for pure Python packages. |
|
155 | 155 | |
|
156 | When distributing magics as part of a package, recommended best practice is to | |
|
157 | execute the registration inside the `load_ipython_extension` as demonstrated in | |
|
158 | the example below, instead of directly in the module (as in the initial example | |
|
159 | with the ``@register_*`` decorators). This means a user will need to explicitly | |
|
160 | choose to load your magic with ``%load_ext``. instead implicitly getting it when | |
|
161 | importing the module. This is particularly relevant if loading your magic has | |
|
162 | side effects, if it is slow to load, or if it might override another magic with | |
|
163 | the same name. | |
|
156 | 164 | |
|
157 | 165 | .. sourcecode:: bash |
|
158 | 166 | |
|
159 | 167 | . |
|
160 | 168 | ├── example_magic |
|
161 | 169 | │ ├── __init__.py |
|
162 | 170 | │ └── abracadabra.py |
|
163 | 171 | └── setup.py |
|
164 | 172 | |
|
165 | 173 | .. sourcecode:: bash |
|
166 | 174 | |
|
167 | 175 | $ cat example_magic/__init__.py |
|
168 | 176 | """An example magic""" |
|
169 | 177 | __version__ = '0.0.1' |
|
170 | 178 | |
|
171 | 179 | from .abracadabra import Abracadabra |
|
172 | 180 | |
|
173 | 181 | def load_ipython_extension(ipython): |
|
174 | 182 | ipython.register_magics(Abracadabra) |
|
175 | 183 | |
|
176 | 184 | .. sourcecode:: bash |
|
177 | 185 | |
|
178 | 186 | $ cat example_magic/abracadabra.py |
|
179 | 187 | from IPython.core.magic import (Magics, magics_class, line_magic, cell_magic) |
|
180 | 188 | |
|
181 | 189 | @magics_class |
|
182 | 190 | class Abracadabra(Magics): |
|
183 | 191 | |
|
184 | 192 | @line_magic |
|
185 | 193 | def abra(self, line): |
|
186 | 194 | return line |
|
187 | 195 | |
|
188 | 196 | @cell_magic |
|
189 | 197 | def cadabra(self, line, cell): |
|
190 | 198 | return line, cell |
|
191 | 199 |
General Comments 0
You need to be logged in to leave comments.
Login now