##// END OF EJS Templates
respect `pylab_import_all` when `--pylab` specified at the command-line...
MinRK -
Show More
@@ -1,406 +1,406 b''
1 # encoding: utf-8
1 # encoding: utf-8
2 """
2 """
3 A mixin for :class:`~IPython.core.application.Application` classes that
3 A mixin for :class:`~IPython.core.application.Application` classes that
4 launch InteractiveShell instances, load extensions, etc.
4 launch InteractiveShell instances, load extensions, etc.
5
5
6 Authors
6 Authors
7 -------
7 -------
8
8
9 * Min Ragan-Kelley
9 * Min Ragan-Kelley
10 """
10 """
11
11
12 #-----------------------------------------------------------------------------
12 #-----------------------------------------------------------------------------
13 # Copyright (C) 2008-2011 The IPython Development Team
13 # Copyright (C) 2008-2011 The IPython Development Team
14 #
14 #
15 # Distributed under the terms of the BSD License. The full license is in
15 # Distributed under the terms of the BSD License. The full license is in
16 # the file COPYING, distributed as part of this software.
16 # the file COPYING, distributed as part of this software.
17 #-----------------------------------------------------------------------------
17 #-----------------------------------------------------------------------------
18
18
19 #-----------------------------------------------------------------------------
19 #-----------------------------------------------------------------------------
20 # Imports
20 # Imports
21 #-----------------------------------------------------------------------------
21 #-----------------------------------------------------------------------------
22
22
23 from __future__ import absolute_import
23 from __future__ import absolute_import
24
24
25 import glob
25 import glob
26 import os
26 import os
27 import sys
27 import sys
28
28
29 from IPython.config.application import boolean_flag
29 from IPython.config.application import boolean_flag
30 from IPython.config.configurable import Configurable
30 from IPython.config.configurable import Configurable
31 from IPython.config.loader import Config
31 from IPython.config.loader import Config
32 from IPython.core import pylabtools
32 from IPython.core import pylabtools
33 from IPython.utils import py3compat
33 from IPython.utils import py3compat
34 from IPython.utils.contexts import preserve_keys
34 from IPython.utils.contexts import preserve_keys
35 from IPython.utils.path import filefind
35 from IPython.utils.path import filefind
36 from IPython.utils.traitlets import (
36 from IPython.utils.traitlets import (
37 Unicode, Instance, List, Bool, CaselessStrEnum, Dict
37 Unicode, Instance, List, Bool, CaselessStrEnum, Dict
38 )
38 )
39
39
40 #-----------------------------------------------------------------------------
40 #-----------------------------------------------------------------------------
41 # Aliases and Flags
41 # Aliases and Flags
42 #-----------------------------------------------------------------------------
42 #-----------------------------------------------------------------------------
43
43
44 backend_keys = sorted(pylabtools.backends.keys())
44 backend_keys = sorted(pylabtools.backends.keys())
45 backend_keys.insert(0, 'auto')
45 backend_keys.insert(0, 'auto')
46
46
47 shell_flags = {}
47 shell_flags = {}
48
48
49 addflag = lambda *args: shell_flags.update(boolean_flag(*args))
49 addflag = lambda *args: shell_flags.update(boolean_flag(*args))
50 addflag('autoindent', 'InteractiveShell.autoindent',
50 addflag('autoindent', 'InteractiveShell.autoindent',
51 'Turn on autoindenting.', 'Turn off autoindenting.'
51 'Turn on autoindenting.', 'Turn off autoindenting.'
52 )
52 )
53 addflag('automagic', 'InteractiveShell.automagic',
53 addflag('automagic', 'InteractiveShell.automagic',
54 """Turn on the auto calling of magic commands. Type %%magic at the
54 """Turn on the auto calling of magic commands. Type %%magic at the
55 IPython prompt for more information.""",
55 IPython prompt for more information.""",
56 'Turn off the auto calling of magic commands.'
56 'Turn off the auto calling of magic commands.'
57 )
57 )
58 addflag('pdb', 'InteractiveShell.pdb',
58 addflag('pdb', 'InteractiveShell.pdb',
59 "Enable auto calling the pdb debugger after every exception.",
59 "Enable auto calling the pdb debugger after every exception.",
60 "Disable auto calling the pdb debugger after every exception."
60 "Disable auto calling the pdb debugger after every exception."
61 )
61 )
62 # pydb flag doesn't do any config, as core.debugger switches on import,
62 # pydb flag doesn't do any config, as core.debugger switches on import,
63 # which is before parsing. This just allows the flag to be passed.
63 # which is before parsing. This just allows the flag to be passed.
64 shell_flags.update(dict(
64 shell_flags.update(dict(
65 pydb = ({},
65 pydb = ({},
66 """Use the third party 'pydb' package as debugger, instead of pdb.
66 """Use the third party 'pydb' package as debugger, instead of pdb.
67 Requires that pydb is installed."""
67 Requires that pydb is installed."""
68 )
68 )
69 ))
69 ))
70 addflag('pprint', 'PlainTextFormatter.pprint',
70 addflag('pprint', 'PlainTextFormatter.pprint',
71 "Enable auto pretty printing of results.",
71 "Enable auto pretty printing of results.",
72 "Disable auto pretty printing of results."
72 "Disable auto pretty printing of results."
73 )
73 )
74 addflag('color-info', 'InteractiveShell.color_info',
74 addflag('color-info', 'InteractiveShell.color_info',
75 """IPython can display information about objects via a set of func-
75 """IPython can display information about objects via a set of func-
76 tions, and optionally can use colors for this, syntax highlighting
76 tions, and optionally can use colors for this, syntax highlighting
77 source code and various other elements. However, because this
77 source code and various other elements. However, because this
78 information is passed through a pager (like 'less') and many pagers get
78 information is passed through a pager (like 'less') and many pagers get
79 confused with color codes, this option is off by default. You can test
79 confused with color codes, this option is off by default. You can test
80 it and turn it on permanently in your ipython_config.py file if it
80 it and turn it on permanently in your ipython_config.py file if it
81 works for you. Test it and turn it on permanently if it works with
81 works for you. Test it and turn it on permanently if it works with
82 your system. The magic function %%color_info allows you to toggle this
82 your system. The magic function %%color_info allows you to toggle this
83 interactively for testing.""",
83 interactively for testing.""",
84 "Disable using colors for info related things."
84 "Disable using colors for info related things."
85 )
85 )
86 addflag('deep-reload', 'InteractiveShell.deep_reload',
86 addflag('deep-reload', 'InteractiveShell.deep_reload',
87 """Enable deep (recursive) reloading by default. IPython can use the
87 """Enable deep (recursive) reloading by default. IPython can use the
88 deep_reload module which reloads changes in modules recursively (it
88 deep_reload module which reloads changes in modules recursively (it
89 replaces the reload() function, so you don't need to change anything to
89 replaces the reload() function, so you don't need to change anything to
90 use it). deep_reload() forces a full reload of modules whose code may
90 use it). deep_reload() forces a full reload of modules whose code may
91 have changed, which the default reload() function does not. When
91 have changed, which the default reload() function does not. When
92 deep_reload is off, IPython will use the normal reload(), but
92 deep_reload is off, IPython will use the normal reload(), but
93 deep_reload will still be available as dreload(). This feature is off
93 deep_reload will still be available as dreload(). This feature is off
94 by default [which means that you have both normal reload() and
94 by default [which means that you have both normal reload() and
95 dreload()].""",
95 dreload()].""",
96 "Disable deep (recursive) reloading by default."
96 "Disable deep (recursive) reloading by default."
97 )
97 )
98 nosep_config = Config()
98 nosep_config = Config()
99 nosep_config.InteractiveShell.separate_in = ''
99 nosep_config.InteractiveShell.separate_in = ''
100 nosep_config.InteractiveShell.separate_out = ''
100 nosep_config.InteractiveShell.separate_out = ''
101 nosep_config.InteractiveShell.separate_out2 = ''
101 nosep_config.InteractiveShell.separate_out2 = ''
102
102
103 shell_flags['nosep']=(nosep_config, "Eliminate all spacing between prompts.")
103 shell_flags['nosep']=(nosep_config, "Eliminate all spacing between prompts.")
104 shell_flags['pylab'] = (
104 shell_flags['pylab'] = (
105 {'InteractiveShellApp' : {'pylab' : 'auto'}},
105 {'InteractiveShellApp' : {'pylab' : 'auto'}},
106 """Pre-load matplotlib and numpy for interactive use with
106 """Pre-load matplotlib and numpy for interactive use with
107 the default matplotlib backend."""
107 the default matplotlib backend."""
108 )
108 )
109 shell_flags['matplotlib'] = (
109 shell_flags['matplotlib'] = (
110 {'InteractiveShellApp' : {'matplotlib' : 'auto'}},
110 {'InteractiveShellApp' : {'matplotlib' : 'auto'}},
111 """Configure matplotlib for interactive use with
111 """Configure matplotlib for interactive use with
112 the default matplotlib backend."""
112 the default matplotlib backend."""
113 )
113 )
114
114
115 # it's possible we don't want short aliases for *all* of these:
115 # it's possible we don't want short aliases for *all* of these:
116 shell_aliases = dict(
116 shell_aliases = dict(
117 autocall='InteractiveShell.autocall',
117 autocall='InteractiveShell.autocall',
118 colors='InteractiveShell.colors',
118 colors='InteractiveShell.colors',
119 logfile='InteractiveShell.logfile',
119 logfile='InteractiveShell.logfile',
120 logappend='InteractiveShell.logappend',
120 logappend='InteractiveShell.logappend',
121 c='InteractiveShellApp.code_to_run',
121 c='InteractiveShellApp.code_to_run',
122 m='InteractiveShellApp.module_to_run',
122 m='InteractiveShellApp.module_to_run',
123 ext='InteractiveShellApp.extra_extension',
123 ext='InteractiveShellApp.extra_extension',
124 gui='InteractiveShellApp.gui',
124 gui='InteractiveShellApp.gui',
125 pylab='InteractiveShellApp.pylab',
125 pylab='InteractiveShellApp.pylab',
126 matplotlib='InteractiveShellApp.matplotlib',
126 matplotlib='InteractiveShellApp.matplotlib',
127 )
127 )
128 shell_aliases['cache-size'] = 'InteractiveShell.cache_size'
128 shell_aliases['cache-size'] = 'InteractiveShell.cache_size'
129
129
130 #-----------------------------------------------------------------------------
130 #-----------------------------------------------------------------------------
131 # Main classes and functions
131 # Main classes and functions
132 #-----------------------------------------------------------------------------
132 #-----------------------------------------------------------------------------
133
133
134 class InteractiveShellApp(Configurable):
134 class InteractiveShellApp(Configurable):
135 """A Mixin for applications that start InteractiveShell instances.
135 """A Mixin for applications that start InteractiveShell instances.
136
136
137 Provides configurables for loading extensions and executing files
137 Provides configurables for loading extensions and executing files
138 as part of configuring a Shell environment.
138 as part of configuring a Shell environment.
139
139
140 The following methods should be called by the :meth:`initialize` method
140 The following methods should be called by the :meth:`initialize` method
141 of the subclass:
141 of the subclass:
142
142
143 - :meth:`init_path`
143 - :meth:`init_path`
144 - :meth:`init_shell` (to be implemented by the subclass)
144 - :meth:`init_shell` (to be implemented by the subclass)
145 - :meth:`init_gui_pylab`
145 - :meth:`init_gui_pylab`
146 - :meth:`init_extensions`
146 - :meth:`init_extensions`
147 - :meth:`init_code`
147 - :meth:`init_code`
148 """
148 """
149 extensions = List(Unicode, config=True,
149 extensions = List(Unicode, config=True,
150 help="A list of dotted module names of IPython extensions to load."
150 help="A list of dotted module names of IPython extensions to load."
151 )
151 )
152 extra_extension = Unicode('', config=True,
152 extra_extension = Unicode('', config=True,
153 help="dotted module name of an IPython extension to load."
153 help="dotted module name of an IPython extension to load."
154 )
154 )
155 def _extra_extension_changed(self, name, old, new):
155 def _extra_extension_changed(self, name, old, new):
156 if new:
156 if new:
157 # add to self.extensions
157 # add to self.extensions
158 self.extensions.append(new)
158 self.extensions.append(new)
159
159
160 # Extensions that are always loaded (not configurable)
160 # Extensions that are always loaded (not configurable)
161 default_extensions = List(Unicode, [u'storemagic'], config=False)
161 default_extensions = List(Unicode, [u'storemagic'], config=False)
162
162
163 exec_files = List(Unicode, config=True,
163 exec_files = List(Unicode, config=True,
164 help="""List of files to run at IPython startup."""
164 help="""List of files to run at IPython startup."""
165 )
165 )
166 file_to_run = Unicode('', config=True,
166 file_to_run = Unicode('', config=True,
167 help="""A file to be run""")
167 help="""A file to be run""")
168
168
169 exec_lines = List(Unicode, config=True,
169 exec_lines = List(Unicode, config=True,
170 help="""lines of code to run at IPython startup."""
170 help="""lines of code to run at IPython startup."""
171 )
171 )
172 code_to_run = Unicode('', config=True,
172 code_to_run = Unicode('', config=True,
173 help="Execute the given command string."
173 help="Execute the given command string."
174 )
174 )
175 module_to_run = Unicode('', config=True,
175 module_to_run = Unicode('', config=True,
176 help="Run the module as a script."
176 help="Run the module as a script."
177 )
177 )
178 gui = CaselessStrEnum(('qt', 'wx', 'gtk', 'glut', 'pyglet', 'osx'), config=True,
178 gui = CaselessStrEnum(('qt', 'wx', 'gtk', 'glut', 'pyglet', 'osx'), config=True,
179 help="Enable GUI event loop integration ('qt', 'wx', 'gtk', 'glut', 'pyglet', 'osx')."
179 help="Enable GUI event loop integration ('qt', 'wx', 'gtk', 'glut', 'pyglet', 'osx')."
180 )
180 )
181 matplotlib = CaselessStrEnum(backend_keys,
181 matplotlib = CaselessStrEnum(backend_keys,
182 config=True,
182 config=True,
183 help="""Configure matplotlib for interactive use with
183 help="""Configure matplotlib for interactive use with
184 the default matplotlib backend."""
184 the default matplotlib backend."""
185 )
185 )
186 pylab = CaselessStrEnum(backend_keys,
186 pylab = CaselessStrEnum(backend_keys,
187 config=True,
187 config=True,
188 help="""Pre-load matplotlib and numpy for interactive use,
188 help="""Pre-load matplotlib and numpy for interactive use,
189 selecting a particular matplotlib backend and loop integration.
189 selecting a particular matplotlib backend and loop integration.
190 """
190 """
191 )
191 )
192 pylab_import_all = Bool(True, config=True,
192 pylab_import_all = Bool(True, config=True,
193 help="""If true, IPython will populate the user namespace with numpy, pylab, etc.
193 help="""If true, IPython will populate the user namespace with numpy, pylab, etc.
194 and an 'import *' is done from numpy and pylab, when using pylab mode.
194 and an 'import *' is done from numpy and pylab, when using pylab mode.
195
195
196 When False, pylab mode should not import any names into the user namespace.
196 When False, pylab mode should not import any names into the user namespace.
197 """
197 """
198 )
198 )
199 shell = Instance('IPython.core.interactiveshell.InteractiveShellABC')
199 shell = Instance('IPython.core.interactiveshell.InteractiveShellABC')
200
200
201 user_ns = Dict(default_value=None)
201 user_ns = Dict(default_value=None)
202 def _user_ns_changed(self, name, old, new):
202 def _user_ns_changed(self, name, old, new):
203 if self.shell is not None:
203 if self.shell is not None:
204 self.shell.user_ns = new
204 self.shell.user_ns = new
205 self.shell.init_user_ns()
205 self.shell.init_user_ns()
206
206
207 def init_path(self):
207 def init_path(self):
208 """Add current working directory, '', to sys.path"""
208 """Add current working directory, '', to sys.path"""
209 if sys.path[0] != '':
209 if sys.path[0] != '':
210 sys.path.insert(0, '')
210 sys.path.insert(0, '')
211
211
212 def init_shell(self):
212 def init_shell(self):
213 raise NotImplementedError("Override in subclasses")
213 raise NotImplementedError("Override in subclasses")
214
214
215 def init_gui_pylab(self):
215 def init_gui_pylab(self):
216 """Enable GUI event loop integration, taking pylab into account."""
216 """Enable GUI event loop integration, taking pylab into account."""
217 enable = False
217 enable = False
218 shell = self.shell
218 shell = self.shell
219 if self.pylab:
219 if self.pylab:
220 enable = shell.enable_pylab
220 enable = lambda key: shell.enable_pylab(key, import_all=self.pylab_import_all)
221 key = self.pylab
221 key = self.pylab
222 elif self.matplotlib:
222 elif self.matplotlib:
223 enable = shell.enable_matplotlib
223 enable = shell.enable_matplotlib
224 key = self.matplotlib
224 key = self.matplotlib
225 elif self.gui:
225 elif self.gui:
226 enable = shell.enable_gui
226 enable = shell.enable_gui
227 key = self.gui
227 key = self.gui
228
228
229 if not enable:
229 if not enable:
230 return
230 return
231
231
232 try:
232 try:
233 r = enable(key)
233 r = enable(key)
234 except ImportError:
234 except ImportError:
235 self.log.warn("Eventloop or matplotlib integration failed. Is matplotlib installed?")
235 self.log.warn("Eventloop or matplotlib integration failed. Is matplotlib installed?")
236 self.shell.showtraceback()
236 self.shell.showtraceback()
237 return
237 return
238 except Exception:
238 except Exception:
239 self.log.warn("GUI event loop or pylab initialization failed")
239 self.log.warn("GUI event loop or pylab initialization failed")
240 self.shell.showtraceback()
240 self.shell.showtraceback()
241 return
241 return
242
242
243 if isinstance(r, tuple):
243 if isinstance(r, tuple):
244 gui, backend = r[:2]
244 gui, backend = r[:2]
245 self.log.info("Enabling GUI event loop integration, "
245 self.log.info("Enabling GUI event loop integration, "
246 "eventloop=%s, matplotlib=%s", gui, backend)
246 "eventloop=%s, matplotlib=%s", gui, backend)
247 if key == "auto":
247 if key == "auto":
248 print ("Using matplotlib backend: %s" % backend)
248 print ("Using matplotlib backend: %s" % backend)
249 else:
249 else:
250 gui = r
250 gui = r
251 self.log.info("Enabling GUI event loop integration, "
251 self.log.info("Enabling GUI event loop integration, "
252 "eventloop=%s", gui)
252 "eventloop=%s", gui)
253
253
254 def init_extensions(self):
254 def init_extensions(self):
255 """Load all IPython extensions in IPythonApp.extensions.
255 """Load all IPython extensions in IPythonApp.extensions.
256
256
257 This uses the :meth:`ExtensionManager.load_extensions` to load all
257 This uses the :meth:`ExtensionManager.load_extensions` to load all
258 the extensions listed in ``self.extensions``.
258 the extensions listed in ``self.extensions``.
259 """
259 """
260 try:
260 try:
261 self.log.debug("Loading IPython extensions...")
261 self.log.debug("Loading IPython extensions...")
262 extensions = self.default_extensions + self.extensions
262 extensions = self.default_extensions + self.extensions
263 for ext in extensions:
263 for ext in extensions:
264 try:
264 try:
265 self.log.info("Loading IPython extension: %s" % ext)
265 self.log.info("Loading IPython extension: %s" % ext)
266 self.shell.extension_manager.load_extension(ext)
266 self.shell.extension_manager.load_extension(ext)
267 except:
267 except:
268 self.log.warn("Error in loading extension: %s" % ext +
268 self.log.warn("Error in loading extension: %s" % ext +
269 "\nCheck your config files in %s" % self.profile_dir.location
269 "\nCheck your config files in %s" % self.profile_dir.location
270 )
270 )
271 self.shell.showtraceback()
271 self.shell.showtraceback()
272 except:
272 except:
273 self.log.warn("Unknown error in loading extensions:")
273 self.log.warn("Unknown error in loading extensions:")
274 self.shell.showtraceback()
274 self.shell.showtraceback()
275
275
276 def init_code(self):
276 def init_code(self):
277 """run the pre-flight code, specified via exec_lines"""
277 """run the pre-flight code, specified via exec_lines"""
278 self._run_startup_files()
278 self._run_startup_files()
279 self._run_exec_lines()
279 self._run_exec_lines()
280 self._run_exec_files()
280 self._run_exec_files()
281 self._run_cmd_line_code()
281 self._run_cmd_line_code()
282 self._run_module()
282 self._run_module()
283
283
284 # flush output, so itwon't be attached to the first cell
284 # flush output, so itwon't be attached to the first cell
285 sys.stdout.flush()
285 sys.stdout.flush()
286 sys.stderr.flush()
286 sys.stderr.flush()
287
287
288 # Hide variables defined here from %who etc.
288 # Hide variables defined here from %who etc.
289 self.shell.user_ns_hidden.update(self.shell.user_ns)
289 self.shell.user_ns_hidden.update(self.shell.user_ns)
290
290
291 def _run_exec_lines(self):
291 def _run_exec_lines(self):
292 """Run lines of code in IPythonApp.exec_lines in the user's namespace."""
292 """Run lines of code in IPythonApp.exec_lines in the user's namespace."""
293 if not self.exec_lines:
293 if not self.exec_lines:
294 return
294 return
295 try:
295 try:
296 self.log.debug("Running code from IPythonApp.exec_lines...")
296 self.log.debug("Running code from IPythonApp.exec_lines...")
297 for line in self.exec_lines:
297 for line in self.exec_lines:
298 try:
298 try:
299 self.log.info("Running code in user namespace: %s" %
299 self.log.info("Running code in user namespace: %s" %
300 line)
300 line)
301 self.shell.run_cell(line, store_history=False)
301 self.shell.run_cell(line, store_history=False)
302 except:
302 except:
303 self.log.warn("Error in executing line in user "
303 self.log.warn("Error in executing line in user "
304 "namespace: %s" % line)
304 "namespace: %s" % line)
305 self.shell.showtraceback()
305 self.shell.showtraceback()
306 except:
306 except:
307 self.log.warn("Unknown error in handling IPythonApp.exec_lines:")
307 self.log.warn("Unknown error in handling IPythonApp.exec_lines:")
308 self.shell.showtraceback()
308 self.shell.showtraceback()
309
309
310 def _exec_file(self, fname):
310 def _exec_file(self, fname):
311 try:
311 try:
312 full_filename = filefind(fname, [u'.', self.ipython_dir])
312 full_filename = filefind(fname, [u'.', self.ipython_dir])
313 except IOError as e:
313 except IOError as e:
314 self.log.warn("File not found: %r"%fname)
314 self.log.warn("File not found: %r"%fname)
315 return
315 return
316 # Make sure that the running script gets a proper sys.argv as if it
316 # Make sure that the running script gets a proper sys.argv as if it
317 # were run from a system shell.
317 # were run from a system shell.
318 save_argv = sys.argv
318 save_argv = sys.argv
319 sys.argv = [full_filename] + self.extra_args[1:]
319 sys.argv = [full_filename] + self.extra_args[1:]
320 # protect sys.argv from potential unicode strings on Python 2:
320 # protect sys.argv from potential unicode strings on Python 2:
321 if not py3compat.PY3:
321 if not py3compat.PY3:
322 sys.argv = [ py3compat.cast_bytes(a) for a in sys.argv ]
322 sys.argv = [ py3compat.cast_bytes(a) for a in sys.argv ]
323 try:
323 try:
324 if os.path.isfile(full_filename):
324 if os.path.isfile(full_filename):
325 self.log.info("Running file in user namespace: %s" %
325 self.log.info("Running file in user namespace: %s" %
326 full_filename)
326 full_filename)
327 # Ensure that __file__ is always defined to match Python
327 # Ensure that __file__ is always defined to match Python
328 # behavior.
328 # behavior.
329 with preserve_keys(self.shell.user_ns, '__file__'):
329 with preserve_keys(self.shell.user_ns, '__file__'):
330 self.shell.user_ns['__file__'] = fname
330 self.shell.user_ns['__file__'] = fname
331 if full_filename.endswith('.ipy'):
331 if full_filename.endswith('.ipy'):
332 self.shell.safe_execfile_ipy(full_filename)
332 self.shell.safe_execfile_ipy(full_filename)
333 else:
333 else:
334 # default to python, even without extension
334 # default to python, even without extension
335 self.shell.safe_execfile(full_filename,
335 self.shell.safe_execfile(full_filename,
336 self.shell.user_ns)
336 self.shell.user_ns)
337 finally:
337 finally:
338 sys.argv = save_argv
338 sys.argv = save_argv
339
339
340 def _run_startup_files(self):
340 def _run_startup_files(self):
341 """Run files from profile startup directory"""
341 """Run files from profile startup directory"""
342 startup_dir = self.profile_dir.startup_dir
342 startup_dir = self.profile_dir.startup_dir
343 startup_files = []
343 startup_files = []
344 if os.environ.get('PYTHONSTARTUP', False):
344 if os.environ.get('PYTHONSTARTUP', False):
345 startup_files.append(os.environ['PYTHONSTARTUP'])
345 startup_files.append(os.environ['PYTHONSTARTUP'])
346 startup_files += glob.glob(os.path.join(startup_dir, '*.py'))
346 startup_files += glob.glob(os.path.join(startup_dir, '*.py'))
347 startup_files += glob.glob(os.path.join(startup_dir, '*.ipy'))
347 startup_files += glob.glob(os.path.join(startup_dir, '*.ipy'))
348 if not startup_files:
348 if not startup_files:
349 return
349 return
350
350
351 self.log.debug("Running startup files from %s...", startup_dir)
351 self.log.debug("Running startup files from %s...", startup_dir)
352 try:
352 try:
353 for fname in sorted(startup_files):
353 for fname in sorted(startup_files):
354 self._exec_file(fname)
354 self._exec_file(fname)
355 except:
355 except:
356 self.log.warn("Unknown error in handling startup files:")
356 self.log.warn("Unknown error in handling startup files:")
357 self.shell.showtraceback()
357 self.shell.showtraceback()
358
358
359 def _run_exec_files(self):
359 def _run_exec_files(self):
360 """Run files from IPythonApp.exec_files"""
360 """Run files from IPythonApp.exec_files"""
361 if not self.exec_files:
361 if not self.exec_files:
362 return
362 return
363
363
364 self.log.debug("Running files in IPythonApp.exec_files...")
364 self.log.debug("Running files in IPythonApp.exec_files...")
365 try:
365 try:
366 for fname in self.exec_files:
366 for fname in self.exec_files:
367 self._exec_file(fname)
367 self._exec_file(fname)
368 except:
368 except:
369 self.log.warn("Unknown error in handling IPythonApp.exec_files:")
369 self.log.warn("Unknown error in handling IPythonApp.exec_files:")
370 self.shell.showtraceback()
370 self.shell.showtraceback()
371
371
372 def _run_cmd_line_code(self):
372 def _run_cmd_line_code(self):
373 """Run code or file specified at the command-line"""
373 """Run code or file specified at the command-line"""
374 if self.code_to_run:
374 if self.code_to_run:
375 line = self.code_to_run
375 line = self.code_to_run
376 try:
376 try:
377 self.log.info("Running code given at command line (c=): %s" %
377 self.log.info("Running code given at command line (c=): %s" %
378 line)
378 line)
379 self.shell.run_cell(line, store_history=False)
379 self.shell.run_cell(line, store_history=False)
380 except:
380 except:
381 self.log.warn("Error in executing line in user namespace: %s" %
381 self.log.warn("Error in executing line in user namespace: %s" %
382 line)
382 line)
383 self.shell.showtraceback()
383 self.shell.showtraceback()
384
384
385 # Like Python itself, ignore the second if the first of these is present
385 # Like Python itself, ignore the second if the first of these is present
386 elif self.file_to_run:
386 elif self.file_to_run:
387 fname = self.file_to_run
387 fname = self.file_to_run
388 try:
388 try:
389 self._exec_file(fname)
389 self._exec_file(fname)
390 except:
390 except:
391 self.log.warn("Error in executing file in user namespace: %s" %
391 self.log.warn("Error in executing file in user namespace: %s" %
392 fname)
392 fname)
393 self.shell.showtraceback()
393 self.shell.showtraceback()
394
394
395 def _run_module(self):
395 def _run_module(self):
396 """Run module specified at the command-line."""
396 """Run module specified at the command-line."""
397 if self.module_to_run:
397 if self.module_to_run:
398 # Make sure that the module gets a proper sys.argv as if it were
398 # Make sure that the module gets a proper sys.argv as if it were
399 # run using `python -m`.
399 # run using `python -m`.
400 save_argv = sys.argv
400 save_argv = sys.argv
401 sys.argv = [sys.executable] + self.extra_args
401 sys.argv = [sys.executable] + self.extra_args
402 try:
402 try:
403 self.shell.safe_run_module(self.module_to_run,
403 self.shell.safe_run_module(self.module_to_run,
404 self.shell.user_ns)
404 self.shell.user_ns)
405 finally:
405 finally:
406 sys.argv = save_argv
406 sys.argv = save_argv
General Comments 0
You need to be logged in to leave comments. Login now