##// END OF EJS Templates
Provide a more useful error for invalid use case....
Jonathan Frederic -
Show More
@@ -1,230 +1,228 b''
1 #!/usr/bin/env python
1 #!/usr/bin/env python
2 """NBConvert is a utility for conversion of .ipynb files.
2 """NBConvert is a utility for conversion of .ipynb files.
3
3
4 Command-line interface for the NbConvert conversion utility.
4 Command-line interface for the NbConvert conversion utility.
5 """
5 """
6 #-----------------------------------------------------------------------------
6 #-----------------------------------------------------------------------------
7 #Copyright (c) 2013, the IPython Development Team.
7 #Copyright (c) 2013, the IPython Development Team.
8 #
8 #
9 #Distributed under the terms of the Modified BSD License.
9 #Distributed under the terms of the Modified BSD License.
10 #
10 #
11 #The full license is in the file COPYING.txt, distributed with this software.
11 #The full license is in the file COPYING.txt, distributed with this software.
12 #-----------------------------------------------------------------------------
12 #-----------------------------------------------------------------------------
13
13
14 #-----------------------------------------------------------------------------
14 #-----------------------------------------------------------------------------
15 #Imports
15 #Imports
16 #-----------------------------------------------------------------------------
16 #-----------------------------------------------------------------------------
17
17
18 # Stdlib imports
18 # Stdlib imports
19 from __future__ import print_function
19 from __future__ import print_function
20 import sys
20 import sys
21 import os
21 import os
22 import glob
22 import glob
23
23
24 # From IPython
24 # From IPython
25 from IPython.core.application import BaseIPythonApplication, base_aliases, base_flags
25 from IPython.core.application import BaseIPythonApplication, base_aliases, base_flags
26 from IPython.config import catch_config_error, Configurable
26 from IPython.config import catch_config_error, Configurable
27 from IPython.utils.traitlets import (
27 from IPython.utils.traitlets import (
28 Unicode, List, Instance, DottedObjectName, Type, CaselessStrEnum,
28 Unicode, List, Instance, DottedObjectName, Type, CaselessStrEnum,
29 )
29 )
30 from IPython.utils.importstring import import_item
30 from IPython.utils.importstring import import_item
31
31
32 from .exporters.export import export_by_name, get_export_names, ExporterNameError
32 from .exporters.export import export_by_name, get_export_names, ExporterNameError
33 from IPython.nbconvert import exporters, transformers, writers
33 from IPython.nbconvert import exporters, transformers, writers
34 from .utils.base import NbConvertBase
34 from .utils.base import NbConvertBase
35
35
36 #-----------------------------------------------------------------------------
36 #-----------------------------------------------------------------------------
37 #Classes and functions
37 #Classes and functions
38 #-----------------------------------------------------------------------------
38 #-----------------------------------------------------------------------------
39
39
40 nbconvert_aliases = {}
40 nbconvert_aliases = {}
41 nbconvert_aliases.update(base_aliases)
41 nbconvert_aliases.update(base_aliases)
42 nbconvert_aliases.update({
42 nbconvert_aliases.update({
43 'format' : 'NbConvertApp.export_format',
43 'format' : 'NbConvertApp.export_format',
44 'notebooks' : 'NbConvertApp.notebooks',
44 'notebooks' : 'NbConvertApp.notebooks',
45 'writer' : 'NbConvertApp.writer_class',
45 'writer' : 'NbConvertApp.writer_class',
46 })
46 })
47
47
48 nbconvert_flags = {}
48 nbconvert_flags = {}
49 nbconvert_flags.update(base_flags)
49 nbconvert_flags.update(base_flags)
50 nbconvert_flags.update({
50 nbconvert_flags.update({
51 'stdout' : (
51 'stdout' : (
52 {'NbConvertApp' : {'writer_class' : "StdoutWriter"}},
52 {'NbConvertApp' : {'writer_class' : "StdoutWriter"}},
53 "Write notebook output to stdout instead of files."
53 "Write notebook output to stdout instead of files."
54 )
54 )
55 })
55 })
56
56
57
57
58 class NbConvertApp(BaseIPythonApplication):
58 class NbConvertApp(BaseIPythonApplication):
59 """Application used to convert to and from notebook file type (*.ipynb)"""
59 """Application used to convert to and from notebook file type (*.ipynb)"""
60
60
61 name = 'ipython-nbconvert'
61 name = 'ipython-nbconvert'
62 aliases = nbconvert_aliases
62 aliases = nbconvert_aliases
63 flags = nbconvert_flags
63 flags = nbconvert_flags
64
64
65 def _classes_default(self):
65 def _classes_default(self):
66 classes = [NbConvertBase]
66 classes = [NbConvertBase]
67 for pkg in (exporters, transformers, writers):
67 for pkg in (exporters, transformers, writers):
68 for name in dir(pkg):
68 for name in dir(pkg):
69 cls = getattr(pkg, name)
69 cls = getattr(pkg, name)
70 if isinstance(cls, type) and issubclass(cls, Configurable):
70 if isinstance(cls, type) and issubclass(cls, Configurable):
71 classes.append(cls)
71 classes.append(cls)
72 return classes
72 return classes
73
73
74 description = Unicode(
74 description = Unicode(
75 u"""This application is used to convert notebook files (*.ipynb)
75 u"""This application is used to convert notebook files (*.ipynb)
76 to various other formats.""")
76 to various other formats.""")
77
77
78 examples = Unicode(u"""
78 examples = Unicode(u"""
79 The simplest way to use nbconvert is
79 The simplest way to use nbconvert is
80
80
81 > ipython nbconvert mynotebook.ipynb
81 > ipython nbconvert mynotebook.ipynb
82
82
83 which will convert mynotebook.ipynb to the default format (probably HTML).
83 which will convert mynotebook.ipynb to the default format (probably HTML).
84
84
85 You can specify the export format with `--format`.
85 You can specify the export format with `--format`.
86 Options include {0}
86 Options include {0}
87
87
88 > ipython nbconvert --format latex mynotebook.ipnynb
88 > ipython nbconvert --format latex mynotebook.ipnynb
89
89
90 You can also pipe the output to stdout, rather than a file
90 You can also pipe the output to stdout, rather than a file
91
91
92 > ipython nbconvert mynotebook.ipynb --stdout
92 > ipython nbconvert mynotebook.ipynb --stdout
93
93
94 Multiple notebooks can be given at the command line in a couple of
94 Multiple notebooks can be given at the command line in a couple of
95 different ways:
95 different ways:
96
96
97 > ipython nbconvert notebook*.ipynb
97 > ipython nbconvert notebook*.ipynb
98 > ipython nbconvert notebook1.ipynb notebook2.ipynb
98 > ipython nbconvert notebook1.ipynb notebook2.ipynb
99
99
100 or you can specify the notebooks list in a config file, containing::
100 or you can specify the notebooks list in a config file, containing::
101
101
102 c.NbConvertApp.notebooks = ["my_notebook.ipynb"]
102 c.NbConvertApp.notebooks = ["my_notebook.ipynb"]
103
103
104 > ipython nbconvert --config mycfg.py
104 > ipython nbconvert --config mycfg.py
105 """.format(get_export_names()))
105 """.format(get_export_names()))
106 # Writer specific variables
106 # Writer specific variables
107 writer = Instance('IPython.nbconvert.writers.base.WriterBase',
107 writer = Instance('IPython.nbconvert.writers.base.WriterBase',
108 help="""Instance of the writer class used to write the
108 help="""Instance of the writer class used to write the
109 results of the conversion.""")
109 results of the conversion.""")
110 writer_class = DottedObjectName('FilesWriter', config=True,
110 writer_class = DottedObjectName('FilesWriter', config=True,
111 help="""Writer class used to write the
111 help="""Writer class used to write the
112 results of the conversion""")
112 results of the conversion""")
113 writer_aliases = {'FilesWriter': 'IPython.nbconvert.writers.files.FilesWriter',
113 writer_aliases = {'FilesWriter': 'IPython.nbconvert.writers.files.FilesWriter',
114 'DebugWriter': 'IPython.nbconvert.writers.debug.DebugWriter',
114 'DebugWriter': 'IPython.nbconvert.writers.debug.DebugWriter',
115 'StdoutWriter': 'IPython.nbconvert.writers.stdout.StdoutWriter'}
115 'StdoutWriter': 'IPython.nbconvert.writers.stdout.StdoutWriter'}
116 writer_factory = Type()
116 writer_factory = Type()
117
117
118 def _writer_class_changed(self, name, old, new):
118 def _writer_class_changed(self, name, old, new):
119 if new in self.writer_aliases:
119 if new in self.writer_aliases:
120 new = self.writer_aliases[new]
120 new = self.writer_aliases[new]
121 self.writer_factory = import_item(new)
121 self.writer_factory = import_item(new)
122
122
123
123
124 # Other configurable variables
124 # Other configurable variables
125 export_format = CaselessStrEnum(get_export_names(),
125 export_format = CaselessStrEnum(get_export_names(),
126 default_value="full_html",
126 default_value="full_html",
127 config=True,
127 config=True,
128 help="""The export format to be used."""
128 help="""The export format to be used."""
129 )
129 )
130
130
131 notebooks = List([], config=True, help="""List of notebooks to convert.
131 notebooks = List([], config=True, help="""List of notebooks to convert.
132 Wildcards are supported.
132 Wildcards are supported.
133 Filenames passed positionally will be added to the list.
133 Filenames passed positionally will be added to the list.
134 """)
134 """)
135
135
136 @catch_config_error
136 @catch_config_error
137 def initialize(self, argv=None):
137 def initialize(self, argv=None):
138 super(NbConvertApp, self).initialize(argv)
138 super(NbConvertApp, self).initialize(argv)
139 self.init_notebooks()
139 self.init_notebooks()
140 self.init_writer()
140 self.init_writer()
141
141
142 def init_notebooks(self):
142 def init_notebooks(self):
143 """Construct the list of notebooks.
143 """Construct the list of notebooks.
144 If notebooks are passed on the command-line,
144 If notebooks are passed on the command-line,
145 they override notebooks specified in config files.
145 they override notebooks specified in config files.
146 Glob each notebook to replace notebook patterns with filenames.
146 Glob each notebook to replace notebook patterns with filenames.
147 """
147 """
148
148
149 # Specifying notebooks on the command-line overrides (rather than adds)
149 # Specifying notebooks on the command-line overrides (rather than adds)
150 # the notebook list
150 # the notebook list
151 if self.extra_args:
151 if self.extra_args:
152 patterns = self.extra_args
152 patterns = self.extra_args
153 else:
153 else:
154 patterns = self.notebooks
154 patterns = self.notebooks
155
155
156 # Use glob to replace all the notebook patterns with filenames.
156 # Use glob to replace all the notebook patterns with filenames.
157 filenames = []
157 filenames = []
158 for pattern in patterns:
158 for pattern in patterns:
159 for filename in glob.glob(pattern):
159 for filename in glob.glob(pattern):
160 if not filename in filenames:
160 if not filename in filenames:
161 filenames.append(filename)
161 filenames.append(filename)
162 self.notebooks = filenames
162 self.notebooks = filenames
163
163
164 def init_writer(self):
164 def init_writer(self):
165 """
165 """
166 Initialize the writer (which is stateless)
166 Initialize the writer (which is stateless)
167 """
167 """
168 self._writer_class_changed(None, self.writer_class, self.writer_class)
168 self._writer_class_changed(None, self.writer_class, self.writer_class)
169 self.writer = self.writer_factory(parent=self)
169 self.writer = self.writer_factory(parent=self)
170
170
171 def start(self):
171 def start(self):
172 """
172 """
173 Ran after initialization completed
173 Ran after initialization completed
174 """
174 """
175 super(NbConvertApp, self).start()
175 super(NbConvertApp, self).start()
176 self.convert_notebooks()
176 self.convert_notebooks()
177
177
178 def convert_notebooks(self):
178 def convert_notebooks(self):
179 """
179 """
180 Convert the notebooks in the self.notebook traitlet
180 Convert the notebooks in the self.notebook traitlet
181 """
181 """
182 # Export each notebook
182 # Export each notebook
183 conversion_success = 0
183 conversion_success = 0
184 for notebook_filename in self.notebooks:
184 for notebook_filename in self.notebooks:
185
185
186 # Get a unique key for the notebook and set it in the resources object.
186 # Get a unique key for the notebook and set it in the resources object.
187 basename = os.path.basename(notebook_filename)
187 basename = os.path.basename(notebook_filename)
188 notebook_name = basename[:basename.rfind('.')]
188 notebook_name = basename[:basename.rfind('.')]
189 resources = {}
189 resources = {}
190 resources['unique_key'] = notebook_name
190 resources['unique_key'] = notebook_name
191 resources['output_files_dir'] = '%s_files' % notebook_name
191 resources['output_files_dir'] = '%s_files' % notebook_name
192
192
193 # Try to export
193 # Try to export
194 try:
194 try:
195 output, resources = export_by_name(self.export_format,
195 output, resources = export_by_name(self.export_format,
196 notebook_filename,
196 notebook_filename,
197 resources=resources,
197 resources=resources,
198 config=self.config)
198 config=self.config)
199 except ExporterNameError as e:
199 except ExporterNameError as e:
200 print("Error: '%s' exporter not found." % self.export_format,
200 print("Error: '%s' exporter not found." % self.export_format,
201 file=sys.stderr)
201 file=sys.stderr)
202 print("Known exporters are:",
202 print("Known exporters are:",
203 "\n\t" + "\n\t".join(get_export_names()),
203 "\n\t" + "\n\t".join(get_export_names()),
204 file=sys.stderr)
204 file=sys.stderr)
205 sys.exit(-1)
205 sys.exit(-1)
206 # except Exception as e:
206 # except Exception as e:
207 # print("Error: could not export '%s'" % notebook_filename, file=sys.stderr)
207 # print("Error: could not export '%s'" % notebook_filename, file=sys.stderr)
208 # print(e, file=sys.stderr)
208 # print(e, file=sys.stderr)
209 else:
209 else:
210 self.writer.write(output, resources, notebook_name=notebook_name)
210 self.writer.write(output, resources, notebook_name=notebook_name)
211 conversion_success += 1
211 conversion_success += 1
212
212
213 # If nothing was converted successfully, help the user.
213 # If nothing was converted successfully, help the user.
214 if conversion_success == 0:
214 if conversion_success == 0:
215
215
216 # No notebooks were specified, show help.
216 # No notebooks were specified, show help.
217 if len(self.notebooks) == 0:
217 if len(self.notebooks) == 0:
218 self.print_help()
218 self.print_examples()
219
219
220 # Notebooks were specified, but not converted successfully. Show how
220 # Show how to access help.
221 # to access help.
221 print('For help, use "ipython nbconvert --help"')
222 else:
223 print('For help, use "ipython nbconvert --help"')
224
222
225
223
226 #-----------------------------------------------------------------------------
224 #-----------------------------------------------------------------------------
227 # Main entry point
225 # Main entry point
228 #-----------------------------------------------------------------------------
226 #-----------------------------------------------------------------------------
229
227
230 launch_new_instance = NbConvertApp.launch_instance
228 launch_new_instance = NbConvertApp.launch_instance
General Comments 0
You need to be logged in to leave comments. Login now