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