Show More
@@ -1,157 +1,159 b'' | |||||
1 | # PYTHON_ARGCOMPLETE_OK |
|
1 | # PYTHON_ARGCOMPLETE_OK | |
2 | """ |
|
2 | """ | |
3 | IPython: tools for interactive and parallel computing in Python. |
|
3 | IPython: tools for interactive and parallel computing in Python. | |
4 |
|
4 | |||
5 | https://ipython.org |
|
5 | https://ipython.org | |
6 | """ |
|
6 | """ | |
7 | #----------------------------------------------------------------------------- |
|
7 | #----------------------------------------------------------------------------- | |
8 | # Copyright (c) 2008-2011, IPython Development Team. |
|
8 | # Copyright (c) 2008-2011, IPython Development Team. | |
9 | # Copyright (c) 2001-2007, Fernando Perez <fernando.perez@colorado.edu> |
|
9 | # Copyright (c) 2001-2007, Fernando Perez <fernando.perez@colorado.edu> | |
10 | # Copyright (c) 2001, Janko Hauser <jhauser@zscout.de> |
|
10 | # Copyright (c) 2001, Janko Hauser <jhauser@zscout.de> | |
11 | # Copyright (c) 2001, Nathaniel Gray <n8gray@caltech.edu> |
|
11 | # Copyright (c) 2001, Nathaniel Gray <n8gray@caltech.edu> | |
12 | # |
|
12 | # | |
13 | # Distributed under the terms of the Modified BSD License. |
|
13 | # Distributed under the terms of the Modified BSD License. | |
14 | # |
|
14 | # | |
15 | # The full license is in the file COPYING.txt, distributed with this software. |
|
15 | # The full license is in the file COPYING.txt, distributed with this software. | |
16 | #----------------------------------------------------------------------------- |
|
16 | #----------------------------------------------------------------------------- | |
17 |
|
17 | |||
18 | #----------------------------------------------------------------------------- |
|
18 | #----------------------------------------------------------------------------- | |
19 | # Imports |
|
19 | # Imports | |
20 | #----------------------------------------------------------------------------- |
|
20 | #----------------------------------------------------------------------------- | |
21 |
|
21 | |||
22 | import sys |
|
22 | import sys | |
23 |
|
23 | |||
24 | #----------------------------------------------------------------------------- |
|
24 | #----------------------------------------------------------------------------- | |
25 | # Setup everything |
|
25 | # Setup everything | |
26 | #----------------------------------------------------------------------------- |
|
26 | #----------------------------------------------------------------------------- | |
27 |
|
27 | |||
28 | # Don't forget to also update setup.py when this changes! |
|
28 | # Don't forget to also update setup.py when this changes! | |
29 | if sys.version_info < (3, 8): |
|
29 | if sys.version_info < (3, 8): | |
30 | raise ImportError( |
|
30 | raise ImportError( | |
31 | """ |
|
31 | """ | |
32 | IPython 8+ supports Python 3.8 and above, following NEP 29. |
|
32 | IPython 8+ supports Python 3.8 and above, following NEP 29. | |
33 | When using Python 2.7, please install IPython 5.x LTS Long Term Support version. |
|
33 | When using Python 2.7, please install IPython 5.x LTS Long Term Support version. | |
34 | Python 3.3 and 3.4 were supported up to IPython 6.x. |
|
34 | Python 3.3 and 3.4 were supported up to IPython 6.x. | |
35 | Python 3.5 was supported with IPython 7.0 to 7.9. |
|
35 | Python 3.5 was supported with IPython 7.0 to 7.9. | |
36 | Python 3.6 was supported with IPython up to 7.16. |
|
36 | Python 3.6 was supported with IPython up to 7.16. | |
37 | Python 3.7 was still supported with the 7.x branch. |
|
37 | Python 3.7 was still supported with the 7.x branch. | |
38 |
|
38 | |||
39 | See IPython `README.rst` file for more information: |
|
39 | See IPython `README.rst` file for more information: | |
40 |
|
40 | |||
41 | https://github.com/ipython/ipython/blob/main/README.rst |
|
41 | https://github.com/ipython/ipython/blob/main/README.rst | |
42 |
|
42 | |||
43 | """ |
|
43 | """ | |
44 | ) |
|
44 | ) | |
45 |
|
45 | |||
46 | #----------------------------------------------------------------------------- |
|
46 | #----------------------------------------------------------------------------- | |
47 | # Setup the top level names |
|
47 | # Setup the top level names | |
48 | #----------------------------------------------------------------------------- |
|
48 | #----------------------------------------------------------------------------- | |
49 |
|
49 | |||
50 | from .core.getipython import get_ipython |
|
50 | from .core.getipython import get_ipython | |
51 | from .core import release |
|
51 | from .core import release | |
52 | from .core.application import Application |
|
52 | from .core.application import Application | |
53 | from .terminal.embed import embed |
|
53 | from .terminal.embed import embed | |
54 |
|
54 | |||
55 | from .core.interactiveshell import InteractiveShell |
|
55 | from .core.interactiveshell import InteractiveShell | |
56 | from .utils.sysinfo import sys_info |
|
56 | from .utils.sysinfo import sys_info | |
57 | from .utils.frame import extract_module_locals |
|
57 | from .utils.frame import extract_module_locals | |
58 |
|
58 | |||
59 | # Release data |
|
59 | # Release data | |
60 | __author__ = '%s <%s>' % (release.author, release.author_email) |
|
60 | __author__ = '%s <%s>' % (release.author, release.author_email) | |
61 | __license__ = release.license |
|
61 | __license__ = release.license | |
62 | __version__ = release.version |
|
62 | __version__ = release.version | |
63 | version_info = release.version_info |
|
63 | version_info = release.version_info | |
64 | # list of CVEs that should have been patched in this release. |
|
64 | # list of CVEs that should have been patched in this release. | |
65 | # this is informational and should not be relied upon. |
|
65 | # this is informational and should not be relied upon. | |
66 | __patched_cves__ = {"CVE-2022-21699", "CVE-2023-24816"} |
|
66 | __patched_cves__ = {"CVE-2022-21699", "CVE-2023-24816"} | |
67 |
|
67 | |||
68 |
|
68 | |||
69 | def embed_kernel(module=None, local_ns=None, **kwargs): |
|
69 | def embed_kernel(module=None, local_ns=None, **kwargs): | |
70 | """Embed and start an IPython kernel in a given scope. |
|
70 | """Embed and start an IPython kernel in a given scope. | |
71 |
|
71 | |||
72 | If you don't want the kernel to initialize the namespace |
|
72 | If you don't want the kernel to initialize the namespace | |
73 | from the scope of the surrounding function, |
|
73 | from the scope of the surrounding function, | |
74 | and/or you want to load full IPython configuration, |
|
74 | and/or you want to load full IPython configuration, | |
75 | you probably want `IPython.start_kernel()` instead. |
|
75 | you probably want `IPython.start_kernel()` instead. | |
76 |
|
76 | |||
77 | Parameters |
|
77 | Parameters | |
78 | ---------- |
|
78 | ---------- | |
79 | module : types.ModuleType, optional |
|
79 | module : types.ModuleType, optional | |
80 | The module to load into IPython globals (default: caller) |
|
80 | The module to load into IPython globals (default: caller) | |
81 | local_ns : dict, optional |
|
81 | local_ns : dict, optional | |
82 | The namespace to load into IPython user namespace (default: caller) |
|
82 | The namespace to load into IPython user namespace (default: caller) | |
83 | **kwargs : various, optional |
|
83 | **kwargs : various, optional | |
84 | Further keyword args are relayed to the IPKernelApp constructor, |
|
84 | Further keyword args are relayed to the IPKernelApp constructor, | |
85 | such as `config`, a traitlets :class:`Config` object (see :ref:`configure_start_ipython`), |
|
85 | such as `config`, a traitlets :class:`Config` object (see :ref:`configure_start_ipython`), | |
86 |
allowing configuration of the |
|
86 | allowing configuration of the kernel (see :ref:`kernel_options`). Will only have an effect | |
87 | on the first embed_kernel call for a given process. |
|
87 | on the first embed_kernel call for a given process. | |
88 | """ |
|
88 | """ | |
89 |
|
89 | |||
90 | (caller_module, caller_locals) = extract_module_locals(1) |
|
90 | (caller_module, caller_locals) = extract_module_locals(1) | |
91 | if module is None: |
|
91 | if module is None: | |
92 | module = caller_module |
|
92 | module = caller_module | |
93 | if local_ns is None: |
|
93 | if local_ns is None: | |
94 | local_ns = caller_locals |
|
94 | local_ns = caller_locals | |
95 |
|
95 | |||
96 | # Only import .zmq when we really need it |
|
96 | # Only import .zmq when we really need it | |
97 | from ipykernel.embed import embed_kernel as real_embed_kernel |
|
97 | from ipykernel.embed import embed_kernel as real_embed_kernel | |
98 | real_embed_kernel(module=module, local_ns=local_ns, **kwargs) |
|
98 | real_embed_kernel(module=module, local_ns=local_ns, **kwargs) | |
99 |
|
99 | |||
100 | def start_ipython(argv=None, **kwargs): |
|
100 | def start_ipython(argv=None, **kwargs): | |
101 | """Launch a normal IPython instance (as opposed to embedded) |
|
101 | """Launch a normal IPython instance (as opposed to embedded) | |
102 |
|
102 | |||
103 | `IPython.embed()` puts a shell in a particular calling scope, |
|
103 | `IPython.embed()` puts a shell in a particular calling scope, | |
104 | such as a function or method for debugging purposes, |
|
104 | such as a function or method for debugging purposes, | |
105 | which is often not desirable. |
|
105 | which is often not desirable. | |
106 |
|
106 | |||
107 | `start_ipython()` does full, regular IPython initialization, |
|
107 | `start_ipython()` does full, regular IPython initialization, | |
108 | including loading startup files, configuration, etc. |
|
108 | including loading startup files, configuration, etc. | |
109 | much of which is skipped by `embed()`. |
|
109 | much of which is skipped by `embed()`. | |
110 |
|
110 | |||
111 | This is a public API method, and will survive implementation changes. |
|
111 | This is a public API method, and will survive implementation changes. | |
112 |
|
112 | |||
113 | Parameters |
|
113 | Parameters | |
114 | ---------- |
|
114 | ---------- | |
115 | argv : list or None, optional |
|
115 | argv : list or None, optional | |
116 | If unspecified or None, IPython will parse command-line options from sys.argv. |
|
116 | If unspecified or None, IPython will parse command-line options from sys.argv. | |
117 | To prevent any command-line parsing, pass an empty list: `argv=[]`. |
|
117 | To prevent any command-line parsing, pass an empty list: `argv=[]`. | |
118 | user_ns : dict, optional |
|
118 | user_ns : dict, optional | |
119 | specify this dictionary to initialize the IPython user namespace with particular values. |
|
119 | specify this dictionary to initialize the IPython user namespace with particular values. | |
120 | **kwargs : various, optional |
|
120 | **kwargs : various, optional | |
121 | Any other kwargs will be passed to the Application constructor, |
|
121 | Any other kwargs will be passed to the Application constructor, | |
122 |
such as `config`, a traitlets :class:`Config` object (see :ref:`configure_start_ipython`) |
|
122 | such as `config`, a traitlets :class:`Config` object (see :ref:`configure_start_ipython`), | |
|
123 | allowing configuration of the instance (see :ref:`terminal_options`). | |||
123 | """ |
|
124 | """ | |
124 | from IPython.terminal.ipapp import launch_new_instance |
|
125 | from IPython.terminal.ipapp import launch_new_instance | |
125 | return launch_new_instance(argv=argv, **kwargs) |
|
126 | return launch_new_instance(argv=argv, **kwargs) | |
126 |
|
127 | |||
127 | def start_kernel(argv=None, **kwargs): |
|
128 | def start_kernel(argv=None, **kwargs): | |
128 | """Launch a normal IPython kernel instance (as opposed to embedded) |
|
129 | """Launch a normal IPython kernel instance (as opposed to embedded) | |
129 |
|
130 | |||
130 | `IPython.embed_kernel()` puts a shell in a particular calling scope, |
|
131 | `IPython.embed_kernel()` puts a shell in a particular calling scope, | |
131 | such as a function or method for debugging purposes, |
|
132 | such as a function or method for debugging purposes, | |
132 | which is often not desirable. |
|
133 | which is often not desirable. | |
133 |
|
134 | |||
134 | `start_kernel()` does full, regular IPython initialization, |
|
135 | `start_kernel()` does full, regular IPython initialization, | |
135 | including loading startup files, configuration, etc. |
|
136 | including loading startup files, configuration, etc. | |
136 | much of which is skipped by `embed_kernel()`. |
|
137 | much of which is skipped by `embed_kernel()`. | |
137 |
|
138 | |||
138 | Parameters |
|
139 | Parameters | |
139 | ---------- |
|
140 | ---------- | |
140 | argv : list or None, optional |
|
141 | argv : list or None, optional | |
141 | If unspecified or None, IPython will parse command-line options from sys.argv. |
|
142 | If unspecified or None, IPython will parse command-line options from sys.argv. | |
142 | To prevent any command-line parsing, pass an empty list: `argv=[]`. |
|
143 | To prevent any command-line parsing, pass an empty list: `argv=[]`. | |
143 | user_ns : dict, optional |
|
144 | user_ns : dict, optional | |
144 | specify this dictionary to initialize the IPython user namespace with particular values. |
|
145 | specify this dictionary to initialize the IPython user namespace with particular values. | |
145 | **kwargs : various, optional |
|
146 | **kwargs : various, optional | |
146 | Any other kwargs will be passed to the Application constructor, |
|
147 | Any other kwargs will be passed to the Application constructor, | |
147 |
such as `config`, a traitlets :class:`Config` object (see :ref:`configure_start_ipython`) |
|
148 | such as `config`, a traitlets :class:`Config` object (see :ref:`configure_start_ipython`), | |
|
149 | allowing configuration of the kernel (see :ref:`kernel_options`). | |||
148 | """ |
|
150 | """ | |
149 | import warnings |
|
151 | import warnings | |
150 |
|
152 | |||
151 | warnings.warn( |
|
153 | warnings.warn( | |
152 | "start_kernel is deprecated since IPython 8.0, use from `ipykernel.kernelapp.launch_new_instance`", |
|
154 | "start_kernel is deprecated since IPython 8.0, use from `ipykernel.kernelapp.launch_new_instance`", | |
153 | DeprecationWarning, |
|
155 | DeprecationWarning, | |
154 | stacklevel=2, |
|
156 | stacklevel=2, | |
155 | ) |
|
157 | ) | |
156 | from ipykernel.kernelapp import launch_new_instance |
|
158 | from ipykernel.kernelapp import launch_new_instance | |
157 | return launch_new_instance(argv=argv, **kwargs) |
|
159 | return launch_new_instance(argv=argv, **kwargs) |
@@ -1,420 +1,420 b'' | |||||
1 | # encoding: utf-8 |
|
1 | # encoding: utf-8 | |
2 | """ |
|
2 | """ | |
3 | An embedded IPython shell. |
|
3 | An embedded IPython shell. | |
4 | """ |
|
4 | """ | |
5 | # Copyright (c) IPython Development Team. |
|
5 | # Copyright (c) IPython Development Team. | |
6 | # Distributed under the terms of the Modified BSD License. |
|
6 | # Distributed under the terms of the Modified BSD License. | |
7 |
|
7 | |||
8 |
|
8 | |||
9 | import sys |
|
9 | import sys | |
10 | import warnings |
|
10 | import warnings | |
11 |
|
11 | |||
12 | from IPython.core import ultratb, compilerop |
|
12 | from IPython.core import ultratb, compilerop | |
13 | from IPython.core import magic_arguments |
|
13 | from IPython.core import magic_arguments | |
14 | from IPython.core.magic import Magics, magics_class, line_magic |
|
14 | from IPython.core.magic import Magics, magics_class, line_magic | |
15 | from IPython.core.interactiveshell import DummyMod, InteractiveShell |
|
15 | from IPython.core.interactiveshell import DummyMod, InteractiveShell | |
16 | from IPython.terminal.interactiveshell import TerminalInteractiveShell |
|
16 | from IPython.terminal.interactiveshell import TerminalInteractiveShell | |
17 | from IPython.terminal.ipapp import load_default_config |
|
17 | from IPython.terminal.ipapp import load_default_config | |
18 |
|
18 | |||
19 | from traitlets import Bool, CBool, Unicode |
|
19 | from traitlets import Bool, CBool, Unicode | |
20 | from IPython.utils.io import ask_yes_no |
|
20 | from IPython.utils.io import ask_yes_no | |
21 |
|
21 | |||
22 | from typing import Set |
|
22 | from typing import Set | |
23 |
|
23 | |||
24 | class KillEmbedded(Exception):pass |
|
24 | class KillEmbedded(Exception):pass | |
25 |
|
25 | |||
26 | # kept for backward compatibility as IPython 6 was released with |
|
26 | # kept for backward compatibility as IPython 6 was released with | |
27 | # the typo. See https://github.com/ipython/ipython/pull/10706 |
|
27 | # the typo. See https://github.com/ipython/ipython/pull/10706 | |
28 | KillEmbeded = KillEmbedded |
|
28 | KillEmbeded = KillEmbedded | |
29 |
|
29 | |||
30 | # This is an additional magic that is exposed in embedded shells. |
|
30 | # This is an additional magic that is exposed in embedded shells. | |
31 | @magics_class |
|
31 | @magics_class | |
32 | class EmbeddedMagics(Magics): |
|
32 | class EmbeddedMagics(Magics): | |
33 |
|
33 | |||
34 | @line_magic |
|
34 | @line_magic | |
35 | @magic_arguments.magic_arguments() |
|
35 | @magic_arguments.magic_arguments() | |
36 | @magic_arguments.argument('-i', '--instance', action='store_true', |
|
36 | @magic_arguments.argument('-i', '--instance', action='store_true', | |
37 | help='Kill instance instead of call location') |
|
37 | help='Kill instance instead of call location') | |
38 | @magic_arguments.argument('-x', '--exit', action='store_true', |
|
38 | @magic_arguments.argument('-x', '--exit', action='store_true', | |
39 | help='Also exit the current session') |
|
39 | help='Also exit the current session') | |
40 | @magic_arguments.argument('-y', '--yes', action='store_true', |
|
40 | @magic_arguments.argument('-y', '--yes', action='store_true', | |
41 | help='Do not ask confirmation') |
|
41 | help='Do not ask confirmation') | |
42 | def kill_embedded(self, parameter_s=''): |
|
42 | def kill_embedded(self, parameter_s=''): | |
43 | """%kill_embedded : deactivate for good the current embedded IPython |
|
43 | """%kill_embedded : deactivate for good the current embedded IPython | |
44 |
|
44 | |||
45 | This function (after asking for confirmation) sets an internal flag so |
|
45 | This function (after asking for confirmation) sets an internal flag so | |
46 | that an embedded IPython will never activate again for the given call |
|
46 | that an embedded IPython will never activate again for the given call | |
47 | location. This is useful to permanently disable a shell that is being |
|
47 | location. This is useful to permanently disable a shell that is being | |
48 | called inside a loop: once you've figured out what you needed from it, |
|
48 | called inside a loop: once you've figured out what you needed from it, | |
49 | you may then kill it and the program will then continue to run without |
|
49 | you may then kill it and the program will then continue to run without | |
50 | the interactive shell interfering again. |
|
50 | the interactive shell interfering again. | |
51 |
|
51 | |||
52 | Kill Instance Option: |
|
52 | Kill Instance Option: | |
53 |
|
53 | |||
54 | If for some reasons you need to kill the location where the instance |
|
54 | If for some reasons you need to kill the location where the instance | |
55 | is created and not called, for example if you create a single |
|
55 | is created and not called, for example if you create a single | |
56 | instance in one place and debug in many locations, you can use the |
|
56 | instance in one place and debug in many locations, you can use the | |
57 | ``--instance`` option to kill this specific instance. Like for the |
|
57 | ``--instance`` option to kill this specific instance. Like for the | |
58 | ``call location`` killing an "instance" should work even if it is |
|
58 | ``call location`` killing an "instance" should work even if it is | |
59 | recreated within a loop. |
|
59 | recreated within a loop. | |
60 |
|
60 | |||
61 | .. note:: |
|
61 | .. note:: | |
62 |
|
62 | |||
63 | This was the default behavior before IPython 5.2 |
|
63 | This was the default behavior before IPython 5.2 | |
64 |
|
64 | |||
65 | """ |
|
65 | """ | |
66 |
|
66 | |||
67 | args = magic_arguments.parse_argstring(self.kill_embedded, parameter_s) |
|
67 | args = magic_arguments.parse_argstring(self.kill_embedded, parameter_s) | |
68 | print(args) |
|
68 | print(args) | |
69 | if args.instance: |
|
69 | if args.instance: | |
70 | # let no ask |
|
70 | # let no ask | |
71 | if not args.yes: |
|
71 | if not args.yes: | |
72 | kill = ask_yes_no( |
|
72 | kill = ask_yes_no( | |
73 | "Are you sure you want to kill this embedded instance? [y/N] ", 'n') |
|
73 | "Are you sure you want to kill this embedded instance? [y/N] ", 'n') | |
74 | else: |
|
74 | else: | |
75 | kill = True |
|
75 | kill = True | |
76 | if kill: |
|
76 | if kill: | |
77 | self.shell._disable_init_location() |
|
77 | self.shell._disable_init_location() | |
78 | print("This embedded IPython instance will not reactivate anymore " |
|
78 | print("This embedded IPython instance will not reactivate anymore " | |
79 | "once you exit.") |
|
79 | "once you exit.") | |
80 | else: |
|
80 | else: | |
81 | if not args.yes: |
|
81 | if not args.yes: | |
82 | kill = ask_yes_no( |
|
82 | kill = ask_yes_no( | |
83 | "Are you sure you want to kill this embedded call_location? [y/N] ", 'n') |
|
83 | "Are you sure you want to kill this embedded call_location? [y/N] ", 'n') | |
84 | else: |
|
84 | else: | |
85 | kill = True |
|
85 | kill = True | |
86 | if kill: |
|
86 | if kill: | |
87 | self.shell.embedded_active = False |
|
87 | self.shell.embedded_active = False | |
88 | print("This embedded IPython call location will not reactivate anymore " |
|
88 | print("This embedded IPython call location will not reactivate anymore " | |
89 | "once you exit.") |
|
89 | "once you exit.") | |
90 |
|
90 | |||
91 | if args.exit: |
|
91 | if args.exit: | |
92 | # Ask-exit does not really ask, it just set internals flags to exit |
|
92 | # Ask-exit does not really ask, it just set internals flags to exit | |
93 | # on next loop. |
|
93 | # on next loop. | |
94 | self.shell.ask_exit() |
|
94 | self.shell.ask_exit() | |
95 |
|
95 | |||
96 |
|
96 | |||
97 | @line_magic |
|
97 | @line_magic | |
98 | def exit_raise(self, parameter_s=''): |
|
98 | def exit_raise(self, parameter_s=''): | |
99 | """%exit_raise Make the current embedded kernel exit and raise and exception. |
|
99 | """%exit_raise Make the current embedded kernel exit and raise and exception. | |
100 |
|
100 | |||
101 | This function sets an internal flag so that an embedded IPython will |
|
101 | This function sets an internal flag so that an embedded IPython will | |
102 | raise a `IPython.terminal.embed.KillEmbedded` Exception on exit, and then exit the current I. This is |
|
102 | raise a `IPython.terminal.embed.KillEmbedded` Exception on exit, and then exit the current I. This is | |
103 | useful to permanently exit a loop that create IPython embed instance. |
|
103 | useful to permanently exit a loop that create IPython embed instance. | |
104 | """ |
|
104 | """ | |
105 |
|
105 | |||
106 | self.shell.should_raise = True |
|
106 | self.shell.should_raise = True | |
107 | self.shell.ask_exit() |
|
107 | self.shell.ask_exit() | |
108 |
|
108 | |||
109 |
|
109 | |||
110 | class _Sentinel: |
|
110 | class _Sentinel: | |
111 | def __init__(self, repr): |
|
111 | def __init__(self, repr): | |
112 | assert isinstance(repr, str) |
|
112 | assert isinstance(repr, str) | |
113 | self.repr = repr |
|
113 | self.repr = repr | |
114 |
|
114 | |||
115 | def __repr__(self): |
|
115 | def __repr__(self): | |
116 | return repr |
|
116 | return repr | |
117 |
|
117 | |||
118 |
|
118 | |||
119 | class InteractiveShellEmbed(TerminalInteractiveShell): |
|
119 | class InteractiveShellEmbed(TerminalInteractiveShell): | |
120 |
|
120 | |||
121 | dummy_mode = Bool(False) |
|
121 | dummy_mode = Bool(False) | |
122 | exit_msg = Unicode('') |
|
122 | exit_msg = Unicode('') | |
123 | embedded = CBool(True) |
|
123 | embedded = CBool(True) | |
124 | should_raise = CBool(False) |
|
124 | should_raise = CBool(False) | |
125 | # Like the base class display_banner is not configurable, but here it |
|
125 | # Like the base class display_banner is not configurable, but here it | |
126 | # is True by default. |
|
126 | # is True by default. | |
127 | display_banner = CBool(True) |
|
127 | display_banner = CBool(True) | |
128 | exit_msg = Unicode() |
|
128 | exit_msg = Unicode() | |
129 |
|
129 | |||
130 | # When embedding, by default we don't change the terminal title |
|
130 | # When embedding, by default we don't change the terminal title | |
131 | term_title = Bool(False, |
|
131 | term_title = Bool(False, | |
132 | help="Automatically set the terminal title" |
|
132 | help="Automatically set the terminal title" | |
133 | ).tag(config=True) |
|
133 | ).tag(config=True) | |
134 |
|
134 | |||
135 | _inactive_locations: Set[str] = set() |
|
135 | _inactive_locations: Set[str] = set() | |
136 |
|
136 | |||
137 | def _disable_init_location(self): |
|
137 | def _disable_init_location(self): | |
138 | """Disable the current Instance creation location""" |
|
138 | """Disable the current Instance creation location""" | |
139 | InteractiveShellEmbed._inactive_locations.add(self._init_location_id) |
|
139 | InteractiveShellEmbed._inactive_locations.add(self._init_location_id) | |
140 |
|
140 | |||
141 | @property |
|
141 | @property | |
142 | def embedded_active(self): |
|
142 | def embedded_active(self): | |
143 | return (self._call_location_id not in InteractiveShellEmbed._inactive_locations)\ |
|
143 | return (self._call_location_id not in InteractiveShellEmbed._inactive_locations)\ | |
144 | and (self._init_location_id not in InteractiveShellEmbed._inactive_locations) |
|
144 | and (self._init_location_id not in InteractiveShellEmbed._inactive_locations) | |
145 |
|
145 | |||
146 | @embedded_active.setter |
|
146 | @embedded_active.setter | |
147 | def embedded_active(self, value): |
|
147 | def embedded_active(self, value): | |
148 | if value: |
|
148 | if value: | |
149 | InteractiveShellEmbed._inactive_locations.discard( |
|
149 | InteractiveShellEmbed._inactive_locations.discard( | |
150 | self._call_location_id) |
|
150 | self._call_location_id) | |
151 | InteractiveShellEmbed._inactive_locations.discard( |
|
151 | InteractiveShellEmbed._inactive_locations.discard( | |
152 | self._init_location_id) |
|
152 | self._init_location_id) | |
153 | else: |
|
153 | else: | |
154 | InteractiveShellEmbed._inactive_locations.add( |
|
154 | InteractiveShellEmbed._inactive_locations.add( | |
155 | self._call_location_id) |
|
155 | self._call_location_id) | |
156 |
|
156 | |||
157 | def __init__(self, **kw): |
|
157 | def __init__(self, **kw): | |
158 | assert ( |
|
158 | assert ( | |
159 | "user_global_ns" not in kw |
|
159 | "user_global_ns" not in kw | |
160 | ), "Key word argument `user_global_ns` has been replaced by `user_module` since IPython 4.0." |
|
160 | ), "Key word argument `user_global_ns` has been replaced by `user_module` since IPython 4.0." | |
161 |
|
161 | |||
162 | clid = kw.pop('_init_location_id', None) |
|
162 | clid = kw.pop('_init_location_id', None) | |
163 | if not clid: |
|
163 | if not clid: | |
164 | frame = sys._getframe(1) |
|
164 | frame = sys._getframe(1) | |
165 | clid = '%s:%s' % (frame.f_code.co_filename, frame.f_lineno) |
|
165 | clid = '%s:%s' % (frame.f_code.co_filename, frame.f_lineno) | |
166 | self._init_location_id = clid |
|
166 | self._init_location_id = clid | |
167 |
|
167 | |||
168 | super(InteractiveShellEmbed,self).__init__(**kw) |
|
168 | super(InteractiveShellEmbed,self).__init__(**kw) | |
169 |
|
169 | |||
170 | # don't use the ipython crash handler so that user exceptions aren't |
|
170 | # don't use the ipython crash handler so that user exceptions aren't | |
171 | # trapped |
|
171 | # trapped | |
172 | sys.excepthook = ultratb.FormattedTB(color_scheme=self.colors, |
|
172 | sys.excepthook = ultratb.FormattedTB(color_scheme=self.colors, | |
173 | mode=self.xmode, |
|
173 | mode=self.xmode, | |
174 | call_pdb=self.pdb) |
|
174 | call_pdb=self.pdb) | |
175 |
|
175 | |||
176 | def init_sys_modules(self): |
|
176 | def init_sys_modules(self): | |
177 | """ |
|
177 | """ | |
178 | Explicitly overwrite :mod:`IPython.core.interactiveshell` to do nothing. |
|
178 | Explicitly overwrite :mod:`IPython.core.interactiveshell` to do nothing. | |
179 | """ |
|
179 | """ | |
180 | pass |
|
180 | pass | |
181 |
|
181 | |||
182 | def init_magics(self): |
|
182 | def init_magics(self): | |
183 | super(InteractiveShellEmbed, self).init_magics() |
|
183 | super(InteractiveShellEmbed, self).init_magics() | |
184 | self.register_magics(EmbeddedMagics) |
|
184 | self.register_magics(EmbeddedMagics) | |
185 |
|
185 | |||
186 | def __call__( |
|
186 | def __call__( | |
187 | self, |
|
187 | self, | |
188 | header="", |
|
188 | header="", | |
189 | local_ns=None, |
|
189 | local_ns=None, | |
190 | module=None, |
|
190 | module=None, | |
191 | dummy=None, |
|
191 | dummy=None, | |
192 | stack_depth=1, |
|
192 | stack_depth=1, | |
193 | compile_flags=None, |
|
193 | compile_flags=None, | |
194 | **kw |
|
194 | **kw | |
195 | ): |
|
195 | ): | |
196 | """Activate the interactive interpreter. |
|
196 | """Activate the interactive interpreter. | |
197 |
|
197 | |||
198 | __call__(self,header='',local_ns=None,module=None,dummy=None) -> Start |
|
198 | __call__(self,header='',local_ns=None,module=None,dummy=None) -> Start | |
199 | the interpreter shell with the given local and global namespaces, and |
|
199 | the interpreter shell with the given local and global namespaces, and | |
200 | optionally print a header string at startup. |
|
200 | optionally print a header string at startup. | |
201 |
|
201 | |||
202 | The shell can be globally activated/deactivated using the |
|
202 | The shell can be globally activated/deactivated using the | |
203 | dummy_mode attribute. This allows you to turn off a shell used |
|
203 | dummy_mode attribute. This allows you to turn off a shell used | |
204 | for debugging globally. |
|
204 | for debugging globally. | |
205 |
|
205 | |||
206 | However, *each* time you call the shell you can override the current |
|
206 | However, *each* time you call the shell you can override the current | |
207 | state of dummy_mode with the optional keyword parameter 'dummy'. For |
|
207 | state of dummy_mode with the optional keyword parameter 'dummy'. For | |
208 | example, if you set dummy mode on with IPShell.dummy_mode = True, you |
|
208 | example, if you set dummy mode on with IPShell.dummy_mode = True, you | |
209 | can still have a specific call work by making it as IPShell(dummy=False). |
|
209 | can still have a specific call work by making it as IPShell(dummy=False). | |
210 | """ |
|
210 | """ | |
211 |
|
211 | |||
212 | # we are called, set the underlying interactiveshell not to exit. |
|
212 | # we are called, set the underlying interactiveshell not to exit. | |
213 | self.keep_running = True |
|
213 | self.keep_running = True | |
214 |
|
214 | |||
215 | # If the user has turned it off, go away |
|
215 | # If the user has turned it off, go away | |
216 | clid = kw.pop('_call_location_id', None) |
|
216 | clid = kw.pop('_call_location_id', None) | |
217 | if not clid: |
|
217 | if not clid: | |
218 | frame = sys._getframe(1) |
|
218 | frame = sys._getframe(1) | |
219 | clid = '%s:%s' % (frame.f_code.co_filename, frame.f_lineno) |
|
219 | clid = '%s:%s' % (frame.f_code.co_filename, frame.f_lineno) | |
220 | self._call_location_id = clid |
|
220 | self._call_location_id = clid | |
221 |
|
221 | |||
222 | if not self.embedded_active: |
|
222 | if not self.embedded_active: | |
223 | return |
|
223 | return | |
224 |
|
224 | |||
225 | # Normal exits from interactive mode set this flag, so the shell can't |
|
225 | # Normal exits from interactive mode set this flag, so the shell can't | |
226 | # re-enter (it checks this variable at the start of interactive mode). |
|
226 | # re-enter (it checks this variable at the start of interactive mode). | |
227 | self.exit_now = False |
|
227 | self.exit_now = False | |
228 |
|
228 | |||
229 | # Allow the dummy parameter to override the global __dummy_mode |
|
229 | # Allow the dummy parameter to override the global __dummy_mode | |
230 | if dummy or (dummy != 0 and self.dummy_mode): |
|
230 | if dummy or (dummy != 0 and self.dummy_mode): | |
231 | return |
|
231 | return | |
232 |
|
232 | |||
233 | # self.banner is auto computed |
|
233 | # self.banner is auto computed | |
234 | if header: |
|
234 | if header: | |
235 | self.old_banner2 = self.banner2 |
|
235 | self.old_banner2 = self.banner2 | |
236 | self.banner2 = self.banner2 + '\n' + header + '\n' |
|
236 | self.banner2 = self.banner2 + '\n' + header + '\n' | |
237 | else: |
|
237 | else: | |
238 | self.old_banner2 = '' |
|
238 | self.old_banner2 = '' | |
239 |
|
239 | |||
240 | if self.display_banner: |
|
240 | if self.display_banner: | |
241 | self.show_banner() |
|
241 | self.show_banner() | |
242 |
|
242 | |||
243 | # Call the embedding code with a stack depth of 1 so it can skip over |
|
243 | # Call the embedding code with a stack depth of 1 so it can skip over | |
244 | # our call and get the original caller's namespaces. |
|
244 | # our call and get the original caller's namespaces. | |
245 | self.mainloop( |
|
245 | self.mainloop( | |
246 | local_ns, module, stack_depth=stack_depth, compile_flags=compile_flags |
|
246 | local_ns, module, stack_depth=stack_depth, compile_flags=compile_flags | |
247 | ) |
|
247 | ) | |
248 |
|
248 | |||
249 | self.banner2 = self.old_banner2 |
|
249 | self.banner2 = self.old_banner2 | |
250 |
|
250 | |||
251 | if self.exit_msg is not None: |
|
251 | if self.exit_msg is not None: | |
252 | print(self.exit_msg) |
|
252 | print(self.exit_msg) | |
253 |
|
253 | |||
254 | if self.should_raise: |
|
254 | if self.should_raise: | |
255 | raise KillEmbedded('Embedded IPython raising error, as user requested.') |
|
255 | raise KillEmbedded('Embedded IPython raising error, as user requested.') | |
256 |
|
256 | |||
257 | def mainloop( |
|
257 | def mainloop( | |
258 | self, |
|
258 | self, | |
259 | local_ns=None, |
|
259 | local_ns=None, | |
260 | module=None, |
|
260 | module=None, | |
261 | stack_depth=0, |
|
261 | stack_depth=0, | |
262 | compile_flags=None, |
|
262 | compile_flags=None, | |
263 | ): |
|
263 | ): | |
264 | """Embeds IPython into a running python program. |
|
264 | """Embeds IPython into a running python program. | |
265 |
|
265 | |||
266 | Parameters |
|
266 | Parameters | |
267 | ---------- |
|
267 | ---------- | |
268 | local_ns, module |
|
268 | local_ns, module | |
269 | Working local namespace (a dict) and module (a module or similar |
|
269 | Working local namespace (a dict) and module (a module or similar | |
270 | object). If given as None, they are automatically taken from the scope |
|
270 | object). If given as None, they are automatically taken from the scope | |
271 | where the shell was called, so that program variables become visible. |
|
271 | where the shell was called, so that program variables become visible. | |
272 | stack_depth : int |
|
272 | stack_depth : int | |
273 | How many levels in the stack to go to looking for namespaces (when |
|
273 | How many levels in the stack to go to looking for namespaces (when | |
274 | local_ns or module is None). This allows an intermediate caller to |
|
274 | local_ns or module is None). This allows an intermediate caller to | |
275 | make sure that this function gets the namespace from the intended |
|
275 | make sure that this function gets the namespace from the intended | |
276 | level in the stack. By default (0) it will get its locals and globals |
|
276 | level in the stack. By default (0) it will get its locals and globals | |
277 | from the immediate caller. |
|
277 | from the immediate caller. | |
278 | compile_flags |
|
278 | compile_flags | |
279 | A bit field identifying the __future__ features |
|
279 | A bit field identifying the __future__ features | |
280 | that are enabled, as passed to the builtin :func:`compile` function. |
|
280 | that are enabled, as passed to the builtin :func:`compile` function. | |
281 | If given as None, they are automatically taken from the scope where |
|
281 | If given as None, they are automatically taken from the scope where | |
282 | the shell was called. |
|
282 | the shell was called. | |
283 |
|
283 | |||
284 | """ |
|
284 | """ | |
285 |
|
285 | |||
286 | # Get locals and globals from caller |
|
286 | # Get locals and globals from caller | |
287 | if ((local_ns is None or module is None or compile_flags is None) |
|
287 | if ((local_ns is None or module is None or compile_flags is None) | |
288 | and self.default_user_namespaces): |
|
288 | and self.default_user_namespaces): | |
289 | call_frame = sys._getframe(stack_depth).f_back |
|
289 | call_frame = sys._getframe(stack_depth).f_back | |
290 |
|
290 | |||
291 | if local_ns is None: |
|
291 | if local_ns is None: | |
292 | local_ns = call_frame.f_locals |
|
292 | local_ns = call_frame.f_locals | |
293 | if module is None: |
|
293 | if module is None: | |
294 | global_ns = call_frame.f_globals |
|
294 | global_ns = call_frame.f_globals | |
295 | try: |
|
295 | try: | |
296 | module = sys.modules[global_ns['__name__']] |
|
296 | module = sys.modules[global_ns['__name__']] | |
297 | except KeyError: |
|
297 | except KeyError: | |
298 | warnings.warn("Failed to get module %s" % \ |
|
298 | warnings.warn("Failed to get module %s" % \ | |
299 | global_ns.get('__name__', 'unknown module') |
|
299 | global_ns.get('__name__', 'unknown module') | |
300 | ) |
|
300 | ) | |
301 | module = DummyMod() |
|
301 | module = DummyMod() | |
302 | module.__dict__ = global_ns |
|
302 | module.__dict__ = global_ns | |
303 | if compile_flags is None: |
|
303 | if compile_flags is None: | |
304 | compile_flags = (call_frame.f_code.co_flags & |
|
304 | compile_flags = (call_frame.f_code.co_flags & | |
305 | compilerop.PyCF_MASK) |
|
305 | compilerop.PyCF_MASK) | |
306 |
|
306 | |||
307 | # Save original namespace and module so we can restore them after |
|
307 | # Save original namespace and module so we can restore them after | |
308 | # embedding; otherwise the shell doesn't shut down correctly. |
|
308 | # embedding; otherwise the shell doesn't shut down correctly. | |
309 | orig_user_module = self.user_module |
|
309 | orig_user_module = self.user_module | |
310 | orig_user_ns = self.user_ns |
|
310 | orig_user_ns = self.user_ns | |
311 | orig_compile_flags = self.compile.flags |
|
311 | orig_compile_flags = self.compile.flags | |
312 |
|
312 | |||
313 | # Update namespaces and fire up interpreter |
|
313 | # Update namespaces and fire up interpreter | |
314 |
|
314 | |||
315 | # The global one is easy, we can just throw it in |
|
315 | # The global one is easy, we can just throw it in | |
316 | if module is not None: |
|
316 | if module is not None: | |
317 | self.user_module = module |
|
317 | self.user_module = module | |
318 |
|
318 | |||
319 | # But the user/local one is tricky: ipython needs it to store internal |
|
319 | # But the user/local one is tricky: ipython needs it to store internal | |
320 | # data, but we also need the locals. We'll throw our hidden variables |
|
320 | # data, but we also need the locals. We'll throw our hidden variables | |
321 | # like _ih and get_ipython() into the local namespace, but delete them |
|
321 | # like _ih and get_ipython() into the local namespace, but delete them | |
322 | # later. |
|
322 | # later. | |
323 | if local_ns is not None: |
|
323 | if local_ns is not None: | |
324 | reentrant_local_ns = {k: v for (k, v) in local_ns.items() if k not in self.user_ns_hidden.keys()} |
|
324 | reentrant_local_ns = {k: v for (k, v) in local_ns.items() if k not in self.user_ns_hidden.keys()} | |
325 | self.user_ns = reentrant_local_ns |
|
325 | self.user_ns = reentrant_local_ns | |
326 | self.init_user_ns() |
|
326 | self.init_user_ns() | |
327 |
|
327 | |||
328 | # Compiler flags |
|
328 | # Compiler flags | |
329 | if compile_flags is not None: |
|
329 | if compile_flags is not None: | |
330 | self.compile.flags = compile_flags |
|
330 | self.compile.flags = compile_flags | |
331 |
|
331 | |||
332 | # make sure the tab-completer has the correct frame information, so it |
|
332 | # make sure the tab-completer has the correct frame information, so it | |
333 | # actually completes using the frame's locals/globals |
|
333 | # actually completes using the frame's locals/globals | |
334 | self.set_completer_frame() |
|
334 | self.set_completer_frame() | |
335 |
|
335 | |||
336 | with self.builtin_trap, self.display_trap: |
|
336 | with self.builtin_trap, self.display_trap: | |
337 | self.interact() |
|
337 | self.interact() | |
338 |
|
338 | |||
339 | # now, purge out the local namespace of IPython's hidden variables. |
|
339 | # now, purge out the local namespace of IPython's hidden variables. | |
340 | if local_ns is not None: |
|
340 | if local_ns is not None: | |
341 | local_ns.update({k: v for (k, v) in self.user_ns.items() if k not in self.user_ns_hidden.keys()}) |
|
341 | local_ns.update({k: v for (k, v) in self.user_ns.items() if k not in self.user_ns_hidden.keys()}) | |
342 |
|
342 | |||
343 |
|
343 | |||
344 | # Restore original namespace so shell can shut down when we exit. |
|
344 | # Restore original namespace so shell can shut down when we exit. | |
345 | self.user_module = orig_user_module |
|
345 | self.user_module = orig_user_module | |
346 | self.user_ns = orig_user_ns |
|
346 | self.user_ns = orig_user_ns | |
347 | self.compile.flags = orig_compile_flags |
|
347 | self.compile.flags = orig_compile_flags | |
348 |
|
348 | |||
349 |
|
349 | |||
350 | def embed(*, header="", compile_flags=None, **kwargs): |
|
350 | def embed(*, header="", compile_flags=None, **kwargs): | |
351 | """Call this to embed IPython at the current point in your program. |
|
351 | """Call this to embed IPython at the current point in your program. | |
352 |
|
352 | |||
353 | The first invocation of this will create an :class:`InteractiveShellEmbed` |
|
353 | The first invocation of this will create an :class:`InteractiveShellEmbed` | |
354 | instance and then call it. Consecutive calls just call the already |
|
354 | instance and then call it. Consecutive calls just call the already | |
355 | created instance. |
|
355 | created instance. | |
356 |
|
356 | |||
357 | If you don't want the kernel to initialize the namespace |
|
357 | If you don't want the kernel to initialize the namespace | |
358 | from the scope of the surrounding function, |
|
358 | from the scope of the surrounding function, | |
359 | and/or you want to load full IPython configuration, |
|
359 | and/or you want to load full IPython configuration, | |
360 | you probably want `IPython.start_ipython()` instead. |
|
360 | you probably want `IPython.start_ipython()` instead. | |
361 |
|
361 | |||
362 | Here is a simple example:: |
|
362 | Here is a simple example:: | |
363 |
|
363 | |||
364 | from IPython import embed |
|
364 | from IPython import embed | |
365 | a = 10 |
|
365 | a = 10 | |
366 | b = 20 |
|
366 | b = 20 | |
367 | embed(header='First time') |
|
367 | embed(header='First time') | |
368 | c = 30 |
|
368 | c = 30 | |
369 | d = 40 |
|
369 | d = 40 | |
370 | embed() |
|
370 | embed() | |
371 |
|
371 | |||
372 | Parameters |
|
372 | Parameters | |
373 | ---------- |
|
373 | ---------- | |
374 |
|
374 | |||
375 | header : str |
|
375 | header : str | |
376 | Optional header string to print at startup. |
|
376 | Optional header string to print at startup. | |
377 | compile_flags |
|
377 | compile_flags | |
378 | Passed to the `compile_flags` parameter of :py:meth:`InteractiveShellEmbed.mainloop()`, |
|
378 | Passed to the `compile_flags` parameter of :py:meth:`InteractiveShellEmbed.mainloop()`, | |
379 | which is called when the :class:`InteractiveShellEmbed` instance is called. |
|
379 | which is called when the :class:`InteractiveShellEmbed` instance is called. | |
380 | **kwargs : various, optional |
|
380 | **kwargs : various, optional | |
381 | Any other kwargs will be passed to the :class:`InteractiveShellEmbed` constructor. |
|
381 | Any other kwargs will be passed to the :class:`InteractiveShellEmbed` constructor. | |
382 | Full customization can be done by passing a traitlets :class:`Config` in as the |
|
382 | Full customization can be done by passing a traitlets :class:`Config` in as the | |
383 | `config` argument (see :ref:`configure_start_ipython`). |
|
383 | `config` argument (see :ref:`configure_start_ipython` and :ref:`terminal_options`). | |
384 | """ |
|
384 | """ | |
385 | config = kwargs.get('config') |
|
385 | config = kwargs.get('config') | |
386 | if config is None: |
|
386 | if config is None: | |
387 | config = load_default_config() |
|
387 | config = load_default_config() | |
388 | config.InteractiveShellEmbed = config.TerminalInteractiveShell |
|
388 | config.InteractiveShellEmbed = config.TerminalInteractiveShell | |
389 | kwargs['config'] = config |
|
389 | kwargs['config'] = config | |
390 | using = kwargs.get('using', 'sync') |
|
390 | using = kwargs.get('using', 'sync') | |
391 | if using : |
|
391 | if using : | |
392 | kwargs['config'].update({'TerminalInteractiveShell':{'loop_runner':using, 'colors':'NoColor', 'autoawait': using!='sync'}}) |
|
392 | kwargs['config'].update({'TerminalInteractiveShell':{'loop_runner':using, 'colors':'NoColor', 'autoawait': using!='sync'}}) | |
393 | #save ps1/ps2 if defined |
|
393 | #save ps1/ps2 if defined | |
394 | ps1 = None |
|
394 | ps1 = None | |
395 | ps2 = None |
|
395 | ps2 = None | |
396 | try: |
|
396 | try: | |
397 | ps1 = sys.ps1 |
|
397 | ps1 = sys.ps1 | |
398 | ps2 = sys.ps2 |
|
398 | ps2 = sys.ps2 | |
399 | except AttributeError: |
|
399 | except AttributeError: | |
400 | pass |
|
400 | pass | |
401 | #save previous instance |
|
401 | #save previous instance | |
402 | saved_shell_instance = InteractiveShell._instance |
|
402 | saved_shell_instance = InteractiveShell._instance | |
403 | if saved_shell_instance is not None: |
|
403 | if saved_shell_instance is not None: | |
404 | cls = type(saved_shell_instance) |
|
404 | cls = type(saved_shell_instance) | |
405 | cls.clear_instance() |
|
405 | cls.clear_instance() | |
406 | frame = sys._getframe(1) |
|
406 | frame = sys._getframe(1) | |
407 | shell = InteractiveShellEmbed.instance(_init_location_id='%s:%s' % ( |
|
407 | shell = InteractiveShellEmbed.instance(_init_location_id='%s:%s' % ( | |
408 | frame.f_code.co_filename, frame.f_lineno), **kwargs) |
|
408 | frame.f_code.co_filename, frame.f_lineno), **kwargs) | |
409 | shell(header=header, stack_depth=2, compile_flags=compile_flags, |
|
409 | shell(header=header, stack_depth=2, compile_flags=compile_flags, | |
410 | _call_location_id='%s:%s' % (frame.f_code.co_filename, frame.f_lineno)) |
|
410 | _call_location_id='%s:%s' % (frame.f_code.co_filename, frame.f_lineno)) | |
411 | InteractiveShellEmbed.clear_instance() |
|
411 | InteractiveShellEmbed.clear_instance() | |
412 | #restore previous instance |
|
412 | #restore previous instance | |
413 | if saved_shell_instance is not None: |
|
413 | if saved_shell_instance is not None: | |
414 | cls = type(saved_shell_instance) |
|
414 | cls = type(saved_shell_instance) | |
415 | cls.clear_instance() |
|
415 | cls.clear_instance() | |
416 | for subclass in cls._walk_mro(): |
|
416 | for subclass in cls._walk_mro(): | |
417 | subclass._instance = saved_shell_instance |
|
417 | subclass._instance = saved_shell_instance | |
418 | if ps1 is not None: |
|
418 | if ps1 is not None: | |
419 | sys.ps1 = ps1 |
|
419 | sys.ps1 = ps1 | |
420 | sys.ps2 = ps2 |
|
420 | sys.ps2 = ps2 |
@@ -1,126 +1,127 b'' | |||||
1 | #!/usr/bin/env python |
|
1 | #!/usr/bin/env python | |
2 |
|
2 | |||
3 | import inspect |
|
3 | import inspect | |
4 | from pathlib import Path |
|
4 | from pathlib import Path | |
5 | from IPython.terminal.ipapp import TerminalIPythonApp |
|
5 | from IPython.terminal.ipapp import TerminalIPythonApp | |
6 | from ipykernel.kernelapp import IPKernelApp |
|
6 | from ipykernel.kernelapp import IPKernelApp | |
7 | from traitlets import Undefined |
|
7 | from traitlets import Undefined | |
8 | from collections import defaultdict |
|
8 | from collections import defaultdict | |
9 |
|
9 | |||
10 | here = (Path(__file__)).parent |
|
10 | here = (Path(__file__)).parent | |
11 | options = here / "source" / "config" / "options" |
|
11 | options = here / "source" / "config" / "options" | |
12 | generated = options / "config-generated.txt" |
|
12 | generated = options / "config-generated.txt" | |
13 |
|
13 | |||
14 | import textwrap |
|
14 | import textwrap | |
15 | indent = lambda text,n: textwrap.indent(text,n*' ') |
|
15 | indent = lambda text,n: textwrap.indent(text,n*' ') | |
16 |
|
16 | |||
17 |
|
17 | |||
18 | def interesting_default_value(dv): |
|
18 | def interesting_default_value(dv): | |
19 | if (dv is None) or (dv is Undefined): |
|
19 | if (dv is None) or (dv is Undefined): | |
20 | return False |
|
20 | return False | |
21 | if isinstance(dv, (str, list, tuple, dict, set)): |
|
21 | if isinstance(dv, (str, list, tuple, dict, set)): | |
22 | return bool(dv) |
|
22 | return bool(dv) | |
23 | return True |
|
23 | return True | |
24 |
|
24 | |||
25 | def format_aliases(aliases): |
|
25 | def format_aliases(aliases): | |
26 | fmted = [] |
|
26 | fmted = [] | |
27 | for a in aliases: |
|
27 | for a in aliases: | |
28 | dashes = '-' if len(a) == 1 else '--' |
|
28 | dashes = '-' if len(a) == 1 else '--' | |
29 | fmted.append('``%s%s``' % (dashes, a)) |
|
29 | fmted.append('``%s%s``' % (dashes, a)) | |
30 | return ', '.join(fmted) |
|
30 | return ', '.join(fmted) | |
31 |
|
31 | |||
32 | def class_config_rst_doc(cls, trait_aliases): |
|
32 | def class_config_rst_doc(cls, trait_aliases): | |
33 | """Generate rST documentation for this class' config options. |
|
33 | """Generate rST documentation for this class' config options. | |
34 |
|
34 | |||
35 | Excludes traits defined on parent classes. |
|
35 | Excludes traits defined on parent classes. | |
36 | """ |
|
36 | """ | |
37 | lines = [] |
|
37 | lines = [] | |
38 | classname = cls.__name__ |
|
38 | classname = cls.__name__ | |
39 | for k, trait in sorted(cls.class_traits(config=True).items()): |
|
39 | for k, trait in sorted(cls.class_traits(config=True).items()): | |
40 | ttype = trait.__class__.__name__ |
|
40 | ttype = trait.__class__.__name__ | |
41 |
|
41 | |||
42 | fullname = classname + '.' + trait.name |
|
42 | fullname = classname + '.' + trait.name | |
43 | lines += ['.. configtrait:: ' + fullname, |
|
43 | lines += ['.. configtrait:: ' + fullname, | |
44 | '' |
|
44 | '' | |
45 | ] |
|
45 | ] | |
46 |
|
46 | |||
47 | help = trait.help.rstrip() or 'No description' |
|
47 | help = trait.help.rstrip() or 'No description' | |
48 | lines.append(indent(inspect.cleandoc(help), 4) + '\n') |
|
48 | lines.append(indent(inspect.cleandoc(help), 4) + '\n') | |
49 |
|
49 | |||
50 | # Choices or type |
|
50 | # Choices or type | |
51 | if 'Enum' in ttype: |
|
51 | if 'Enum' in ttype: | |
52 | # include Enum choices |
|
52 | # include Enum choices | |
53 | lines.append(indent( |
|
53 | lines.append(indent( | |
54 | ':options: ' + ', '.join('``%r``' % x for x in trait.values), 4)) |
|
54 | ':options: ' + ', '.join('``%r``' % x for x in trait.values), 4)) | |
55 | else: |
|
55 | else: | |
56 | lines.append(indent(':trait type: ' + ttype, 4)) |
|
56 | lines.append(indent(':trait type: ' + ttype, 4)) | |
57 |
|
57 | |||
58 | # Default value |
|
58 | # Default value | |
59 | # Ignore boring default values like None, [] or '' |
|
59 | # Ignore boring default values like None, [] or '' | |
60 | if interesting_default_value(trait.default_value): |
|
60 | if interesting_default_value(trait.default_value): | |
61 | try: |
|
61 | try: | |
62 | dvr = trait.default_value_repr() |
|
62 | dvr = trait.default_value_repr() | |
63 | except Exception: |
|
63 | except Exception: | |
64 | dvr = None # ignore defaults we can't construct |
|
64 | dvr = None # ignore defaults we can't construct | |
65 | if dvr is not None: |
|
65 | if dvr is not None: | |
66 | if len(dvr) > 64: |
|
66 | if len(dvr) > 64: | |
67 | dvr = dvr[:61] + '...' |
|
67 | dvr = dvr[:61] + '...' | |
68 | # Double up backslashes, so they get to the rendered docs |
|
68 | # Double up backslashes, so they get to the rendered docs | |
69 | dvr = dvr.replace('\\n', '\\\\n') |
|
69 | dvr = dvr.replace('\\n', '\\\\n') | |
70 | lines.append(indent(':default: ``%s``' % dvr, 4)) |
|
70 | lines.append(indent(':default: ``%s``' % dvr, 4)) | |
71 |
|
71 | |||
72 | # Command line aliases |
|
72 | # Command line aliases | |
73 | if trait_aliases[fullname]: |
|
73 | if trait_aliases[fullname]: | |
74 | fmt_aliases = format_aliases(trait_aliases[fullname]) |
|
74 | fmt_aliases = format_aliases(trait_aliases[fullname]) | |
75 | lines.append(indent(':CLI option: ' + fmt_aliases, 4)) |
|
75 | lines.append(indent(':CLI option: ' + fmt_aliases, 4)) | |
76 |
|
76 | |||
77 | # Blank line |
|
77 | # Blank line | |
78 | lines.append('') |
|
78 | lines.append('') | |
79 |
|
79 | |||
80 | return '\n'.join(lines) |
|
80 | return '\n'.join(lines) | |
81 |
|
81 | |||
82 | def reverse_aliases(app): |
|
82 | def reverse_aliases(app): | |
83 | """Produce a mapping of trait names to lists of command line aliases. |
|
83 | """Produce a mapping of trait names to lists of command line aliases. | |
84 | """ |
|
84 | """ | |
85 | res = defaultdict(list) |
|
85 | res = defaultdict(list) | |
86 | for alias, trait in app.aliases.items(): |
|
86 | for alias, trait in app.aliases.items(): | |
87 | res[trait].append(alias) |
|
87 | res[trait].append(alias) | |
88 |
|
88 | |||
89 | # Flags also often act as aliases for a boolean trait. |
|
89 | # Flags also often act as aliases for a boolean trait. | |
90 | # Treat flags which set one trait to True as aliases. |
|
90 | # Treat flags which set one trait to True as aliases. | |
91 | for flag, (cfg, _) in app.flags.items(): |
|
91 | for flag, (cfg, _) in app.flags.items(): | |
92 | if len(cfg) == 1: |
|
92 | if len(cfg) == 1: | |
93 | classname = list(cfg)[0] |
|
93 | classname = list(cfg)[0] | |
94 | cls_cfg = cfg[classname] |
|
94 | cls_cfg = cfg[classname] | |
95 | if len(cls_cfg) == 1: |
|
95 | if len(cls_cfg) == 1: | |
96 | traitname = list(cls_cfg)[0] |
|
96 | traitname = list(cls_cfg)[0] | |
97 | if cls_cfg[traitname] is True: |
|
97 | if cls_cfg[traitname] is True: | |
98 | res[classname+'.'+traitname].append(flag) |
|
98 | res[classname+'.'+traitname].append(flag) | |
99 |
|
99 | |||
100 | return res |
|
100 | return res | |
101 |
|
101 | |||
102 | def write_doc(name, title, app, preamble=None): |
|
102 | def write_doc(name, title, app, preamble=None): | |
103 | trait_aliases = reverse_aliases(app) |
|
103 | trait_aliases = reverse_aliases(app) | |
104 | filename = options / (name + ".rst") |
|
104 | filename = options / (name + ".rst") | |
105 | with open(filename, "w", encoding="utf-8") as f: |
|
105 | with open(filename, "w", encoding="utf-8") as f: | |
|
106 | f.write(".. _" + name + "_options:" + "\n\n") | |||
106 | f.write(title + "\n") |
|
107 | f.write(title + "\n") | |
107 | f.write(("=" * len(title)) + "\n") |
|
108 | f.write(("=" * len(title)) + "\n") | |
108 | f.write("\n") |
|
109 | f.write("\n") | |
109 | if preamble is not None: |
|
110 | if preamble is not None: | |
110 | f.write(preamble + '\n\n') |
|
111 | f.write(preamble + '\n\n') | |
111 | #f.write(app.document_config_options()) |
|
112 | #f.write(app.document_config_options()) | |
112 |
|
113 | |||
113 | for c in app._classes_inc_parents(): |
|
114 | for c in app._classes_inc_parents(): | |
114 | f.write(class_config_rst_doc(c, trait_aliases)) |
|
115 | f.write(class_config_rst_doc(c, trait_aliases)) | |
115 | f.write('\n') |
|
116 | f.write('\n') | |
116 |
|
117 | |||
117 |
|
118 | |||
118 | if __name__ == '__main__': |
|
119 | if __name__ == '__main__': | |
119 | # Touch this file for the make target |
|
120 | # Touch this file for the make target | |
120 | Path(generated).write_text("", encoding="utf-8") |
|
121 | Path(generated).write_text("", encoding="utf-8") | |
121 |
|
122 | |||
122 | write_doc('terminal', 'Terminal IPython options', TerminalIPythonApp()) |
|
123 | write_doc('terminal', 'Terminal IPython options', TerminalIPythonApp()) | |
123 | write_doc('kernel', 'IPython kernel options', IPKernelApp(), |
|
124 | write_doc('kernel', 'IPython kernel options', IPKernelApp(), | |
124 | preamble=("These options can be used in :file:`ipython_kernel_config.py`. " |
|
125 | preamble=("These options can be used in :file:`ipython_kernel_config.py`. " | |
125 | "The kernel also respects any options in `ipython_config.py`"), |
|
126 | "The kernel also respects any options in `ipython_config.py`"), | |
126 | ) |
|
127 | ) |
General Comments 0
You need to be logged in to leave comments.
Login now