| 16 | |
| 17 | |
| 18 | class RichOutput: |
| 19 | def __init__(self, data=None, metadata=None, transient=None, update=False): |
| 20 | self.data = data or {} |
| 21 | self.metadata = metadata or {} |
| 22 | self.transient = transient or {} |
| 23 | self.update = update |
| 24 | |
| 25 | def display(self): |
| 26 | from IPython.display import publish_display_data |
| 27 | publish_display_data(data=self.data, metadata=self.metadata, |
| 28 | transient=self.transient, update=self.update) |
| 29 | |
| 30 | def _repr_mime_(self, mime): |
| 31 | if mime not in self.data: |
| 32 | return |
| 33 | data = self.data[mime] |
| 34 | if mime in self.metadata: |
| 35 | return data, self.metadata[mime] |
| 36 | else: |
| 37 | return data |
| 38 | |
| 39 | def _repr_mimebundle_(self, include=None, exclude=None): |
| 40 | return self.data, self.metadata |
| 41 | |
| 42 | def _repr_html_(self): |
| 43 | return self._repr_mime_("text/html") |
| 44 | |
| 45 | def _repr_latex_(self): |
| 46 | return self._repr_mime_("text/latex") |
| 47 | |
| 48 | def _repr_json_(self): |
| 49 | return self._repr_mime_("application/json") |
| 50 | |
| 51 | def _repr_javascript_(self): |
| 52 | return self._repr_mime_("application/javascript") |
| 53 | |
| 54 | def _repr_png_(self): |
| 55 | return self._repr_mime_("image/png") |
| 56 | |
| 57 | def _repr_jpeg_(self): |
| 58 | return self._repr_mime_("image/jpeg") |
| 59 | |
| 60 | def _repr_svg_(self): |
| 61 | return self._repr_mime_("image/svg+xml") |
| 62 | |
| 63 | |
| 64 | class CapturedIO: |