##// END OF EJS Templates
CLN: core/display: pep8 specifies double newlines between module-level callables
Wes Turner -
Show More
@@ -1,1342 +1,1347
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2 """Top-level display functions for displaying object in different formats."""
2 """Top-level display functions for displaying object in different formats."""
3
3
4 # Copyright (c) IPython Development Team.
4 # Copyright (c) IPython Development Team.
5 # Distributed under the terms of the Modified BSD License.
5 # Distributed under the terms of the Modified BSD License.
6
6
7
7
8 from binascii import b2a_base64, hexlify
8 from binascii import b2a_base64, hexlify
9 import html
9 import html
10 import json
10 import json
11 import mimetypes
11 import mimetypes
12 import os
12 import os
13 import struct
13 import struct
14 import warnings
14 import warnings
15 from copy import deepcopy
15 from copy import deepcopy
16 from os.path import splitext
16 from os.path import splitext
17 from pathlib import Path, PurePath
17 from pathlib import Path, PurePath
18
18
19 from IPython.utils.py3compat import cast_unicode
19 from IPython.utils.py3compat import cast_unicode
20 from IPython.testing.skipdoctest import skip_doctest
20 from IPython.testing.skipdoctest import skip_doctest
21 from . import display_functions
21 from . import display_functions
22
22
23
23
24 __all__ = ['display_pretty', 'display_html', 'display_markdown',
24 __all__ = ['display_pretty', 'display_html', 'display_markdown',
25 'display_svg', 'display_png', 'display_jpeg', 'display_webp',
25 'display_svg', 'display_png', 'display_jpeg', 'display_webp',
26 'display_latex', 'display_json',
26 'display_latex', 'display_json',
27 'display_javascript', 'display_pdf', 'DisplayObject', 'TextDisplayObject',
27 'display_javascript', 'display_pdf', 'DisplayObject', 'TextDisplayObject',
28 'Pretty', 'HTML', 'Markdown', 'Math', 'Latex', 'SVG', 'ProgressBar', 'JSON',
28 'Pretty', 'HTML', 'Markdown', 'Math', 'Latex', 'SVG', 'ProgressBar', 'JSON',
29 'GeoJSON', 'Javascript', 'Image', 'set_matplotlib_formats',
29 'GeoJSON', 'Javascript', 'Image', 'set_matplotlib_formats',
30 'set_matplotlib_close',
30 'set_matplotlib_close',
31 'Video']
31 'Video']
32
32
33 _deprecated_names = ["display", "clear_output", "publish_display_data", "update_display", "DisplayHandle"]
33 _deprecated_names = ["display", "clear_output", "publish_display_data", "update_display", "DisplayHandle"]
34
34
35 __all__ = __all__ + _deprecated_names
35 __all__ = __all__ + _deprecated_names
36
36
37
37
38 # ----- warn to import from IPython.display -----
38 # ----- warn to import from IPython.display -----
39
39
40 from warnings import warn
40 from warnings import warn
41
41
42
42
43 def __getattr__(name):
43 def __getattr__(name):
44 if name in _deprecated_names:
44 if name in _deprecated_names:
45 warn(
45 warn(
46 f"Importing {name} from IPython.core.display is deprecated since IPython 7.14, please import from IPython.display",
46 f"Importing {name} from IPython.core.display is deprecated since IPython 7.14, please import from IPython.display",
47 DeprecationWarning,
47 DeprecationWarning,
48 stacklevel=2,
48 stacklevel=2,
49 )
49 )
50 return getattr(display_functions, name)
50 return getattr(display_functions, name)
51
51
52 if name in globals().keys():
52 if name in globals().keys():
53 return globals()[name]
53 return globals()[name]
54 else:
54 else:
55 raise AttributeError(f"module {__name__} has no attribute {name}")
55 raise AttributeError(f"module {__name__} has no attribute {name}")
56
56
57
57
58 #-----------------------------------------------------------------------------
58 #-----------------------------------------------------------------------------
59 # utility functions
59 # utility functions
60 #-----------------------------------------------------------------------------
60 #-----------------------------------------------------------------------------
61
61
62 def _safe_exists(path):
62 def _safe_exists(path):
63 """Check path, but don't let exceptions raise"""
63 """Check path, but don't let exceptions raise"""
64 try:
64 try:
65 return os.path.exists(path)
65 return os.path.exists(path)
66 except Exception:
66 except Exception:
67 return False
67 return False
68
68
69
69
70 def _display_mimetype(mimetype, objs, raw=False, metadata=None):
70 def _display_mimetype(mimetype, objs, raw=False, metadata=None):
71 """internal implementation of all display_foo methods
71 """internal implementation of all display_foo methods
72
72
73 Parameters
73 Parameters
74 ----------
74 ----------
75 mimetype : str
75 mimetype : str
76 The mimetype to be published (e.g. 'image/png')
76 The mimetype to be published (e.g. 'image/png')
77 *objs : object
77 *objs : object
78 The Python objects to display, or if raw=True raw text data to
78 The Python objects to display, or if raw=True raw text data to
79 display.
79 display.
80 raw : bool
80 raw : bool
81 Are the data objects raw data or Python objects that need to be
81 Are the data objects raw data or Python objects that need to be
82 formatted before display? [default: False]
82 formatted before display? [default: False]
83 metadata : dict (optional)
83 metadata : dict (optional)
84 Metadata to be associated with the specific mimetype output.
84 Metadata to be associated with the specific mimetype output.
85 """
85 """
86 if metadata:
86 if metadata:
87 metadata = {mimetype: metadata}
87 metadata = {mimetype: metadata}
88 if raw:
88 if raw:
89 # turn list of pngdata into list of { 'image/png': pngdata }
89 # turn list of pngdata into list of { 'image/png': pngdata }
90 objs = [ {mimetype: obj} for obj in objs ]
90 objs = [ {mimetype: obj} for obj in objs ]
91 display_functions.display(*objs, raw=raw, metadata=metadata, include=[mimetype])
91 display_functions.display(*objs, raw=raw, metadata=metadata, include=[mimetype])
92
92
93 #-----------------------------------------------------------------------------
93 #-----------------------------------------------------------------------------
94 # Main functions
94 # Main functions
95 #-----------------------------------------------------------------------------
95 #-----------------------------------------------------------------------------
96
96
97
97
98 def display_pretty(*objs, **kwargs):
98 def display_pretty(*objs, **kwargs):
99 """Display the pretty (default) representation of an object.
99 """Display the pretty (default) representation of an object.
100
100
101 Parameters
101 Parameters
102 ----------
102 ----------
103 *objs : object
103 *objs : object
104 The Python objects to display, or if raw=True raw text data to
104 The Python objects to display, or if raw=True raw text data to
105 display.
105 display.
106 raw : bool
106 raw : bool
107 Are the data objects raw data or Python objects that need to be
107 Are the data objects raw data or Python objects that need to be
108 formatted before display? [default: False]
108 formatted before display? [default: False]
109 metadata : dict (optional)
109 metadata : dict (optional)
110 Metadata to be associated with the specific mimetype output.
110 Metadata to be associated with the specific mimetype output.
111 """
111 """
112 _display_mimetype('text/plain', objs, **kwargs)
112 _display_mimetype('text/plain', objs, **kwargs)
113
113
114
114
115 def display_html(*objs, **kwargs):
115 def display_html(*objs, **kwargs):
116 """Display the HTML representation of an object.
116 """Display the HTML representation of an object.
117
117
118 Note: If raw=False and the object does not have a HTML
118 Note: If raw=False and the object does not have a HTML
119 representation, no HTML will be shown.
119 representation, no HTML will be shown.
120
120
121 Parameters
121 Parameters
122 ----------
122 ----------
123 *objs : object
123 *objs : object
124 The Python objects to display, or if raw=True raw HTML data to
124 The Python objects to display, or if raw=True raw HTML data to
125 display.
125 display.
126 raw : bool
126 raw : bool
127 Are the data objects raw data or Python objects that need to be
127 Are the data objects raw data or Python objects that need to be
128 formatted before display? [default: False]
128 formatted before display? [default: False]
129 metadata : dict (optional)
129 metadata : dict (optional)
130 Metadata to be associated with the specific mimetype output.
130 Metadata to be associated with the specific mimetype output.
131 """
131 """
132 _display_mimetype('text/html', objs, **kwargs)
132 _display_mimetype('text/html', objs, **kwargs)
133
133
134
134
135 def display_markdown(*objs, **kwargs):
135 def display_markdown(*objs, **kwargs):
136 """Displays the Markdown representation of an object.
136 """Displays the Markdown representation of an object.
137
137
138 Parameters
138 Parameters
139 ----------
139 ----------
140 *objs : object
140 *objs : object
141 The Python objects to display, or if raw=True raw markdown data to
141 The Python objects to display, or if raw=True raw markdown data to
142 display.
142 display.
143 raw : bool
143 raw : bool
144 Are the data objects raw data or Python objects that need to be
144 Are the data objects raw data or Python objects that need to be
145 formatted before display? [default: False]
145 formatted before display? [default: False]
146 metadata : dict (optional)
146 metadata : dict (optional)
147 Metadata to be associated with the specific mimetype output.
147 Metadata to be associated with the specific mimetype output.
148 """
148 """
149
149
150 _display_mimetype('text/markdown', objs, **kwargs)
150 _display_mimetype('text/markdown', objs, **kwargs)
151
151
152
152
153 def display_svg(*objs, **kwargs):
153 def display_svg(*objs, **kwargs):
154 """Display the SVG representation of an object.
154 """Display the SVG representation of an object.
155
155
156 Parameters
156 Parameters
157 ----------
157 ----------
158 *objs : object
158 *objs : object
159 The Python objects to display, or if raw=True raw svg data to
159 The Python objects to display, or if raw=True raw svg data to
160 display.
160 display.
161 raw : bool
161 raw : bool
162 Are the data objects raw data or Python objects that need to be
162 Are the data objects raw data or Python objects that need to be
163 formatted before display? [default: False]
163 formatted before display? [default: False]
164 metadata : dict (optional)
164 metadata : dict (optional)
165 Metadata to be associated with the specific mimetype output.
165 Metadata to be associated with the specific mimetype output.
166 """
166 """
167 _display_mimetype('image/svg+xml', objs, **kwargs)
167 _display_mimetype('image/svg+xml', objs, **kwargs)
168
168
169
169
170 def display_png(*objs, **kwargs):
170 def display_png(*objs, **kwargs):
171 """Display the PNG representation of an object.
171 """Display the PNG representation of an object.
172
172
173 Parameters
173 Parameters
174 ----------
174 ----------
175 *objs : object
175 *objs : object
176 The Python objects to display, or if raw=True raw png data to
176 The Python objects to display, or if raw=True raw png data to
177 display.
177 display.
178 raw : bool
178 raw : bool
179 Are the data objects raw data or Python objects that need to be
179 Are the data objects raw data or Python objects that need to be
180 formatted before display? [default: False]
180 formatted before display? [default: False]
181 metadata : dict (optional)
181 metadata : dict (optional)
182 Metadata to be associated with the specific mimetype output.
182 Metadata to be associated with the specific mimetype output.
183 """
183 """
184 _display_mimetype('image/png', objs, **kwargs)
184 _display_mimetype('image/png', objs, **kwargs)
185
185
186
186
187 def display_jpeg(*objs, **kwargs):
187 def display_jpeg(*objs, **kwargs):
188 """Display the JPEG representation of an object.
188 """Display the JPEG representation of an object.
189
189
190 Parameters
190 Parameters
191 ----------
191 ----------
192 *objs : object
192 *objs : object
193 The Python objects to display, or if raw=True raw JPEG data to
193 The Python objects to display, or if raw=True raw JPEG data to
194 display.
194 display.
195 raw : bool
195 raw : bool
196 Are the data objects raw data or Python objects that need to be
196 Are the data objects raw data or Python objects that need to be
197 formatted before display? [default: False]
197 formatted before display? [default: False]
198 metadata : dict (optional)
198 metadata : dict (optional)
199 Metadata to be associated with the specific mimetype output.
199 Metadata to be associated with the specific mimetype output.
200 """
200 """
201 _display_mimetype('image/jpeg', objs, **kwargs)
201 _display_mimetype('image/jpeg', objs, **kwargs)
202
202
203
203
204 def display_webp(*objs, **kwargs):
204 def display_webp(*objs, **kwargs):
205 """Display the WEBP representation of an object.
205 """Display the WEBP representation of an object.
206
206
207 Parameters
207 Parameters
208 ----------
208 ----------
209 *objs : object
209 *objs : object
210 The Python objects to display, or if raw=True raw JPEG data to
210 The Python objects to display, or if raw=True raw JPEG data to
211 display.
211 display.
212 raw : bool
212 raw : bool
213 Are the data objects raw data or Python objects that need to be
213 Are the data objects raw data or Python objects that need to be
214 formatted before display? [default: False]
214 formatted before display? [default: False]
215 metadata : dict (optional)
215 metadata : dict (optional)
216 Metadata to be associated with the specific mimetype output.
216 Metadata to be associated with the specific mimetype output.
217 """
217 """
218 _display_mimetype('image/webp', objs, **kwargs)
218 _display_mimetype('image/webp', objs, **kwargs)
219
219
220
220
221 def display_latex(*objs, **kwargs):
221 def display_latex(*objs, **kwargs):
222 """Display the LaTeX representation of an object.
222 """Display the LaTeX representation of an object.
223
223
224 Parameters
224 Parameters
225 ----------
225 ----------
226 *objs : object
226 *objs : object
227 The Python objects to display, or if raw=True raw latex data to
227 The Python objects to display, or if raw=True raw latex data to
228 display.
228 display.
229 raw : bool
229 raw : bool
230 Are the data objects raw data or Python objects that need to be
230 Are the data objects raw data or Python objects that need to be
231 formatted before display? [default: False]
231 formatted before display? [default: False]
232 metadata : dict (optional)
232 metadata : dict (optional)
233 Metadata to be associated with the specific mimetype output.
233 Metadata to be associated with the specific mimetype output.
234 """
234 """
235 _display_mimetype('text/latex', objs, **kwargs)
235 _display_mimetype('text/latex', objs, **kwargs)
236
236
237
237
238 def display_json(*objs, **kwargs):
238 def display_json(*objs, **kwargs):
239 """Display the JSON representation of an object.
239 """Display the JSON representation of an object.
240
240
241 Note that not many frontends support displaying JSON.
241 Note that not many frontends support displaying JSON.
242
242
243 Parameters
243 Parameters
244 ----------
244 ----------
245 *objs : object
245 *objs : object
246 The Python objects to display, or if raw=True raw json data to
246 The Python objects to display, or if raw=True raw json data to
247 display.
247 display.
248 raw : bool
248 raw : bool
249 Are the data objects raw data or Python objects that need to be
249 Are the data objects raw data or Python objects that need to be
250 formatted before display? [default: False]
250 formatted before display? [default: False]
251 metadata : dict (optional)
251 metadata : dict (optional)
252 Metadata to be associated with the specific mimetype output.
252 Metadata to be associated with the specific mimetype output.
253 """
253 """
254 _display_mimetype('application/json', objs, **kwargs)
254 _display_mimetype('application/json', objs, **kwargs)
255
255
256
256
257 def display_javascript(*objs, **kwargs):
257 def display_javascript(*objs, **kwargs):
258 """Display the Javascript representation of an object.
258 """Display the Javascript representation of an object.
259
259
260 Parameters
260 Parameters
261 ----------
261 ----------
262 *objs : object
262 *objs : object
263 The Python objects to display, or if raw=True raw javascript data to
263 The Python objects to display, or if raw=True raw javascript data to
264 display.
264 display.
265 raw : bool
265 raw : bool
266 Are the data objects raw data or Python objects that need to be
266 Are the data objects raw data or Python objects that need to be
267 formatted before display? [default: False]
267 formatted before display? [default: False]
268 metadata : dict (optional)
268 metadata : dict (optional)
269 Metadata to be associated with the specific mimetype output.
269 Metadata to be associated with the specific mimetype output.
270 """
270 """
271 _display_mimetype('application/javascript', objs, **kwargs)
271 _display_mimetype('application/javascript', objs, **kwargs)
272
272
273
273
274 def display_pdf(*objs, **kwargs):
274 def display_pdf(*objs, **kwargs):
275 """Display the PDF representation of an object.
275 """Display the PDF representation of an object.
276
276
277 Parameters
277 Parameters
278 ----------
278 ----------
279 *objs : object
279 *objs : object
280 The Python objects to display, or if raw=True raw javascript data to
280 The Python objects to display, or if raw=True raw javascript data to
281 display.
281 display.
282 raw : bool
282 raw : bool
283 Are the data objects raw data or Python objects that need to be
283 Are the data objects raw data or Python objects that need to be
284 formatted before display? [default: False]
284 formatted before display? [default: False]
285 metadata : dict (optional)
285 metadata : dict (optional)
286 Metadata to be associated with the specific mimetype output.
286 Metadata to be associated with the specific mimetype output.
287 """
287 """
288 _display_mimetype('application/pdf', objs, **kwargs)
288 _display_mimetype('application/pdf', objs, **kwargs)
289
289
290
290
291 #-----------------------------------------------------------------------------
291 #-----------------------------------------------------------------------------
292 # Smart classes
292 # Smart classes
293 #-----------------------------------------------------------------------------
293 #-----------------------------------------------------------------------------
294
294
295
295
296 class DisplayObject(object):
296 class DisplayObject(object):
297 """An object that wraps data to be displayed."""
297 """An object that wraps data to be displayed."""
298
298
299 _read_flags = 'r'
299 _read_flags = 'r'
300 _show_mem_addr = False
300 _show_mem_addr = False
301 metadata = None
301 metadata = None
302
302
303 def __init__(self, data=None, url=None, filename=None, metadata=None):
303 def __init__(self, data=None, url=None, filename=None, metadata=None):
304 """Create a display object given raw data.
304 """Create a display object given raw data.
305
305
306 When this object is returned by an expression or passed to the
306 When this object is returned by an expression or passed to the
307 display function, it will result in the data being displayed
307 display function, it will result in the data being displayed
308 in the frontend. The MIME type of the data should match the
308 in the frontend. The MIME type of the data should match the
309 subclasses used, so the Png subclass should be used for 'image/png'
309 subclasses used, so the Png subclass should be used for 'image/png'
310 data. If the data is a URL, the data will first be downloaded
310 data. If the data is a URL, the data will first be downloaded
311 and then displayed.
311 and then displayed.
312
312
313 Parameters
313 Parameters
314 ----------
314 ----------
315 data : unicode, str or bytes
315 data : unicode, str or bytes
316 The raw data or a URL or file to load the data from
316 The raw data or a URL or file to load the data from
317 url : unicode
317 url : unicode
318 A URL to download the data from.
318 A URL to download the data from.
319 filename : unicode
319 filename : unicode
320 Path to a local file to load the data from.
320 Path to a local file to load the data from.
321 metadata : dict
321 metadata : dict
322 Dict of metadata associated to be the object when displayed
322 Dict of metadata associated to be the object when displayed
323 """
323 """
324 if isinstance(data, (Path, PurePath)):
324 if isinstance(data, (Path, PurePath)):
325 data = str(data)
325 data = str(data)
326
326
327 if data is not None and isinstance(data, str):
327 if data is not None and isinstance(data, str):
328 if data.startswith('http') and url is None:
328 if data.startswith('http') and url is None:
329 url = data
329 url = data
330 filename = None
330 filename = None
331 data = None
331 data = None
332 elif _safe_exists(data) and filename is None:
332 elif _safe_exists(data) and filename is None:
333 url = None
333 url = None
334 filename = data
334 filename = data
335 data = None
335 data = None
336
336
337 self.url = url
337 self.url = url
338 self.filename = filename
338 self.filename = filename
339 # because of @data.setter methods in
339 # because of @data.setter methods in
340 # subclasses ensure url and filename are set
340 # subclasses ensure url and filename are set
341 # before assigning to self.data
341 # before assigning to self.data
342 self.data = data
342 self.data = data
343
343
344 if metadata is not None:
344 if metadata is not None:
345 self.metadata = metadata
345 self.metadata = metadata
346 elif self.metadata is None:
346 elif self.metadata is None:
347 self.metadata = {}
347 self.metadata = {}
348
348
349 self.reload()
349 self.reload()
350 self._check_data()
350 self._check_data()
351
351
352 def __repr__(self):
352 def __repr__(self):
353 if not self._show_mem_addr:
353 if not self._show_mem_addr:
354 cls = self.__class__
354 cls = self.__class__
355 r = "<%s.%s object>" % (cls.__module__, cls.__name__)
355 r = "<%s.%s object>" % (cls.__module__, cls.__name__)
356 else:
356 else:
357 r = super(DisplayObject, self).__repr__()
357 r = super(DisplayObject, self).__repr__()
358 return r
358 return r
359
359
360 def _check_data(self):
360 def _check_data(self):
361 """Override in subclasses if there's something to check."""
361 """Override in subclasses if there's something to check."""
362 pass
362 pass
363
363
364 def _data_and_metadata(self):
364 def _data_and_metadata(self):
365 """shortcut for returning metadata with shape information, if defined"""
365 """shortcut for returning metadata with shape information, if defined"""
366 if self.metadata:
366 if self.metadata:
367 return self.data, deepcopy(self.metadata)
367 return self.data, deepcopy(self.metadata)
368 else:
368 else:
369 return self.data
369 return self.data
370
370
371 def reload(self):
371 def reload(self):
372 """Reload the raw data from file or URL."""
372 """Reload the raw data from file or URL."""
373 if self.filename is not None:
373 if self.filename is not None:
374 encoding = None if "b" in self._read_flags else "utf-8"
374 encoding = None if "b" in self._read_flags else "utf-8"
375 with open(self.filename, self._read_flags, encoding=encoding) as f:
375 with open(self.filename, self._read_flags, encoding=encoding) as f:
376 self.data = f.read()
376 self.data = f.read()
377 elif self.url is not None:
377 elif self.url is not None:
378 # Deferred import
378 # Deferred import
379 from urllib.request import urlopen
379 from urllib.request import urlopen
380 response = urlopen(self.url)
380 response = urlopen(self.url)
381 data = response.read()
381 data = response.read()
382 # extract encoding from header, if there is one:
382 # extract encoding from header, if there is one:
383 encoding = None
383 encoding = None
384 if 'content-type' in response.headers:
384 if 'content-type' in response.headers:
385 for sub in response.headers['content-type'].split(';'):
385 for sub in response.headers['content-type'].split(';'):
386 sub = sub.strip()
386 sub = sub.strip()
387 if sub.startswith('charset'):
387 if sub.startswith('charset'):
388 encoding = sub.split('=')[-1].strip()
388 encoding = sub.split('=')[-1].strip()
389 break
389 break
390 if 'content-encoding' in response.headers:
390 if 'content-encoding' in response.headers:
391 # TODO: do deflate?
391 # TODO: do deflate?
392 if 'gzip' in response.headers['content-encoding']:
392 if 'gzip' in response.headers['content-encoding']:
393 import gzip
393 import gzip
394 from io import BytesIO
394 from io import BytesIO
395
395
396 # assume utf-8 if encoding is not specified
396 # assume utf-8 if encoding is not specified
397 with gzip.open(
397 with gzip.open(
398 BytesIO(data), "rt", encoding=encoding or "utf-8"
398 BytesIO(data), "rt", encoding=encoding or "utf-8"
399 ) as fp:
399 ) as fp:
400 encoding = None
400 encoding = None
401 data = fp.read()
401 data = fp.read()
402
402
403 # decode data, if an encoding was specified
403 # decode data, if an encoding was specified
404 # We only touch self.data once since
404 # We only touch self.data once since
405 # subclasses such as SVG have @data.setter methods
405 # subclasses such as SVG have @data.setter methods
406 # that transform self.data into ... well svg.
406 # that transform self.data into ... well svg.
407 if encoding:
407 if encoding:
408 self.data = data.decode(encoding, 'replace')
408 self.data = data.decode(encoding, 'replace')
409 else:
409 else:
410 self.data = data
410 self.data = data
411
411
412
412
413 class TextDisplayObject(DisplayObject):
413 class TextDisplayObject(DisplayObject):
414 """Create a text display object given raw data.
414 """Create a text display object given raw data.
415
415
416 Parameters
416 Parameters
417 ----------
417 ----------
418 data : str or unicode
418 data : str or unicode
419 The raw data or a URL or file to load the data from.
419 The raw data or a URL or file to load the data from.
420 url : unicode
420 url : unicode
421 A URL to download the data from.
421 A URL to download the data from.
422 filename : unicode
422 filename : unicode
423 Path to a local file to load the data from.
423 Path to a local file to load the data from.
424 metadata : dict
424 metadata : dict
425 Dict of metadata associated to be the object when displayed
425 Dict of metadata associated to be the object when displayed
426 """
426 """
427 def _check_data(self):
427 def _check_data(self):
428 if self.data is not None and not isinstance(self.data, str):
428 if self.data is not None and not isinstance(self.data, str):
429 raise TypeError("%s expects text, not %r" % (self.__class__.__name__, self.data))
429 raise TypeError("%s expects text, not %r" % (self.__class__.__name__, self.data))
430
430
431 class Pretty(TextDisplayObject):
431 class Pretty(TextDisplayObject):
432
432
433 def _repr_pretty_(self, pp, cycle):
433 def _repr_pretty_(self, pp, cycle):
434 return pp.text(self.data)
434 return pp.text(self.data)
435
435
436
436
437 class HTML(TextDisplayObject):
437 class HTML(TextDisplayObject):
438
438
439 def __init__(self, data=None, url=None, filename=None, metadata=None):
439 def __init__(self, data=None, url=None, filename=None, metadata=None):
440 def warn():
440 def warn():
441 if not data:
441 if not data:
442 return False
442 return False
443
443
444 #
444 #
445 # Avoid calling lower() on the entire data, because it could be a
445 # Avoid calling lower() on the entire data, because it could be a
446 # long string and we're only interested in its beginning and end.
446 # long string and we're only interested in its beginning and end.
447 #
447 #
448 prefix = data[:10].lower()
448 prefix = data[:10].lower()
449 suffix = data[-10:].lower()
449 suffix = data[-10:].lower()
450 return prefix.startswith("<iframe ") and suffix.endswith("</iframe>")
450 return prefix.startswith("<iframe ") and suffix.endswith("</iframe>")
451
451
452 if warn():
452 if warn():
453 warnings.warn("Consider using IPython.display.IFrame instead")
453 warnings.warn("Consider using IPython.display.IFrame instead")
454 super(HTML, self).__init__(data=data, url=url, filename=filename, metadata=metadata)
454 super(HTML, self).__init__(data=data, url=url, filename=filename, metadata=metadata)
455
455
456 def _repr_html_(self):
456 def _repr_html_(self):
457 return self._data_and_metadata()
457 return self._data_and_metadata()
458
458
459 def __html__(self):
459 def __html__(self):
460 """
460 """
461 This method exists to inform other HTML-using modules (e.g. Markupsafe,
461 This method exists to inform other HTML-using modules (e.g. Markupsafe,
462 htmltag, etc) that this object is HTML and does not need things like
462 htmltag, etc) that this object is HTML and does not need things like
463 special characters (<>&) escaped.
463 special characters (<>&) escaped.
464 """
464 """
465 return self._repr_html_()
465 return self._repr_html_()
466
466
467
467
468 class Markdown(TextDisplayObject):
468 class Markdown(TextDisplayObject):
469
469
470 def _repr_markdown_(self):
470 def _repr_markdown_(self):
471 return self._data_and_metadata()
471 return self._data_and_metadata()
472
472
473
473
474 class Math(TextDisplayObject):
474 class Math(TextDisplayObject):
475
475
476 def _repr_latex_(self):
476 def _repr_latex_(self):
477 s = r"$\displaystyle %s$" % self.data.strip('$')
477 s = r"$\displaystyle %s$" % self.data.strip('$')
478 if self.metadata:
478 if self.metadata:
479 return s, deepcopy(self.metadata)
479 return s, deepcopy(self.metadata)
480 else:
480 else:
481 return s
481 return s
482
482
483
483
484 class Latex(TextDisplayObject):
484 class Latex(TextDisplayObject):
485
485
486 def _repr_latex_(self):
486 def _repr_latex_(self):
487 return self._data_and_metadata()
487 return self._data_and_metadata()
488
488
489
489
490 class SVG(DisplayObject):
490 class SVG(DisplayObject):
491 """Embed an SVG into the display.
491 """Embed an SVG into the display.
492
492
493 Note if you just want to view a svg image via a URL use `:class:Image` with
493 Note if you just want to view a svg image via a URL use `:class:Image` with
494 a url=URL keyword argument.
494 a url=URL keyword argument.
495 """
495 """
496
496
497 _read_flags = 'rb'
497 _read_flags = 'rb'
498 # wrap data in a property, which extracts the <svg> tag, discarding
498 # wrap data in a property, which extracts the <svg> tag, discarding
499 # document headers
499 # document headers
500 _data = None
500 _data = None
501
501
502 @property
502 @property
503 def data(self):
503 def data(self):
504 return self._data
504 return self._data
505
505
506 @data.setter
506 @data.setter
507 def data(self, svg):
507 def data(self, svg):
508 if svg is None:
508 if svg is None:
509 self._data = None
509 self._data = None
510 return
510 return
511 # parse into dom object
511 # parse into dom object
512 from xml.dom import minidom
512 from xml.dom import minidom
513 x = minidom.parseString(svg)
513 x = minidom.parseString(svg)
514 # get svg tag (should be 1)
514 # get svg tag (should be 1)
515 found_svg = x.getElementsByTagName('svg')
515 found_svg = x.getElementsByTagName('svg')
516 if found_svg:
516 if found_svg:
517 svg = found_svg[0].toxml()
517 svg = found_svg[0].toxml()
518 else:
518 else:
519 # fallback on the input, trust the user
519 # fallback on the input, trust the user
520 # but this is probably an error.
520 # but this is probably an error.
521 pass
521 pass
522 svg = cast_unicode(svg)
522 svg = cast_unicode(svg)
523 self._data = svg
523 self._data = svg
524
524
525 def _repr_svg_(self):
525 def _repr_svg_(self):
526 return self._data_and_metadata()
526 return self._data_and_metadata()
527
527
528 class ProgressBar(DisplayObject):
528 class ProgressBar(DisplayObject):
529 """Progressbar supports displaying a progressbar like element
529 """Progressbar supports displaying a progressbar like element
530 """
530 """
531 def __init__(self, total):
531 def __init__(self, total):
532 """Creates a new progressbar
532 """Creates a new progressbar
533
533
534 Parameters
534 Parameters
535 ----------
535 ----------
536 total : int
536 total : int
537 maximum size of the progressbar
537 maximum size of the progressbar
538 """
538 """
539 self.total = total
539 self.total = total
540 self._progress = 0
540 self._progress = 0
541 self.html_width = '60ex'
541 self.html_width = '60ex'
542 self.text_width = 60
542 self.text_width = 60
543 self._display_id = hexlify(os.urandom(8)).decode('ascii')
543 self._display_id = hexlify(os.urandom(8)).decode('ascii')
544
544
545 def __repr__(self):
545 def __repr__(self):
546 fraction = self.progress / self.total
546 fraction = self.progress / self.total
547 filled = '=' * int(fraction * self.text_width)
547 filled = '=' * int(fraction * self.text_width)
548 rest = ' ' * (self.text_width - len(filled))
548 rest = ' ' * (self.text_width - len(filled))
549 return '[{}{}] {}/{}'.format(
549 return '[{}{}] {}/{}'.format(
550 filled, rest,
550 filled, rest,
551 self.progress, self.total,
551 self.progress, self.total,
552 )
552 )
553
553
554 def _repr_html_(self):
554 def _repr_html_(self):
555 return "<progress style='width:{}' max='{}' value='{}'></progress>".format(
555 return "<progress style='width:{}' max='{}' value='{}'></progress>".format(
556 self.html_width, self.total, self.progress)
556 self.html_width, self.total, self.progress)
557
557
558 def display(self):
558 def display(self):
559 display_functions.display(self, display_id=self._display_id)
559 display_functions.display(self, display_id=self._display_id)
560
560
561 def update(self):
561 def update(self):
562 display_functions.display(self, display_id=self._display_id, update=True)
562 display_functions.display(self, display_id=self._display_id, update=True)
563
563
564 @property
564 @property
565 def progress(self):
565 def progress(self):
566 return self._progress
566 return self._progress
567
567
568 @progress.setter
568 @progress.setter
569 def progress(self, value):
569 def progress(self, value):
570 self._progress = value
570 self._progress = value
571 self.update()
571 self.update()
572
572
573 def __iter__(self):
573 def __iter__(self):
574 self.display()
574 self.display()
575 self._progress = -1 # First iteration is 0
575 self._progress = -1 # First iteration is 0
576 return self
576 return self
577
577
578 def __next__(self):
578 def __next__(self):
579 """Returns current value and increments display by one."""
579 """Returns current value and increments display by one."""
580 self.progress += 1
580 self.progress += 1
581 if self.progress < self.total:
581 if self.progress < self.total:
582 return self.progress
582 return self.progress
583 else:
583 else:
584 raise StopIteration()
584 raise StopIteration()
585
585
586 class JSON(DisplayObject):
586 class JSON(DisplayObject):
587 """JSON expects a JSON-able dict or list
587 """JSON expects a JSON-able dict or list
588
588
589 not an already-serialized JSON string.
589 not an already-serialized JSON string.
590
590
591 Scalar types (None, number, string) are not allowed, only dict or list containers.
591 Scalar types (None, number, string) are not allowed, only dict or list containers.
592 """
592 """
593 # wrap data in a property, which warns about passing already-serialized JSON
593 # wrap data in a property, which warns about passing already-serialized JSON
594 _data = None
594 _data = None
595 def __init__(self, data=None, url=None, filename=None, expanded=False, metadata=None, root='root', **kwargs):
595 def __init__(self, data=None, url=None, filename=None, expanded=False, metadata=None, root='root', **kwargs):
596 """Create a JSON display object given raw data.
596 """Create a JSON display object given raw data.
597
597
598 Parameters
598 Parameters
599 ----------
599 ----------
600 data : dict or list
600 data : dict or list
601 JSON data to display. Not an already-serialized JSON string.
601 JSON data to display. Not an already-serialized JSON string.
602 Scalar types (None, number, string) are not allowed, only dict
602 Scalar types (None, number, string) are not allowed, only dict
603 or list containers.
603 or list containers.
604 url : unicode
604 url : unicode
605 A URL to download the data from.
605 A URL to download the data from.
606 filename : unicode
606 filename : unicode
607 Path to a local file to load the data from.
607 Path to a local file to load the data from.
608 expanded : boolean
608 expanded : boolean
609 Metadata to control whether a JSON display component is expanded.
609 Metadata to control whether a JSON display component is expanded.
610 metadata : dict
610 metadata : dict
611 Specify extra metadata to attach to the json display object.
611 Specify extra metadata to attach to the json display object.
612 root : str
612 root : str
613 The name of the root element of the JSON tree
613 The name of the root element of the JSON tree
614 """
614 """
615 self.metadata = {
615 self.metadata = {
616 'expanded': expanded,
616 'expanded': expanded,
617 'root': root,
617 'root': root,
618 }
618 }
619 if metadata:
619 if metadata:
620 self.metadata.update(metadata)
620 self.metadata.update(metadata)
621 if kwargs:
621 if kwargs:
622 self.metadata.update(kwargs)
622 self.metadata.update(kwargs)
623 super(JSON, self).__init__(data=data, url=url, filename=filename)
623 super(JSON, self).__init__(data=data, url=url, filename=filename)
624
624
625 def _check_data(self):
625 def _check_data(self):
626 if self.data is not None and not isinstance(self.data, (dict, list)):
626 if self.data is not None and not isinstance(self.data, (dict, list)):
627 raise TypeError("%s expects JSONable dict or list, not %r" % (self.__class__.__name__, self.data))
627 raise TypeError("%s expects JSONable dict or list, not %r" % (self.__class__.__name__, self.data))
628
628
629 @property
629 @property
630 def data(self):
630 def data(self):
631 return self._data
631 return self._data
632
632
633 @data.setter
633 @data.setter
634 def data(self, data):
634 def data(self, data):
635 if isinstance(data, (Path, PurePath)):
635 if isinstance(data, (Path, PurePath)):
636 data = str(data)
636 data = str(data)
637
637
638 if isinstance(data, str):
638 if isinstance(data, str):
639 if self.filename is None and self.url is None:
639 if self.filename is None and self.url is None:
640 warnings.warn("JSON expects JSONable dict or list, not JSON strings")
640 warnings.warn("JSON expects JSONable dict or list, not JSON strings")
641 data = json.loads(data)
641 data = json.loads(data)
642 self._data = data
642 self._data = data
643
643
644 def _data_and_metadata(self):
644 def _data_and_metadata(self):
645 return self.data, self.metadata
645 return self.data, self.metadata
646
646
647 def _repr_json_(self):
647 def _repr_json_(self):
648 return self._data_and_metadata()
648 return self._data_and_metadata()
649
649
650
650
651 _css_t = """var link = document.createElement("link");
651 _css_t = """var link = document.createElement("link");
652 link.rel = "stylesheet";
652 link.rel = "stylesheet";
653 link.type = "text/css";
653 link.type = "text/css";
654 link.href = "%s";
654 link.href = "%s";
655 document.head.appendChild(link);
655 document.head.appendChild(link);
656 """
656 """
657
657
658 _lib_t1 = """new Promise(function(resolve, reject) {
658 _lib_t1 = """new Promise(function(resolve, reject) {
659 var script = document.createElement("script");
659 var script = document.createElement("script");
660 script.onload = resolve;
660 script.onload = resolve;
661 script.onerror = reject;
661 script.onerror = reject;
662 script.src = "%s";
662 script.src = "%s";
663 document.head.appendChild(script);
663 document.head.appendChild(script);
664 }).then(() => {
664 }).then(() => {
665 """
665 """
666
666
667 _lib_t2 = """
667 _lib_t2 = """
668 });"""
668 });"""
669
669
670 class GeoJSON(JSON):
670 class GeoJSON(JSON):
671 """GeoJSON expects JSON-able dict
671 """GeoJSON expects JSON-able dict
672
672
673 not an already-serialized JSON string.
673 not an already-serialized JSON string.
674
674
675 Scalar types (None, number, string) are not allowed, only dict containers.
675 Scalar types (None, number, string) are not allowed, only dict containers.
676 """
676 """
677
677
678 def __init__(self, *args, **kwargs):
678 def __init__(self, *args, **kwargs):
679 """Create a GeoJSON display object given raw data.
679 """Create a GeoJSON display object given raw data.
680
680
681 Parameters
681 Parameters
682 ----------
682 ----------
683 data : dict or list
683 data : dict or list
684 VegaLite data. Not an already-serialized JSON string.
684 VegaLite data. Not an already-serialized JSON string.
685 Scalar types (None, number, string) are not allowed, only dict
685 Scalar types (None, number, string) are not allowed, only dict
686 or list containers.
686 or list containers.
687 url_template : string
687 url_template : string
688 Leaflet TileLayer URL template: http://leafletjs.com/reference.html#url-template
688 Leaflet TileLayer URL template: http://leafletjs.com/reference.html#url-template
689 layer_options : dict
689 layer_options : dict
690 Leaflet TileLayer options: http://leafletjs.com/reference.html#tilelayer-options
690 Leaflet TileLayer options: http://leafletjs.com/reference.html#tilelayer-options
691 url : unicode
691 url : unicode
692 A URL to download the data from.
692 A URL to download the data from.
693 filename : unicode
693 filename : unicode
694 Path to a local file to load the data from.
694 Path to a local file to load the data from.
695 metadata : dict
695 metadata : dict
696 Specify extra metadata to attach to the json display object.
696 Specify extra metadata to attach to the json display object.
697
697
698 Examples
698 Examples
699 --------
699 --------
700 The following will display an interactive map of Mars with a point of
700 The following will display an interactive map of Mars with a point of
701 interest on frontend that do support GeoJSON display.
701 interest on frontend that do support GeoJSON display.
702
702
703 >>> from IPython.display import GeoJSON
703 >>> from IPython.display import GeoJSON
704
704
705 >>> GeoJSON(data={
705 >>> GeoJSON(data={
706 ... "type": "Feature",
706 ... "type": "Feature",
707 ... "geometry": {
707 ... "geometry": {
708 ... "type": "Point",
708 ... "type": "Point",
709 ... "coordinates": [-81.327, 296.038]
709 ... "coordinates": [-81.327, 296.038]
710 ... }
710 ... }
711 ... },
711 ... },
712 ... url_template="http://s3-eu-west-1.amazonaws.com/whereonmars.cartodb.net/{basemap_id}/{z}/{x}/{y}.png",
712 ... url_template="http://s3-eu-west-1.amazonaws.com/whereonmars.cartodb.net/{basemap_id}/{z}/{x}/{y}.png",
713 ... layer_options={
713 ... layer_options={
714 ... "basemap_id": "celestia_mars-shaded-16k_global",
714 ... "basemap_id": "celestia_mars-shaded-16k_global",
715 ... "attribution" : "Celestia/praesepe",
715 ... "attribution" : "Celestia/praesepe",
716 ... "minZoom" : 0,
716 ... "minZoom" : 0,
717 ... "maxZoom" : 18,
717 ... "maxZoom" : 18,
718 ... })
718 ... })
719 <IPython.core.display.GeoJSON object>
719 <IPython.core.display.GeoJSON object>
720
720
721 In the terminal IPython, you will only see the text representation of
721 In the terminal IPython, you will only see the text representation of
722 the GeoJSON object.
722 the GeoJSON object.
723
723
724 """
724 """
725
725
726 super(GeoJSON, self).__init__(*args, **kwargs)
726 super(GeoJSON, self).__init__(*args, **kwargs)
727
727
728
728
729 def _ipython_display_(self):
729 def _ipython_display_(self):
730 bundle = {
730 bundle = {
731 'application/geo+json': self.data,
731 'application/geo+json': self.data,
732 'text/plain': '<IPython.display.GeoJSON object>'
732 'text/plain': '<IPython.display.GeoJSON object>'
733 }
733 }
734 metadata = {
734 metadata = {
735 'application/geo+json': self.metadata
735 'application/geo+json': self.metadata
736 }
736 }
737 display_functions.display(bundle, metadata=metadata, raw=True)
737 display_functions.display(bundle, metadata=metadata, raw=True)
738
738
739 class Javascript(TextDisplayObject):
739 class Javascript(TextDisplayObject):
740
740
741 def __init__(self, data=None, url=None, filename=None, lib=None, css=None):
741 def __init__(self, data=None, url=None, filename=None, lib=None, css=None):
742 """Create a Javascript display object given raw data.
742 """Create a Javascript display object given raw data.
743
743
744 When this object is returned by an expression or passed to the
744 When this object is returned by an expression or passed to the
745 display function, it will result in the data being displayed
745 display function, it will result in the data being displayed
746 in the frontend. If the data is a URL, the data will first be
746 in the frontend. If the data is a URL, the data will first be
747 downloaded and then displayed.
747 downloaded and then displayed.
748
748
749 In the Notebook, the containing element will be available as `element`,
749 In the Notebook, the containing element will be available as `element`,
750 and jQuery will be available. Content appended to `element` will be
750 and jQuery will be available. Content appended to `element` will be
751 visible in the output area.
751 visible in the output area.
752
752
753 Parameters
753 Parameters
754 ----------
754 ----------
755 data : unicode, str or bytes
755 data : unicode, str or bytes
756 The Javascript source code or a URL to download it from.
756 The Javascript source code or a URL to download it from.
757 url : unicode
757 url : unicode
758 A URL to download the data from.
758 A URL to download the data from.
759 filename : unicode
759 filename : unicode
760 Path to a local file to load the data from.
760 Path to a local file to load the data from.
761 lib : list or str
761 lib : list or str
762 A sequence of Javascript library URLs to load asynchronously before
762 A sequence of Javascript library URLs to load asynchronously before
763 running the source code. The full URLs of the libraries should
763 running the source code. The full URLs of the libraries should
764 be given. A single Javascript library URL can also be given as a
764 be given. A single Javascript library URL can also be given as a
765 string.
765 string.
766 css : list or str
766 css : list or str
767 A sequence of css files to load before running the source code.
767 A sequence of css files to load before running the source code.
768 The full URLs of the css files should be given. A single css URL
768 The full URLs of the css files should be given. A single css URL
769 can also be given as a string.
769 can also be given as a string.
770 """
770 """
771 if isinstance(lib, str):
771 if isinstance(lib, str):
772 lib = [lib]
772 lib = [lib]
773 elif lib is None:
773 elif lib is None:
774 lib = []
774 lib = []
775 if isinstance(css, str):
775 if isinstance(css, str):
776 css = [css]
776 css = [css]
777 elif css is None:
777 elif css is None:
778 css = []
778 css = []
779 if not isinstance(lib, (list,tuple)):
779 if not isinstance(lib, (list,tuple)):
780 raise TypeError('expected sequence, got: %r' % lib)
780 raise TypeError('expected sequence, got: %r' % lib)
781 if not isinstance(css, (list,tuple)):
781 if not isinstance(css, (list,tuple)):
782 raise TypeError('expected sequence, got: %r' % css)
782 raise TypeError('expected sequence, got: %r' % css)
783 self.lib = lib
783 self.lib = lib
784 self.css = css
784 self.css = css
785 super(Javascript, self).__init__(data=data, url=url, filename=filename)
785 super(Javascript, self).__init__(data=data, url=url, filename=filename)
786
786
787 def _repr_javascript_(self):
787 def _repr_javascript_(self):
788 r = ''
788 r = ''
789 for c in self.css:
789 for c in self.css:
790 r += _css_t % c
790 r += _css_t % c
791 for l in self.lib:
791 for l in self.lib:
792 r += _lib_t1 % l
792 r += _lib_t1 % l
793 r += self.data
793 r += self.data
794 r += _lib_t2*len(self.lib)
794 r += _lib_t2*len(self.lib)
795 return r
795 return r
796
796
797
797 # constants for identifying png/jpeg/gif/webp data
798 # constants for identifying png/jpeg/gif/webp data
798 _PNG = b'\x89PNG\r\n\x1a\n'
799 _PNG = b'\x89PNG\r\n\x1a\n'
799 _JPEG = b'\xff\xd8'
800 _JPEG = b'\xff\xd8'
800 _GIF1 = b"GIF87a"
801 _GIF1 = b"GIF87a"
801 _GIF2 = b"GIF89a"
802 _GIF2 = b"GIF89a"
802 _WEBP = b'WEBP'
803 _WEBP = b'WEBP'
803
804
805
804 def _pngxy(data):
806 def _pngxy(data):
805 """read the (width, height) from a PNG header"""
807 """read the (width, height) from a PNG header"""
806 ihdr = data.index(b'IHDR')
808 ihdr = data.index(b'IHDR')
807 # next 8 bytes are width/height
809 # next 8 bytes are width/height
808 return struct.unpack('>ii', data[ihdr+4:ihdr+12])
810 return struct.unpack('>ii', data[ihdr+4:ihdr+12])
809
811
812
810 def _jpegxy(data):
813 def _jpegxy(data):
811 """read the (width, height) from a JPEG header"""
814 """read the (width, height) from a JPEG header"""
812 # adapted from http://www.64lines.com/jpeg-width-height
815 # adapted from http://www.64lines.com/jpeg-width-height
813
816
814 idx = 4
817 idx = 4
815 while True:
818 while True:
816 block_size = struct.unpack('>H', data[idx:idx+2])[0]
819 block_size = struct.unpack('>H', data[idx:idx+2])[0]
817 idx = idx + block_size
820 idx = idx + block_size
818 if data[idx:idx+2] == b'\xFF\xC0':
821 if data[idx:idx+2] == b'\xFF\xC0':
819 # found Start of Frame
822 # found Start of Frame
820 iSOF = idx
823 iSOF = idx
821 break
824 break
822 else:
825 else:
823 # read another block
826 # read another block
824 idx += 2
827 idx += 2
825
828
826 h, w = struct.unpack('>HH', data[iSOF+5:iSOF+9])
829 h, w = struct.unpack('>HH', data[iSOF+5:iSOF+9])
827 return w, h
830 return w, h
828
831
832
829 def _gifxy(data):
833 def _gifxy(data):
830 """read the (width, height) from a GIF header"""
834 """read the (width, height) from a GIF header"""
831 return struct.unpack('<HH', data[6:10])
835 return struct.unpack('<HH', data[6:10])
832
836
837
833 def _webpxy(data):
838 def _webpxy(data):
834 """read the (width, height) from a WEBP header"""
839 """read the (width, height) from a WEBP header"""
835 if data[12:16] == b"VP8 ":
840 if data[12:16] == b"VP8 ":
836 width, height = struct.unpack('<HH', data[24:30])
841 width, height = struct.unpack('<HH', data[24:30])
837 width = (width & 0x3fff)
842 width = (width & 0x3fff)
838 height = (height & 0x3fff)
843 height = (height & 0x3fff)
839 return (width, height)
844 return (width, height)
840 elif data[12:16] == b"VP8L":
845 elif data[12:16] == b"VP8L":
841 size_info = struct.unpack('<I', data[21:25])[0]
846 size_info = struct.unpack('<I', data[21:25])[0]
842 width = 1 + ((size_info & 0x3F) << 8) | (size_info >> 24)
847 width = 1 + ((size_info & 0x3F) << 8) | (size_info >> 24)
843 height = 1 + ((((size_info >> 8) & 0xF) << 10) |
848 height = 1 + ((((size_info >> 8) & 0xF) << 10) |
844 (((size_info >> 14) & 0x3FC) << 2) |
849 (((size_info >> 14) & 0x3FC) << 2) |
845 ((size_info >> 22) & 0x3))
850 ((size_info >> 22) & 0x3))
846 return (width, height)
851 return (width, height)
847 else:
852 else:
848 raise ValueError("Not a valid WEBP header")
853 raise ValueError("Not a valid WEBP header")
849
854
850
855
851 class Image(DisplayObject):
856 class Image(DisplayObject):
852
857
853 _read_flags = 'rb'
858 _read_flags = 'rb'
854 _FMT_JPEG = u'jpeg'
859 _FMT_JPEG = u'jpeg'
855 _FMT_PNG = u'png'
860 _FMT_PNG = u'png'
856 _FMT_GIF = u'gif'
861 _FMT_GIF = u'gif'
857 _FMT_WEBP = u'webp'
862 _FMT_WEBP = u'webp'
858 _ACCEPTABLE_EMBEDDINGS = [_FMT_JPEG, _FMT_PNG, _FMT_GIF, _FMT_WEBP]
863 _ACCEPTABLE_EMBEDDINGS = [_FMT_JPEG, _FMT_PNG, _FMT_GIF, _FMT_WEBP]
859 _MIMETYPES = {
864 _MIMETYPES = {
860 _FMT_PNG: 'image/png',
865 _FMT_PNG: 'image/png',
861 _FMT_JPEG: 'image/jpeg',
866 _FMT_JPEG: 'image/jpeg',
862 _FMT_GIF: 'image/gif',
867 _FMT_GIF: 'image/gif',
863 _FMT_WEBP: 'image/webp',
868 _FMT_WEBP: 'image/webp',
864 }
869 }
865
870
866 def __init__(
871 def __init__(
867 self,
872 self,
868 data=None,
873 data=None,
869 url=None,
874 url=None,
870 filename=None,
875 filename=None,
871 format=None,
876 format=None,
872 embed=None,
877 embed=None,
873 width=None,
878 width=None,
874 height=None,
879 height=None,
875 retina=False,
880 retina=False,
876 unconfined=False,
881 unconfined=False,
877 metadata=None,
882 metadata=None,
878 alt=None,
883 alt=None,
879 ):
884 ):
880 """Create a PNG/JPEG/GIF image object given raw data.
885 """Create a PNG/JPEG/GIF image object given raw data.
881
886
882 When this object is returned by an input cell or passed to the
887 When this object is returned by an input cell or passed to the
883 display function, it will result in the image being displayed
888 display function, it will result in the image being displayed
884 in the frontend.
889 in the frontend.
885
890
886 Parameters
891 Parameters
887 ----------
892 ----------
888 data : unicode, str or bytes
893 data : unicode, str or bytes
889 The raw image data or a URL or filename to load the data from.
894 The raw image data or a URL or filename to load the data from.
890 This always results in embedded image data.
895 This always results in embedded image data.
891
896
892 url : unicode
897 url : unicode
893 A URL to download the data from. If you specify `url=`,
898 A URL to download the data from. If you specify `url=`,
894 the image data will not be embedded unless you also specify `embed=True`.
899 the image data will not be embedded unless you also specify `embed=True`.
895
900
896 filename : unicode
901 filename : unicode
897 Path to a local file to load the data from.
902 Path to a local file to load the data from.
898 Images from a file are always embedded.
903 Images from a file are always embedded.
899
904
900 format : unicode
905 format : unicode
901 The format of the image data (png/jpeg/jpg/gif/webp). If a filename or URL is given
906 The format of the image data (png/jpeg/jpg/gif/webp). If a filename or URL is given
902 for format will be inferred from the filename extension.
907 for format will be inferred from the filename extension.
903
908
904 embed : bool
909 embed : bool
905 Should the image data be embedded using a data URI (True) or be
910 Should the image data be embedded using a data URI (True) or be
906 loaded using an <img> tag. Set this to True if you want the image
911 loaded using an <img> tag. Set this to True if you want the image
907 to be viewable later with no internet connection in the notebook.
912 to be viewable later with no internet connection in the notebook.
908
913
909 Default is `True`, unless the keyword argument `url` is set, then
914 Default is `True`, unless the keyword argument `url` is set, then
910 default value is `False`.
915 default value is `False`.
911
916
912 Note that QtConsole is not able to display images if `embed` is set to `False`
917 Note that QtConsole is not able to display images if `embed` is set to `False`
913
918
914 width : int
919 width : int
915 Width in pixels to which to constrain the image in html
920 Width in pixels to which to constrain the image in html
916
921
917 height : int
922 height : int
918 Height in pixels to which to constrain the image in html
923 Height in pixels to which to constrain the image in html
919
924
920 retina : bool
925 retina : bool
921 Automatically set the width and height to half of the measured
926 Automatically set the width and height to half of the measured
922 width and height.
927 width and height.
923 This only works for embedded images because it reads the width/height
928 This only works for embedded images because it reads the width/height
924 from image data.
929 from image data.
925 For non-embedded images, you can just set the desired display width
930 For non-embedded images, you can just set the desired display width
926 and height directly.
931 and height directly.
927
932
928 unconfined : bool
933 unconfined : bool
929 Set unconfined=True to disable max-width confinement of the image.
934 Set unconfined=True to disable max-width confinement of the image.
930
935
931 metadata : dict
936 metadata : dict
932 Specify extra metadata to attach to the image.
937 Specify extra metadata to attach to the image.
933
938
934 alt : unicode
939 alt : unicode
935 Alternative text for the image, for use by screen readers.
940 Alternative text for the image, for use by screen readers.
936
941
937 Examples
942 Examples
938 --------
943 --------
939 embedded image data, works in qtconsole and notebook
944 embedded image data, works in qtconsole and notebook
940 when passed positionally, the first arg can be any of raw image data,
945 when passed positionally, the first arg can be any of raw image data,
941 a URL, or a filename from which to load image data.
946 a URL, or a filename from which to load image data.
942 The result is always embedding image data for inline images.
947 The result is always embedding image data for inline images.
943
948
944 >>> Image('https://www.google.fr/images/srpr/logo3w.png') # doctest: +SKIP
949 >>> Image('https://www.google.fr/images/srpr/logo3w.png') # doctest: +SKIP
945 <IPython.core.display.Image object>
950 <IPython.core.display.Image object>
946
951
947 >>> Image('/path/to/image.jpg')
952 >>> Image('/path/to/image.jpg')
948 <IPython.core.display.Image object>
953 <IPython.core.display.Image object>
949
954
950 >>> Image(b'RAW_PNG_DATA...')
955 >>> Image(b'RAW_PNG_DATA...')
951 <IPython.core.display.Image object>
956 <IPython.core.display.Image object>
952
957
953 Specifying Image(url=...) does not embed the image data,
958 Specifying Image(url=...) does not embed the image data,
954 it only generates ``<img>`` tag with a link to the source.
959 it only generates ``<img>`` tag with a link to the source.
955 This will not work in the qtconsole or offline.
960 This will not work in the qtconsole or offline.
956
961
957 >>> Image(url='https://www.google.fr/images/srpr/logo3w.png')
962 >>> Image(url='https://www.google.fr/images/srpr/logo3w.png')
958 <IPython.core.display.Image object>
963 <IPython.core.display.Image object>
959
964
960 """
965 """
961 if isinstance(data, (Path, PurePath)):
966 if isinstance(data, (Path, PurePath)):
962 data = str(data)
967 data = str(data)
963
968
964 if filename is not None:
969 if filename is not None:
965 ext = self._find_ext(filename)
970 ext = self._find_ext(filename)
966 elif url is not None:
971 elif url is not None:
967 ext = self._find_ext(url)
972 ext = self._find_ext(url)
968 elif data is None:
973 elif data is None:
969 raise ValueError("No image data found. Expecting filename, url, or data.")
974 raise ValueError("No image data found. Expecting filename, url, or data.")
970 elif isinstance(data, str) and (
975 elif isinstance(data, str) and (
971 data.startswith('http') or _safe_exists(data)
976 data.startswith('http') or _safe_exists(data)
972 ):
977 ):
973 ext = self._find_ext(data)
978 ext = self._find_ext(data)
974 else:
979 else:
975 ext = None
980 ext = None
976
981
977 if format is None:
982 if format is None:
978 if ext is not None:
983 if ext is not None:
979 if ext == u'jpg' or ext == u'jpeg':
984 if ext == u'jpg' or ext == u'jpeg':
980 format = self._FMT_JPEG
985 format = self._FMT_JPEG
981 elif ext == u'png':
986 elif ext == u'png':
982 format = self._FMT_PNG
987 format = self._FMT_PNG
983 elif ext == u'gif':
988 elif ext == u'gif':
984 format = self._FMT_GIF
989 format = self._FMT_GIF
985 elif ext == u'webp':
990 elif ext == u'webp':
986 format = self._FMT_WEBP
991 format = self._FMT_WEBP
987 else:
992 else:
988 format = ext.lower()
993 format = ext.lower()
989 elif isinstance(data, bytes):
994 elif isinstance(data, bytes):
990 # infer image type from image data header,
995 # infer image type from image data header,
991 # only if format has not been specified.
996 # only if format has not been specified.
992 if data[:2] == _JPEG:
997 if data[:2] == _JPEG:
993 format = self._FMT_JPEG
998 format = self._FMT_JPEG
994 elif data[:8] == _PNG:
999 elif data[:8] == _PNG:
995 format = self._FMT_PNG
1000 format = self._FMT_PNG
996 elif data[8:12] == _WEBP:
1001 elif data[8:12] == _WEBP:
997 format = self._FMT_WEBP
1002 format = self._FMT_WEBP
998 elif data[:6] == self._GIF1 or data[:6] == self._GIF2:
1003 elif data[:6] == self._GIF1 or data[:6] == self._GIF2:
999 format = self._FMT_GIF
1004 format = self._FMT_GIF
1000
1005
1001 # failed to detect format, default png
1006 # failed to detect format, default png
1002 if format is None:
1007 if format is None:
1003 format = self._FMT_PNG
1008 format = self._FMT_PNG
1004
1009
1005 if format.lower() == 'jpg':
1010 if format.lower() == 'jpg':
1006 # jpg->jpeg
1011 # jpg->jpeg
1007 format = self._FMT_JPEG
1012 format = self._FMT_JPEG
1008
1013
1009 self.format = format.lower()
1014 self.format = format.lower()
1010 self.embed = embed if embed is not None else (url is None)
1015 self.embed = embed if embed is not None else (url is None)
1011
1016
1012 if self.embed and self.format not in self._ACCEPTABLE_EMBEDDINGS:
1017 if self.embed and self.format not in self._ACCEPTABLE_EMBEDDINGS:
1013 raise ValueError("Cannot embed the '%s' image format" % (self.format))
1018 raise ValueError("Cannot embed the '%s' image format" % (self.format))
1014 if self.embed:
1019 if self.embed:
1015 self._mimetype = self._MIMETYPES.get(self.format)
1020 self._mimetype = self._MIMETYPES.get(self.format)
1016
1021
1017 self.width = width
1022 self.width = width
1018 self.height = height
1023 self.height = height
1019 self.retina = retina
1024 self.retina = retina
1020 self.unconfined = unconfined
1025 self.unconfined = unconfined
1021 self.alt = alt
1026 self.alt = alt
1022 super(Image, self).__init__(data=data, url=url, filename=filename,
1027 super(Image, self).__init__(data=data, url=url, filename=filename,
1023 metadata=metadata)
1028 metadata=metadata)
1024
1029
1025 if self.width is None and self.metadata.get('width', {}):
1030 if self.width is None and self.metadata.get('width', {}):
1026 self.width = metadata['width']
1031 self.width = metadata['width']
1027
1032
1028 if self.height is None and self.metadata.get('height', {}):
1033 if self.height is None and self.metadata.get('height', {}):
1029 self.height = metadata['height']
1034 self.height = metadata['height']
1030
1035
1031 if self.alt is None and self.metadata.get("alt", {}):
1036 if self.alt is None and self.metadata.get("alt", {}):
1032 self.alt = metadata["alt"]
1037 self.alt = metadata["alt"]
1033
1038
1034 if retina:
1039 if retina:
1035 self._retina_shape()
1040 self._retina_shape()
1036
1041
1037
1042
1038 def _retina_shape(self):
1043 def _retina_shape(self):
1039 """load pixel-doubled width and height from image data"""
1044 """load pixel-doubled width and height from image data"""
1040 if not self.embed:
1045 if not self.embed:
1041 return
1046 return
1042 if self.format == self._FMT_PNG:
1047 if self.format == self._FMT_PNG:
1043 w, h = _pngxy(self.data)
1048 w, h = _pngxy(self.data)
1044 elif self.format == self._FMT_JPEG:
1049 elif self.format == self._FMT_JPEG:
1045 w, h = _jpegxy(self.data)
1050 w, h = _jpegxy(self.data)
1046 elif self.format == self._FMT_GIF:
1051 elif self.format == self._FMT_GIF:
1047 w, h = _gifxy(self.data)
1052 w, h = _gifxy(self.data)
1048 else:
1053 else:
1049 # retina only supports png
1054 # retina only supports png
1050 return
1055 return
1051 self.width = w // 2
1056 self.width = w // 2
1052 self.height = h // 2
1057 self.height = h // 2
1053
1058
1054 def reload(self):
1059 def reload(self):
1055 """Reload the raw data from file or URL."""
1060 """Reload the raw data from file or URL."""
1056 if self.embed:
1061 if self.embed:
1057 super(Image,self).reload()
1062 super(Image,self).reload()
1058 if self.retina:
1063 if self.retina:
1059 self._retina_shape()
1064 self._retina_shape()
1060
1065
1061 def _repr_html_(self):
1066 def _repr_html_(self):
1062 if not self.embed:
1067 if not self.embed:
1063 width = height = klass = alt = ""
1068 width = height = klass = alt = ""
1064 if self.width:
1069 if self.width:
1065 width = ' width="%d"' % self.width
1070 width = ' width="%d"' % self.width
1066 if self.height:
1071 if self.height:
1067 height = ' height="%d"' % self.height
1072 height = ' height="%d"' % self.height
1068 if self.unconfined:
1073 if self.unconfined:
1069 klass = ' class="unconfined"'
1074 klass = ' class="unconfined"'
1070 if self.alt:
1075 if self.alt:
1071 alt = ' alt="%s"' % html.escape(self.alt)
1076 alt = ' alt="%s"' % html.escape(self.alt)
1072 return '<img src="{url}"{width}{height}{klass}{alt}/>'.format(
1077 return '<img src="{url}"{width}{height}{klass}{alt}/>'.format(
1073 url=self.url,
1078 url=self.url,
1074 width=width,
1079 width=width,
1075 height=height,
1080 height=height,
1076 klass=klass,
1081 klass=klass,
1077 alt=alt,
1082 alt=alt,
1078 )
1083 )
1079
1084
1080 def _repr_mimebundle_(self, include=None, exclude=None):
1085 def _repr_mimebundle_(self, include=None, exclude=None):
1081 """Return the image as a mimebundle
1086 """Return the image as a mimebundle
1082
1087
1083 Any new mimetype support should be implemented here.
1088 Any new mimetype support should be implemented here.
1084 """
1089 """
1085 if self.embed:
1090 if self.embed:
1086 mimetype = self._mimetype
1091 mimetype = self._mimetype
1087 data, metadata = self._data_and_metadata(always_both=True)
1092 data, metadata = self._data_and_metadata(always_both=True)
1088 if metadata:
1093 if metadata:
1089 metadata = {mimetype: metadata}
1094 metadata = {mimetype: metadata}
1090 return {mimetype: data}, metadata
1095 return {mimetype: data}, metadata
1091 else:
1096 else:
1092 return {'text/html': self._repr_html_()}
1097 return {'text/html': self._repr_html_()}
1093
1098
1094 def _data_and_metadata(self, always_both=False):
1099 def _data_and_metadata(self, always_both=False):
1095 """shortcut for returning metadata with shape information, if defined"""
1100 """shortcut for returning metadata with shape information, if defined"""
1096 try:
1101 try:
1097 b64_data = b2a_base64(self.data, newline=False).decode("ascii")
1102 b64_data = b2a_base64(self.data, newline=False).decode("ascii")
1098 except TypeError as e:
1103 except TypeError as e:
1099 raise FileNotFoundError(
1104 raise FileNotFoundError(
1100 "No such file or directory: '%s'" % (self.data)) from e
1105 "No such file or directory: '%s'" % (self.data)) from e
1101 md = {}
1106 md = {}
1102 if self.metadata:
1107 if self.metadata:
1103 md.update(self.metadata)
1108 md.update(self.metadata)
1104 if self.width:
1109 if self.width:
1105 md['width'] = self.width
1110 md['width'] = self.width
1106 if self.height:
1111 if self.height:
1107 md['height'] = self.height
1112 md['height'] = self.height
1108 if self.unconfined:
1113 if self.unconfined:
1109 md['unconfined'] = self.unconfined
1114 md['unconfined'] = self.unconfined
1110 if self.alt:
1115 if self.alt:
1111 md["alt"] = self.alt
1116 md["alt"] = self.alt
1112 if md or always_both:
1117 if md or always_both:
1113 return b64_data, md
1118 return b64_data, md
1114 else:
1119 else:
1115 return b64_data
1120 return b64_data
1116
1121
1117 def _repr_png_(self):
1122 def _repr_png_(self):
1118 if self.embed and self.format == self._FMT_PNG:
1123 if self.embed and self.format == self._FMT_PNG:
1119 return self._data_and_metadata()
1124 return self._data_and_metadata()
1120
1125
1121 def _repr_jpeg_(self):
1126 def _repr_jpeg_(self):
1122 if self.embed and self.format == self._FMT_JPEG:
1127 if self.embed and self.format == self._FMT_JPEG:
1123 return self._data_and_metadata()
1128 return self._data_and_metadata()
1124
1129
1125 def _find_ext(self, s):
1130 def _find_ext(self, s):
1126 base, ext = splitext(s)
1131 base, ext = splitext(s)
1127
1132
1128 if not ext:
1133 if not ext:
1129 return base
1134 return base
1130
1135
1131 # `splitext` includes leading period, so we skip it
1136 # `splitext` includes leading period, so we skip it
1132 return ext[1:].lower()
1137 return ext[1:].lower()
1133
1138
1134
1139
1135 class Video(DisplayObject):
1140 class Video(DisplayObject):
1136
1141
1137 def __init__(self, data=None, url=None, filename=None, embed=False,
1142 def __init__(self, data=None, url=None, filename=None, embed=False,
1138 mimetype=None, width=None, height=None, html_attributes="controls"):
1143 mimetype=None, width=None, height=None, html_attributes="controls"):
1139 """Create a video object given raw data or an URL.
1144 """Create a video object given raw data or an URL.
1140
1145
1141 When this object is returned by an input cell or passed to the
1146 When this object is returned by an input cell or passed to the
1142 display function, it will result in the video being displayed
1147 display function, it will result in the video being displayed
1143 in the frontend.
1148 in the frontend.
1144
1149
1145 Parameters
1150 Parameters
1146 ----------
1151 ----------
1147 data : unicode, str or bytes
1152 data : unicode, str or bytes
1148 The raw video data or a URL or filename to load the data from.
1153 The raw video data or a URL or filename to load the data from.
1149 Raw data will require passing ``embed=True``.
1154 Raw data will require passing ``embed=True``.
1150
1155
1151 url : unicode
1156 url : unicode
1152 A URL for the video. If you specify ``url=``,
1157 A URL for the video. If you specify ``url=``,
1153 the image data will not be embedded.
1158 the image data will not be embedded.
1154
1159
1155 filename : unicode
1160 filename : unicode
1156 Path to a local file containing the video.
1161 Path to a local file containing the video.
1157 Will be interpreted as a local URL unless ``embed=True``.
1162 Will be interpreted as a local URL unless ``embed=True``.
1158
1163
1159 embed : bool
1164 embed : bool
1160 Should the video be embedded using a data URI (True) or be
1165 Should the video be embedded using a data URI (True) or be
1161 loaded using a <video> tag (False).
1166 loaded using a <video> tag (False).
1162
1167
1163 Since videos are large, embedding them should be avoided, if possible.
1168 Since videos are large, embedding them should be avoided, if possible.
1164 You must confirm embedding as your intention by passing ``embed=True``.
1169 You must confirm embedding as your intention by passing ``embed=True``.
1165
1170
1166 Local files can be displayed with URLs without embedding the content, via::
1171 Local files can be displayed with URLs without embedding the content, via::
1167
1172
1168 Video('./video.mp4')
1173 Video('./video.mp4')
1169
1174
1170 mimetype : unicode
1175 mimetype : unicode
1171 Specify the mimetype for embedded videos.
1176 Specify the mimetype for embedded videos.
1172 Default will be guessed from file extension, if available.
1177 Default will be guessed from file extension, if available.
1173
1178
1174 width : int
1179 width : int
1175 Width in pixels to which to constrain the video in HTML.
1180 Width in pixels to which to constrain the video in HTML.
1176 If not supplied, defaults to the width of the video.
1181 If not supplied, defaults to the width of the video.
1177
1182
1178 height : int
1183 height : int
1179 Height in pixels to which to constrain the video in html.
1184 Height in pixels to which to constrain the video in html.
1180 If not supplied, defaults to the height of the video.
1185 If not supplied, defaults to the height of the video.
1181
1186
1182 html_attributes : str
1187 html_attributes : str
1183 Attributes for the HTML ``<video>`` block.
1188 Attributes for the HTML ``<video>`` block.
1184 Default: ``"controls"`` to get video controls.
1189 Default: ``"controls"`` to get video controls.
1185 Other examples: ``"controls muted"`` for muted video with controls,
1190 Other examples: ``"controls muted"`` for muted video with controls,
1186 ``"loop autoplay"`` for looping autoplaying video without controls.
1191 ``"loop autoplay"`` for looping autoplaying video without controls.
1187
1192
1188 Examples
1193 Examples
1189 --------
1194 --------
1190 ::
1195 ::
1191
1196
1192 Video('https://archive.org/download/Sita_Sings_the_Blues/Sita_Sings_the_Blues_small.mp4')
1197 Video('https://archive.org/download/Sita_Sings_the_Blues/Sita_Sings_the_Blues_small.mp4')
1193 Video('path/to/video.mp4')
1198 Video('path/to/video.mp4')
1194 Video('path/to/video.mp4', embed=True)
1199 Video('path/to/video.mp4', embed=True)
1195 Video('path/to/video.mp4', embed=True, html_attributes="controls muted autoplay")
1200 Video('path/to/video.mp4', embed=True, html_attributes="controls muted autoplay")
1196 Video(b'raw-videodata', embed=True)
1201 Video(b'raw-videodata', embed=True)
1197 """
1202 """
1198 if isinstance(data, (Path, PurePath)):
1203 if isinstance(data, (Path, PurePath)):
1199 data = str(data)
1204 data = str(data)
1200
1205
1201 if url is None and isinstance(data, str) and data.startswith(('http:', 'https:')):
1206 if url is None and isinstance(data, str) and data.startswith(('http:', 'https:')):
1202 url = data
1207 url = data
1203 data = None
1208 data = None
1204 elif data is not None and os.path.exists(data):
1209 elif data is not None and os.path.exists(data):
1205 filename = data
1210 filename = data
1206 data = None
1211 data = None
1207
1212
1208 if data and not embed:
1213 if data and not embed:
1209 msg = ''.join([
1214 msg = ''.join([
1210 "To embed videos, you must pass embed=True ",
1215 "To embed videos, you must pass embed=True ",
1211 "(this may make your notebook files huge)\n",
1216 "(this may make your notebook files huge)\n",
1212 "Consider passing Video(url='...')",
1217 "Consider passing Video(url='...')",
1213 ])
1218 ])
1214 raise ValueError(msg)
1219 raise ValueError(msg)
1215
1220
1216 self.mimetype = mimetype
1221 self.mimetype = mimetype
1217 self.embed = embed
1222 self.embed = embed
1218 self.width = width
1223 self.width = width
1219 self.height = height
1224 self.height = height
1220 self.html_attributes = html_attributes
1225 self.html_attributes = html_attributes
1221 super(Video, self).__init__(data=data, url=url, filename=filename)
1226 super(Video, self).__init__(data=data, url=url, filename=filename)
1222
1227
1223 def _repr_html_(self):
1228 def _repr_html_(self):
1224 width = height = ''
1229 width = height = ''
1225 if self.width:
1230 if self.width:
1226 width = ' width="%d"' % self.width
1231 width = ' width="%d"' % self.width
1227 if self.height:
1232 if self.height:
1228 height = ' height="%d"' % self.height
1233 height = ' height="%d"' % self.height
1229
1234
1230 # External URLs and potentially local files are not embedded into the
1235 # External URLs and potentially local files are not embedded into the
1231 # notebook output.
1236 # notebook output.
1232 if not self.embed:
1237 if not self.embed:
1233 url = self.url if self.url is not None else self.filename
1238 url = self.url if self.url is not None else self.filename
1234 output = """<video src="{0}" {1} {2} {3}>
1239 output = """<video src="{0}" {1} {2} {3}>
1235 Your browser does not support the <code>video</code> element.
1240 Your browser does not support the <code>video</code> element.
1236 </video>""".format(url, self.html_attributes, width, height)
1241 </video>""".format(url, self.html_attributes, width, height)
1237 return output
1242 return output
1238
1243
1239 # Embedded videos are base64-encoded.
1244 # Embedded videos are base64-encoded.
1240 mimetype = self.mimetype
1245 mimetype = self.mimetype
1241 if self.filename is not None:
1246 if self.filename is not None:
1242 if not mimetype:
1247 if not mimetype:
1243 mimetype, _ = mimetypes.guess_type(self.filename)
1248 mimetype, _ = mimetypes.guess_type(self.filename)
1244
1249
1245 with open(self.filename, 'rb') as f:
1250 with open(self.filename, 'rb') as f:
1246 video = f.read()
1251 video = f.read()
1247 else:
1252 else:
1248 video = self.data
1253 video = self.data
1249 if isinstance(video, str):
1254 if isinstance(video, str):
1250 # unicode input is already b64-encoded
1255 # unicode input is already b64-encoded
1251 b64_video = video
1256 b64_video = video
1252 else:
1257 else:
1253 b64_video = b2a_base64(video, newline=False).decode("ascii").rstrip()
1258 b64_video = b2a_base64(video, newline=False).decode("ascii").rstrip()
1254
1259
1255 output = """<video {0} {1} {2}>
1260 output = """<video {0} {1} {2}>
1256 <source src="data:{3};base64,{4}" type="{3}">
1261 <source src="data:{3};base64,{4}" type="{3}">
1257 Your browser does not support the video tag.
1262 Your browser does not support the video tag.
1258 </video>""".format(self.html_attributes, width, height, mimetype, b64_video)
1263 </video>""".format(self.html_attributes, width, height, mimetype, b64_video)
1259 return output
1264 return output
1260
1265
1261 def reload(self):
1266 def reload(self):
1262 # TODO
1267 # TODO
1263 pass
1268 pass
1264
1269
1265
1270
1266 @skip_doctest
1271 @skip_doctest
1267 def set_matplotlib_formats(*formats, **kwargs):
1272 def set_matplotlib_formats(*formats, **kwargs):
1268 """
1273 """
1269 .. deprecated:: 7.23
1274 .. deprecated:: 7.23
1270
1275
1271 use `matplotlib_inline.backend_inline.set_matplotlib_formats()`
1276 use `matplotlib_inline.backend_inline.set_matplotlib_formats()`
1272
1277
1273 Select figure formats for the inline backend. Optionally pass quality for JPEG.
1278 Select figure formats for the inline backend. Optionally pass quality for JPEG.
1274
1279
1275 For example, this enables PNG and JPEG output with a JPEG quality of 90%::
1280 For example, this enables PNG and JPEG output with a JPEG quality of 90%::
1276
1281
1277 In [1]: set_matplotlib_formats('png', 'jpeg', quality=90)
1282 In [1]: set_matplotlib_formats('png', 'jpeg', quality=90)
1278
1283
1279 To set this in your config files use the following::
1284 To set this in your config files use the following::
1280
1285
1281 c.InlineBackend.figure_formats = {'png', 'jpeg'}
1286 c.InlineBackend.figure_formats = {'png', 'jpeg'}
1282 c.InlineBackend.print_figure_kwargs.update({'quality' : 90})
1287 c.InlineBackend.print_figure_kwargs.update({'quality' : 90})
1283
1288
1284 Parameters
1289 Parameters
1285 ----------
1290 ----------
1286 *formats : strs
1291 *formats : strs
1287 One or more figure formats to enable: 'png', 'retina', 'jpeg', 'svg', 'pdf'.
1292 One or more figure formats to enable: 'png', 'retina', 'jpeg', 'svg', 'pdf'.
1288 **kwargs
1293 **kwargs
1289 Keyword args will be relayed to ``figure.canvas.print_figure``.
1294 Keyword args will be relayed to ``figure.canvas.print_figure``.
1290 """
1295 """
1291 warnings.warn(
1296 warnings.warn(
1292 "`set_matplotlib_formats` is deprecated since IPython 7.23, directly "
1297 "`set_matplotlib_formats` is deprecated since IPython 7.23, directly "
1293 "use `matplotlib_inline.backend_inline.set_matplotlib_formats()`",
1298 "use `matplotlib_inline.backend_inline.set_matplotlib_formats()`",
1294 DeprecationWarning,
1299 DeprecationWarning,
1295 stacklevel=2,
1300 stacklevel=2,
1296 )
1301 )
1297
1302
1298 from matplotlib_inline.backend_inline import (
1303 from matplotlib_inline.backend_inline import (
1299 set_matplotlib_formats as set_matplotlib_formats_orig,
1304 set_matplotlib_formats as set_matplotlib_formats_orig,
1300 )
1305 )
1301
1306
1302 set_matplotlib_formats_orig(*formats, **kwargs)
1307 set_matplotlib_formats_orig(*formats, **kwargs)
1303
1308
1304 @skip_doctest
1309 @skip_doctest
1305 def set_matplotlib_close(close=True):
1310 def set_matplotlib_close(close=True):
1306 """
1311 """
1307 .. deprecated:: 7.23
1312 .. deprecated:: 7.23
1308
1313
1309 use `matplotlib_inline.backend_inline.set_matplotlib_close()`
1314 use `matplotlib_inline.backend_inline.set_matplotlib_close()`
1310
1315
1311 Set whether the inline backend closes all figures automatically or not.
1316 Set whether the inline backend closes all figures automatically or not.
1312
1317
1313 By default, the inline backend used in the IPython Notebook will close all
1318 By default, the inline backend used in the IPython Notebook will close all
1314 matplotlib figures automatically after each cell is run. This means that
1319 matplotlib figures automatically after each cell is run. This means that
1315 plots in different cells won't interfere. Sometimes, you may want to make
1320 plots in different cells won't interfere. Sometimes, you may want to make
1316 a plot in one cell and then refine it in later cells. This can be accomplished
1321 a plot in one cell and then refine it in later cells. This can be accomplished
1317 by::
1322 by::
1318
1323
1319 In [1]: set_matplotlib_close(False)
1324 In [1]: set_matplotlib_close(False)
1320
1325
1321 To set this in your config files use the following::
1326 To set this in your config files use the following::
1322
1327
1323 c.InlineBackend.close_figures = False
1328 c.InlineBackend.close_figures = False
1324
1329
1325 Parameters
1330 Parameters
1326 ----------
1331 ----------
1327 close : bool
1332 close : bool
1328 Should all matplotlib figures be automatically closed after each cell is
1333 Should all matplotlib figures be automatically closed after each cell is
1329 run?
1334 run?
1330 """
1335 """
1331 warnings.warn(
1336 warnings.warn(
1332 "`set_matplotlib_close` is deprecated since IPython 7.23, directly "
1337 "`set_matplotlib_close` is deprecated since IPython 7.23, directly "
1333 "use `matplotlib_inline.backend_inline.set_matplotlib_close()`",
1338 "use `matplotlib_inline.backend_inline.set_matplotlib_close()`",
1334 DeprecationWarning,
1339 DeprecationWarning,
1335 stacklevel=2,
1340 stacklevel=2,
1336 )
1341 )
1337
1342
1338 from matplotlib_inline.backend_inline import (
1343 from matplotlib_inline.backend_inline import (
1339 set_matplotlib_close as set_matplotlib_close_orig,
1344 set_matplotlib_close as set_matplotlib_close_orig,
1340 )
1345 )
1341
1346
1342 set_matplotlib_close_orig(close)
1347 set_matplotlib_close_orig(close)
General Comments 0
You need to be logged in to leave comments. Login now