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