##// END OF EJS Templates
Added missing quote in iframe height property.
Cyrille Rossant -
Show More
@@ -1,533 +1,533 b''
1 """Various display related classes.
1 """Various display related classes.
2
2
3 Authors : MinRK, gregcaporaso, dannystaple
3 Authors : MinRK, gregcaporaso, dannystaple
4 """
4 """
5 from os.path import exists, isfile, splitext, abspath, join, isdir
5 from os.path import exists, isfile, splitext, abspath, join, isdir
6 from os import walk, sep
6 from os import walk, sep
7
7
8 from IPython.core.display import DisplayObject
8 from IPython.core.display import DisplayObject
9
9
10 __all__ = ['Audio', 'IFrame', 'YouTubeVideo', 'VimeoVideo', 'ScribdDocument',
10 __all__ = ['Audio', 'IFrame', 'YouTubeVideo', 'VimeoVideo', 'ScribdDocument',
11 'FileLink', 'FileLinks']
11 'FileLink', 'FileLinks']
12
12
13
13
14 class Audio(DisplayObject):
14 class Audio(DisplayObject):
15 """Create an audio object.
15 """Create an audio object.
16
16
17 When this object is returned by an input cell or passed to the
17 When this object is returned by an input cell or passed to the
18 display function, it will result in Audio controls being displayed
18 display function, it will result in Audio controls being displayed
19 in the frontend (only works in the notebook).
19 in the frontend (only works in the notebook).
20
20
21 Parameters
21 Parameters
22 ----------
22 ----------
23 data : numpy array, list, unicode, str or bytes
23 data : numpy array, list, unicode, str or bytes
24 Can be one of
24 Can be one of
25
25
26 * Numpy 1d array containing the desired waveform (mono)
26 * Numpy 1d array containing the desired waveform (mono)
27 * Numpy 2d array containing waveforms for each channel.
27 * Numpy 2d array containing waveforms for each channel.
28 Shape=(NCHAN, NSAMPLES). For the standard channel order, see
28 Shape=(NCHAN, NSAMPLES). For the standard channel order, see
29 http://msdn.microsoft.com/en-us/library/windows/hardware/dn653308(v=vs.85).aspx
29 http://msdn.microsoft.com/en-us/library/windows/hardware/dn653308(v=vs.85).aspx
30 * List of float or integer representing the waveform (mono)
30 * List of float or integer representing the waveform (mono)
31 * String containing the filename
31 * String containing the filename
32 * Bytestring containing raw PCM data or
32 * Bytestring containing raw PCM data or
33 * URL pointing to a file on the web.
33 * URL pointing to a file on the web.
34
34
35 If the array option is used the waveform will be normalized.
35 If the array option is used the waveform will be normalized.
36
36
37 If a filename or url is used the format support will be browser
37 If a filename or url is used the format support will be browser
38 dependent.
38 dependent.
39 url : unicode
39 url : unicode
40 A URL to download the data from.
40 A URL to download the data from.
41 filename : unicode
41 filename : unicode
42 Path to a local file to load the data from.
42 Path to a local file to load the data from.
43 embed : boolean
43 embed : boolean
44 Should the image data be embedded using a data URI (True) or should
44 Should the image data be embedded using a data URI (True) or should
45 the original source be referenced. Set this to True if you want the
45 the original source be referenced. Set this to True if you want the
46 audio to playable later with no internet connection in the notebook.
46 audio to playable later with no internet connection in the notebook.
47
47
48 Default is `True`, unless the keyword argument `url` is set, then
48 Default is `True`, unless the keyword argument `url` is set, then
49 default value is `False`.
49 default value is `False`.
50 rate : integer
50 rate : integer
51 The sampling rate of the raw data.
51 The sampling rate of the raw data.
52 Only required when data parameter is being used as an array
52 Only required when data parameter is being used as an array
53 autoplay : bool
53 autoplay : bool
54 Set to True if the audio should immediately start playing.
54 Set to True if the audio should immediately start playing.
55 Default is `False`.
55 Default is `False`.
56
56
57 Examples
57 Examples
58 --------
58 --------
59 ::
59 ::
60
60
61 # Generate a sound
61 # Generate a sound
62 import numpy as np
62 import numpy as np
63 framerate = 44100
63 framerate = 44100
64 t = np.linspace(0,5,framerate*5)
64 t = np.linspace(0,5,framerate*5)
65 data = np.sin(2*np.pi*220*t) + np.sin(2*np.pi*224*t))
65 data = np.sin(2*np.pi*220*t) + np.sin(2*np.pi*224*t))
66 Audio(data,rate=framerate)
66 Audio(data,rate=framerate)
67
67
68 # Can also do stereo or more channels
68 # Can also do stereo or more channels
69 dataleft = np.sin(2*np.pi*220*t)
69 dataleft = np.sin(2*np.pi*220*t)
70 dataright = np.sin(2*np.pi*224*t)
70 dataright = np.sin(2*np.pi*224*t)
71 Audio([dataleft, dataright],rate=framerate)
71 Audio([dataleft, dataright],rate=framerate)
72
72
73 Audio("http://www.nch.com.au/acm/8k16bitpcm.wav") # From URL
73 Audio("http://www.nch.com.au/acm/8k16bitpcm.wav") # From URL
74 Audio(url="http://www.w3schools.com/html/horse.ogg")
74 Audio(url="http://www.w3schools.com/html/horse.ogg")
75
75
76 Audio('/path/to/sound.wav') # From file
76 Audio('/path/to/sound.wav') # From file
77 Audio(filename='/path/to/sound.ogg')
77 Audio(filename='/path/to/sound.ogg')
78
78
79 Audio(b'RAW_WAV_DATA..) # From bytes
79 Audio(b'RAW_WAV_DATA..) # From bytes
80 Audio(data=b'RAW_WAV_DATA..)
80 Audio(data=b'RAW_WAV_DATA..)
81
81
82 """
82 """
83 _read_flags = 'rb'
83 _read_flags = 'rb'
84
84
85 def __init__(self, data=None, filename=None, url=None, embed=None, rate=None, autoplay=False):
85 def __init__(self, data=None, filename=None, url=None, embed=None, rate=None, autoplay=False):
86 if filename is None and url is None and data is None:
86 if filename is None and url is None and data is None:
87 raise ValueError("No image data found. Expecting filename, url, or data.")
87 raise ValueError("No image data found. Expecting filename, url, or data.")
88 if embed is False and url is None:
88 if embed is False and url is None:
89 raise ValueError("No url found. Expecting url when embed=False")
89 raise ValueError("No url found. Expecting url when embed=False")
90
90
91 if url is not None and embed is not True:
91 if url is not None and embed is not True:
92 self.embed = False
92 self.embed = False
93 else:
93 else:
94 self.embed = True
94 self.embed = True
95 self.autoplay = autoplay
95 self.autoplay = autoplay
96 super(Audio, self).__init__(data=data, url=url, filename=filename)
96 super(Audio, self).__init__(data=data, url=url, filename=filename)
97
97
98 if self.data is not None and not isinstance(self.data, bytes):
98 if self.data is not None and not isinstance(self.data, bytes):
99 self.data = self._make_wav(data,rate)
99 self.data = self._make_wav(data,rate)
100
100
101 def reload(self):
101 def reload(self):
102 """Reload the raw data from file or URL."""
102 """Reload the raw data from file or URL."""
103 import mimetypes
103 import mimetypes
104 if self.embed:
104 if self.embed:
105 super(Audio, self).reload()
105 super(Audio, self).reload()
106
106
107 if self.filename is not None:
107 if self.filename is not None:
108 self.mimetype = mimetypes.guess_type(self.filename)[0]
108 self.mimetype = mimetypes.guess_type(self.filename)[0]
109 elif self.url is not None:
109 elif self.url is not None:
110 self.mimetype = mimetypes.guess_type(self.url)[0]
110 self.mimetype = mimetypes.guess_type(self.url)[0]
111 else:
111 else:
112 self.mimetype = "audio/wav"
112 self.mimetype = "audio/wav"
113
113
114 def _make_wav(self, data, rate):
114 def _make_wav(self, data, rate):
115 """ Transform a numpy array to a PCM bytestring """
115 """ Transform a numpy array to a PCM bytestring """
116 import struct
116 import struct
117 from io import BytesIO
117 from io import BytesIO
118 import wave
118 import wave
119
119
120 try:
120 try:
121 import numpy as np
121 import numpy as np
122
122
123 data = np.array(data, dtype=float)
123 data = np.array(data, dtype=float)
124 if len(data.shape) == 1:
124 if len(data.shape) == 1:
125 nchan = 1
125 nchan = 1
126 elif len(data.shape) == 2:
126 elif len(data.shape) == 2:
127 # In wave files,channels are interleaved. E.g.,
127 # In wave files,channels are interleaved. E.g.,
128 # "L1R1L2R2..." for stereo. See
128 # "L1R1L2R2..." for stereo. See
129 # http://msdn.microsoft.com/en-us/library/windows/hardware/dn653308(v=vs.85).aspx
129 # http://msdn.microsoft.com/en-us/library/windows/hardware/dn653308(v=vs.85).aspx
130 # for channel ordering
130 # for channel ordering
131 nchan = data.shape[0]
131 nchan = data.shape[0]
132 data = data.T.ravel()
132 data = data.T.ravel()
133 else:
133 else:
134 raise ValueError('Array audio input must be a 1D or 2D array')
134 raise ValueError('Array audio input must be a 1D or 2D array')
135 scaled = np.int16(data/np.max(np.abs(data))*32767).tolist()
135 scaled = np.int16(data/np.max(np.abs(data))*32767).tolist()
136 except ImportError:
136 except ImportError:
137 # check that it is a "1D" list
137 # check that it is a "1D" list
138 idata = iter(data) # fails if not an iterable
138 idata = iter(data) # fails if not an iterable
139 try:
139 try:
140 iter(idata.next())
140 iter(idata.next())
141 raise TypeError('Only lists of mono audio are '
141 raise TypeError('Only lists of mono audio are '
142 'supported if numpy is not installed')
142 'supported if numpy is not installed')
143 except TypeError:
143 except TypeError:
144 # this means it's not a nested list, which is what we want
144 # this means it's not a nested list, which is what we want
145 pass
145 pass
146 maxabsvalue = float(max([abs(x) for x in data]))
146 maxabsvalue = float(max([abs(x) for x in data]))
147 scaled = [int(x/maxabsvalue*32767) for x in data]
147 scaled = [int(x/maxabsvalue*32767) for x in data]
148 nchan = 1
148 nchan = 1
149
149
150 fp = BytesIO()
150 fp = BytesIO()
151 waveobj = wave.open(fp,mode='wb')
151 waveobj = wave.open(fp,mode='wb')
152 waveobj.setnchannels(nchan)
152 waveobj.setnchannels(nchan)
153 waveobj.setframerate(rate)
153 waveobj.setframerate(rate)
154 waveobj.setsampwidth(2)
154 waveobj.setsampwidth(2)
155 waveobj.setcomptype('NONE','NONE')
155 waveobj.setcomptype('NONE','NONE')
156 waveobj.writeframes(b''.join([struct.pack('<h',x) for x in scaled]))
156 waveobj.writeframes(b''.join([struct.pack('<h',x) for x in scaled]))
157 val = fp.getvalue()
157 val = fp.getvalue()
158 waveobj.close()
158 waveobj.close()
159
159
160 return val
160 return val
161
161
162 def _data_and_metadata(self):
162 def _data_and_metadata(self):
163 """shortcut for returning metadata with url information, if defined"""
163 """shortcut for returning metadata with url information, if defined"""
164 md = {}
164 md = {}
165 if self.url:
165 if self.url:
166 md['url'] = self.url
166 md['url'] = self.url
167 if md:
167 if md:
168 return self.data, md
168 return self.data, md
169 else:
169 else:
170 return self.data
170 return self.data
171
171
172 def _repr_html_(self):
172 def _repr_html_(self):
173 src = """
173 src = """
174 <audio controls="controls" {autoplay}>
174 <audio controls="controls" {autoplay}>
175 <source src="{src}" type="{type}" />
175 <source src="{src}" type="{type}" />
176 Your browser does not support the audio element.
176 Your browser does not support the audio element.
177 </audio>
177 </audio>
178 """
178 """
179 return src.format(src=self.src_attr(),type=self.mimetype, autoplay=self.autoplay_attr())
179 return src.format(src=self.src_attr(),type=self.mimetype, autoplay=self.autoplay_attr())
180
180
181 def src_attr(self):
181 def src_attr(self):
182 import base64
182 import base64
183 if self.embed and (self.data is not None):
183 if self.embed and (self.data is not None):
184 data = base64=base64.b64encode(self.data).decode('ascii')
184 data = base64=base64.b64encode(self.data).decode('ascii')
185 return """data:{type};base64,{base64}""".format(type=self.mimetype,
185 return """data:{type};base64,{base64}""".format(type=self.mimetype,
186 base64=data)
186 base64=data)
187 elif self.url is not None:
187 elif self.url is not None:
188 return self.url
188 return self.url
189 else:
189 else:
190 return ""
190 return ""
191
191
192 def autoplay_attr(self):
192 def autoplay_attr(self):
193 if(self.autoplay):
193 if(self.autoplay):
194 return 'autoplay="autoplay"'
194 return 'autoplay="autoplay"'
195 else:
195 else:
196 return ''
196 return ''
197
197
198 class IFrame(object):
198 class IFrame(object):
199 """
199 """
200 Generic class to embed an iframe in an IPython notebook
200 Generic class to embed an iframe in an IPython notebook
201 """
201 """
202
202
203 iframe = """
203 iframe = """
204 <iframe
204 <iframe
205 width="{width}"
205 width="{width}"
206 height={height}"
206 height="{height}"
207 src="{src}{params}"
207 src="{src}{params}"
208 frameborder="0"
208 frameborder="0"
209 allowfullscreen
209 allowfullscreen
210 ></iframe>
210 ></iframe>
211 """
211 """
212
212
213 def __init__(self, src, width, height, **kwargs):
213 def __init__(self, src, width, height, **kwargs):
214 self.src = src
214 self.src = src
215 self.width = width
215 self.width = width
216 self.height = height
216 self.height = height
217 self.params = kwargs
217 self.params = kwargs
218
218
219 def _repr_html_(self):
219 def _repr_html_(self):
220 """return the embed iframe"""
220 """return the embed iframe"""
221 if self.params:
221 if self.params:
222 try:
222 try:
223 from urllib.parse import urlencode # Py 3
223 from urllib.parse import urlencode # Py 3
224 except ImportError:
224 except ImportError:
225 from urllib import urlencode
225 from urllib import urlencode
226 params = "?" + urlencode(self.params)
226 params = "?" + urlencode(self.params)
227 else:
227 else:
228 params = ""
228 params = ""
229 return self.iframe.format(src=self.src,
229 return self.iframe.format(src=self.src,
230 width=self.width,
230 width=self.width,
231 height=self.height,
231 height=self.height,
232 params=params)
232 params=params)
233
233
234 class YouTubeVideo(IFrame):
234 class YouTubeVideo(IFrame):
235 """Class for embedding a YouTube Video in an IPython session, based on its video id.
235 """Class for embedding a YouTube Video in an IPython session, based on its video id.
236
236
237 e.g. to embed the video from https://www.youtube.com/watch?v=foo , you would
237 e.g. to embed the video from https://www.youtube.com/watch?v=foo , you would
238 do::
238 do::
239
239
240 vid = YouTubeVideo("foo")
240 vid = YouTubeVideo("foo")
241 display(vid)
241 display(vid)
242
242
243 To start from 30 seconds::
243 To start from 30 seconds::
244
244
245 vid = YouTubeVideo("abc", start=30)
245 vid = YouTubeVideo("abc", start=30)
246 display(vid)
246 display(vid)
247
247
248 To calculate seconds from time as hours, minutes, seconds use
248 To calculate seconds from time as hours, minutes, seconds use
249 :class:`datetime.timedelta`::
249 :class:`datetime.timedelta`::
250
250
251 start=int(timedelta(hours=1, minutes=46, seconds=40).total_seconds())
251 start=int(timedelta(hours=1, minutes=46, seconds=40).total_seconds())
252
252
253 Other parameters can be provided as documented at
253 Other parameters can be provided as documented at
254 https://developers.google.com/youtube/player_parameters#parameter-subheader
254 https://developers.google.com/youtube/player_parameters#parameter-subheader
255 """
255 """
256
256
257 def __init__(self, id, width=400, height=300, **kwargs):
257 def __init__(self, id, width=400, height=300, **kwargs):
258 src = "https://www.youtube.com/embed/{0}".format(id)
258 src = "https://www.youtube.com/embed/{0}".format(id)
259 super(YouTubeVideo, self).__init__(src, width, height, **kwargs)
259 super(YouTubeVideo, self).__init__(src, width, height, **kwargs)
260
260
261 class VimeoVideo(IFrame):
261 class VimeoVideo(IFrame):
262 """
262 """
263 Class for embedding a Vimeo video in an IPython session, based on its video id.
263 Class for embedding a Vimeo video in an IPython session, based on its video id.
264 """
264 """
265
265
266 def __init__(self, id, width=400, height=300, **kwargs):
266 def __init__(self, id, width=400, height=300, **kwargs):
267 src="https://player.vimeo.com/video/{0}".format(id)
267 src="https://player.vimeo.com/video/{0}".format(id)
268 super(VimeoVideo, self).__init__(src, width, height, **kwargs)
268 super(VimeoVideo, self).__init__(src, width, height, **kwargs)
269
269
270 class ScribdDocument(IFrame):
270 class ScribdDocument(IFrame):
271 """
271 """
272 Class for embedding a Scribd document in an IPython session
272 Class for embedding a Scribd document in an IPython session
273
273
274 Use the start_page params to specify a starting point in the document
274 Use the start_page params to specify a starting point in the document
275 Use the view_mode params to specify display type one off scroll | slideshow | book
275 Use the view_mode params to specify display type one off scroll | slideshow | book
276
276
277 e.g to Display Wes' foundational paper about PANDAS in book mode from page 3
277 e.g to Display Wes' foundational paper about PANDAS in book mode from page 3
278
278
279 ScribdDocument(71048089, width=800, height=400, start_page=3, view_mode="book")
279 ScribdDocument(71048089, width=800, height=400, start_page=3, view_mode="book")
280 """
280 """
281
281
282 def __init__(self, id, width=400, height=300, **kwargs):
282 def __init__(self, id, width=400, height=300, **kwargs):
283 src="https://www.scribd.com/embeds/{0}/content".format(id)
283 src="https://www.scribd.com/embeds/{0}/content".format(id)
284 super(ScribdDocument, self).__init__(src, width, height, **kwargs)
284 super(ScribdDocument, self).__init__(src, width, height, **kwargs)
285
285
286 class FileLink(object):
286 class FileLink(object):
287 """Class for embedding a local file link in an IPython session, based on path
287 """Class for embedding a local file link in an IPython session, based on path
288
288
289 e.g. to embed a link that was generated in the IPython notebook as my/data.txt
289 e.g. to embed a link that was generated in the IPython notebook as my/data.txt
290
290
291 you would do::
291 you would do::
292
292
293 local_file = FileLink("my/data.txt")
293 local_file = FileLink("my/data.txt")
294 display(local_file)
294 display(local_file)
295
295
296 or in the HTML notebook, just::
296 or in the HTML notebook, just::
297
297
298 FileLink("my/data.txt")
298 FileLink("my/data.txt")
299 """
299 """
300
300
301 html_link_str = "<a href='%s' target='_blank'>%s</a>"
301 html_link_str = "<a href='%s' target='_blank'>%s</a>"
302
302
303 def __init__(self,
303 def __init__(self,
304 path,
304 path,
305 url_prefix='',
305 url_prefix='',
306 result_html_prefix='',
306 result_html_prefix='',
307 result_html_suffix='<br>'):
307 result_html_suffix='<br>'):
308 """
308 """
309 Parameters
309 Parameters
310 ----------
310 ----------
311 path : str
311 path : str
312 path to the file or directory that should be formatted
312 path to the file or directory that should be formatted
313 directory_prefix : str
313 directory_prefix : str
314 prefix to be prepended to all files to form a working link [default:
314 prefix to be prepended to all files to form a working link [default:
315 'files']
315 'files']
316 result_html_prefix : str
316 result_html_prefix : str
317 text to append to beginning to link [default: none]
317 text to append to beginning to link [default: none]
318 result_html_suffix : str
318 result_html_suffix : str
319 text to append at the end of link [default: '<br>']
319 text to append at the end of link [default: '<br>']
320 """
320 """
321 if isdir(path):
321 if isdir(path):
322 raise ValueError("Cannot display a directory using FileLink. "
322 raise ValueError("Cannot display a directory using FileLink. "
323 "Use FileLinks to display '%s'." % path)
323 "Use FileLinks to display '%s'." % path)
324 self.path = path
324 self.path = path
325 self.url_prefix = url_prefix
325 self.url_prefix = url_prefix
326 self.result_html_prefix = result_html_prefix
326 self.result_html_prefix = result_html_prefix
327 self.result_html_suffix = result_html_suffix
327 self.result_html_suffix = result_html_suffix
328
328
329 def _format_path(self):
329 def _format_path(self):
330 fp = ''.join([self.url_prefix,self.path])
330 fp = ''.join([self.url_prefix,self.path])
331 return ''.join([self.result_html_prefix,
331 return ''.join([self.result_html_prefix,
332 self.html_link_str % (fp, self.path),
332 self.html_link_str % (fp, self.path),
333 self.result_html_suffix])
333 self.result_html_suffix])
334
334
335 def _repr_html_(self):
335 def _repr_html_(self):
336 """return html link to file
336 """return html link to file
337 """
337 """
338 if not exists(self.path):
338 if not exists(self.path):
339 return ("Path (<tt>%s</tt>) doesn't exist. "
339 return ("Path (<tt>%s</tt>) doesn't exist. "
340 "It may still be in the process of "
340 "It may still be in the process of "
341 "being generated, or you may have the "
341 "being generated, or you may have the "
342 "incorrect path." % self.path)
342 "incorrect path." % self.path)
343
343
344 return self._format_path()
344 return self._format_path()
345
345
346 def __repr__(self):
346 def __repr__(self):
347 """return absolute path to file
347 """return absolute path to file
348 """
348 """
349 return abspath(self.path)
349 return abspath(self.path)
350
350
351 class FileLinks(FileLink):
351 class FileLinks(FileLink):
352 """Class for embedding local file links in an IPython session, based on path
352 """Class for embedding local file links in an IPython session, based on path
353
353
354 e.g. to embed links to files that were generated in the IPython notebook
354 e.g. to embed links to files that were generated in the IPython notebook
355 under ``my/data``, you would do::
355 under ``my/data``, you would do::
356
356
357 local_files = FileLinks("my/data")
357 local_files = FileLinks("my/data")
358 display(local_files)
358 display(local_files)
359
359
360 or in the HTML notebook, just::
360 or in the HTML notebook, just::
361
361
362 FileLinks("my/data")
362 FileLinks("my/data")
363 """
363 """
364 def __init__(self,
364 def __init__(self,
365 path,
365 path,
366 url_prefix='',
366 url_prefix='',
367 included_suffixes=None,
367 included_suffixes=None,
368 result_html_prefix='',
368 result_html_prefix='',
369 result_html_suffix='<br>',
369 result_html_suffix='<br>',
370 notebook_display_formatter=None,
370 notebook_display_formatter=None,
371 terminal_display_formatter=None):
371 terminal_display_formatter=None):
372 """
372 """
373 See :class:`FileLink` for the ``path``, ``url_prefix``,
373 See :class:`FileLink` for the ``path``, ``url_prefix``,
374 ``result_html_prefix`` and ``result_html_suffix`` parameters.
374 ``result_html_prefix`` and ``result_html_suffix`` parameters.
375
375
376 included_suffixes : list
376 included_suffixes : list
377 Filename suffixes to include when formatting output [default: include
377 Filename suffixes to include when formatting output [default: include
378 all files]
378 all files]
379
379
380 notebook_display_formatter : function
380 notebook_display_formatter : function
381 Used to format links for display in the notebook. See discussion of
381 Used to format links for display in the notebook. See discussion of
382 formatter functions below.
382 formatter functions below.
383
383
384 terminal_display_formatter : function
384 terminal_display_formatter : function
385 Used to format links for display in the terminal. See discussion of
385 Used to format links for display in the terminal. See discussion of
386 formatter functions below.
386 formatter functions below.
387
387
388 Formatter functions must be of the form::
388 Formatter functions must be of the form::
389
389
390 f(dirname, fnames, included_suffixes)
390 f(dirname, fnames, included_suffixes)
391
391
392 dirname : str
392 dirname : str
393 The name of a directory
393 The name of a directory
394 fnames : list
394 fnames : list
395 The files in that directory
395 The files in that directory
396 included_suffixes : list
396 included_suffixes : list
397 The file suffixes that should be included in the output (passing None
397 The file suffixes that should be included in the output (passing None
398 meansto include all suffixes in the output in the built-in formatters)
398 meansto include all suffixes in the output in the built-in formatters)
399
399
400 The function should return a list of lines that will be printed in the
400 The function should return a list of lines that will be printed in the
401 notebook (if passing notebook_display_formatter) or the terminal (if
401 notebook (if passing notebook_display_formatter) or the terminal (if
402 passing terminal_display_formatter). This function is iterated over for
402 passing terminal_display_formatter). This function is iterated over for
403 each directory in self.path. Default formatters are in place, can be
403 each directory in self.path. Default formatters are in place, can be
404 passed here to support alternative formatting.
404 passed here to support alternative formatting.
405
405
406 """
406 """
407 if isfile(path):
407 if isfile(path):
408 raise ValueError("Cannot display a file using FileLinks. "
408 raise ValueError("Cannot display a file using FileLinks. "
409 "Use FileLink to display '%s'." % path)
409 "Use FileLink to display '%s'." % path)
410 self.included_suffixes = included_suffixes
410 self.included_suffixes = included_suffixes
411 # remove trailing slashs for more consistent output formatting
411 # remove trailing slashs for more consistent output formatting
412 path = path.rstrip('/')
412 path = path.rstrip('/')
413
413
414 self.path = path
414 self.path = path
415 self.url_prefix = url_prefix
415 self.url_prefix = url_prefix
416 self.result_html_prefix = result_html_prefix
416 self.result_html_prefix = result_html_prefix
417 self.result_html_suffix = result_html_suffix
417 self.result_html_suffix = result_html_suffix
418
418
419 self.notebook_display_formatter = \
419 self.notebook_display_formatter = \
420 notebook_display_formatter or self._get_notebook_display_formatter()
420 notebook_display_formatter or self._get_notebook_display_formatter()
421 self.terminal_display_formatter = \
421 self.terminal_display_formatter = \
422 terminal_display_formatter or self._get_terminal_display_formatter()
422 terminal_display_formatter or self._get_terminal_display_formatter()
423
423
424 def _get_display_formatter(self,
424 def _get_display_formatter(self,
425 dirname_output_format,
425 dirname_output_format,
426 fname_output_format,
426 fname_output_format,
427 fp_format,
427 fp_format,
428 fp_cleaner=None):
428 fp_cleaner=None):
429 """ generate built-in formatter function
429 """ generate built-in formatter function
430
430
431 this is used to define both the notebook and terminal built-in
431 this is used to define both the notebook and terminal built-in
432 formatters as they only differ by some wrapper text for each entry
432 formatters as they only differ by some wrapper text for each entry
433
433
434 dirname_output_format: string to use for formatting directory
434 dirname_output_format: string to use for formatting directory
435 names, dirname will be substituted for a single "%s" which
435 names, dirname will be substituted for a single "%s" which
436 must appear in this string
436 must appear in this string
437 fname_output_format: string to use for formatting file names,
437 fname_output_format: string to use for formatting file names,
438 if a single "%s" appears in the string, fname will be substituted
438 if a single "%s" appears in the string, fname will be substituted
439 if two "%s" appear in the string, the path to fname will be
439 if two "%s" appear in the string, the path to fname will be
440 substituted for the first and fname will be substituted for the
440 substituted for the first and fname will be substituted for the
441 second
441 second
442 fp_format: string to use for formatting filepaths, must contain
442 fp_format: string to use for formatting filepaths, must contain
443 exactly two "%s" and the dirname will be subsituted for the first
443 exactly two "%s" and the dirname will be subsituted for the first
444 and fname will be substituted for the second
444 and fname will be substituted for the second
445 """
445 """
446 def f(dirname, fnames, included_suffixes=None):
446 def f(dirname, fnames, included_suffixes=None):
447 result = []
447 result = []
448 # begin by figuring out which filenames, if any,
448 # begin by figuring out which filenames, if any,
449 # are going to be displayed
449 # are going to be displayed
450 display_fnames = []
450 display_fnames = []
451 for fname in fnames:
451 for fname in fnames:
452 if (isfile(join(dirname,fname)) and
452 if (isfile(join(dirname,fname)) and
453 (included_suffixes == None or
453 (included_suffixes == None or
454 splitext(fname)[1] in included_suffixes)):
454 splitext(fname)[1] in included_suffixes)):
455 display_fnames.append(fname)
455 display_fnames.append(fname)
456
456
457 if len(display_fnames) == 0:
457 if len(display_fnames) == 0:
458 # if there are no filenames to display, don't print anything
458 # if there are no filenames to display, don't print anything
459 # (not even the directory name)
459 # (not even the directory name)
460 pass
460 pass
461 else:
461 else:
462 # otherwise print the formatted directory name followed by
462 # otherwise print the formatted directory name followed by
463 # the formatted filenames
463 # the formatted filenames
464 dirname_output_line = dirname_output_format % dirname
464 dirname_output_line = dirname_output_format % dirname
465 result.append(dirname_output_line)
465 result.append(dirname_output_line)
466 for fname in display_fnames:
466 for fname in display_fnames:
467 fp = fp_format % (dirname,fname)
467 fp = fp_format % (dirname,fname)
468 if fp_cleaner is not None:
468 if fp_cleaner is not None:
469 fp = fp_cleaner(fp)
469 fp = fp_cleaner(fp)
470 try:
470 try:
471 # output can include both a filepath and a filename...
471 # output can include both a filepath and a filename...
472 fname_output_line = fname_output_format % (fp, fname)
472 fname_output_line = fname_output_format % (fp, fname)
473 except TypeError:
473 except TypeError:
474 # ... or just a single filepath
474 # ... or just a single filepath
475 fname_output_line = fname_output_format % fname
475 fname_output_line = fname_output_format % fname
476 result.append(fname_output_line)
476 result.append(fname_output_line)
477 return result
477 return result
478 return f
478 return f
479
479
480 def _get_notebook_display_formatter(self,
480 def _get_notebook_display_formatter(self,
481 spacer="&nbsp;&nbsp;"):
481 spacer="&nbsp;&nbsp;"):
482 """ generate function to use for notebook formatting
482 """ generate function to use for notebook formatting
483 """
483 """
484 dirname_output_format = \
484 dirname_output_format = \
485 self.result_html_prefix + "%s/" + self.result_html_suffix
485 self.result_html_prefix + "%s/" + self.result_html_suffix
486 fname_output_format = \
486 fname_output_format = \
487 self.result_html_prefix + spacer + self.html_link_str + self.result_html_suffix
487 self.result_html_prefix + spacer + self.html_link_str + self.result_html_suffix
488 fp_format = self.url_prefix + '%s/%s'
488 fp_format = self.url_prefix + '%s/%s'
489 if sep == "\\":
489 if sep == "\\":
490 # Working on a platform where the path separator is "\", so
490 # Working on a platform where the path separator is "\", so
491 # must convert these to "/" for generating a URI
491 # must convert these to "/" for generating a URI
492 def fp_cleaner(fp):
492 def fp_cleaner(fp):
493 # Replace all occurences of backslash ("\") with a forward
493 # Replace all occurences of backslash ("\") with a forward
494 # slash ("/") - this is necessary on windows when a path is
494 # slash ("/") - this is necessary on windows when a path is
495 # provided as input, but we must link to a URI
495 # provided as input, but we must link to a URI
496 return fp.replace('\\','/')
496 return fp.replace('\\','/')
497 else:
497 else:
498 fp_cleaner = None
498 fp_cleaner = None
499
499
500 return self._get_display_formatter(dirname_output_format,
500 return self._get_display_formatter(dirname_output_format,
501 fname_output_format,
501 fname_output_format,
502 fp_format,
502 fp_format,
503 fp_cleaner)
503 fp_cleaner)
504
504
505 def _get_terminal_display_formatter(self,
505 def _get_terminal_display_formatter(self,
506 spacer=" "):
506 spacer=" "):
507 """ generate function to use for terminal formatting
507 """ generate function to use for terminal formatting
508 """
508 """
509 dirname_output_format = "%s/"
509 dirname_output_format = "%s/"
510 fname_output_format = spacer + "%s"
510 fname_output_format = spacer + "%s"
511 fp_format = '%s/%s'
511 fp_format = '%s/%s'
512
512
513 return self._get_display_formatter(dirname_output_format,
513 return self._get_display_formatter(dirname_output_format,
514 fname_output_format,
514 fname_output_format,
515 fp_format)
515 fp_format)
516
516
517 def _format_path(self):
517 def _format_path(self):
518 result_lines = []
518 result_lines = []
519 walked_dir = list(walk(self.path))
519 walked_dir = list(walk(self.path))
520 walked_dir.sort()
520 walked_dir.sort()
521 for dirname, subdirs, fnames in walked_dir:
521 for dirname, subdirs, fnames in walked_dir:
522 result_lines += self.notebook_display_formatter(dirname, fnames, self.included_suffixes)
522 result_lines += self.notebook_display_formatter(dirname, fnames, self.included_suffixes)
523 return '\n'.join(result_lines)
523 return '\n'.join(result_lines)
524
524
525 def __repr__(self):
525 def __repr__(self):
526 """return newline-separated absolute paths
526 """return newline-separated absolute paths
527 """
527 """
528 result_lines = []
528 result_lines = []
529 walked_dir = list(walk(self.path))
529 walked_dir = list(walk(self.path))
530 walked_dir.sort()
530 walked_dir.sort()
531 for dirname, subdirs, fnames in walked_dir:
531 for dirname, subdirs, fnames in walked_dir:
532 result_lines += self.terminal_display_formatter(dirname, fnames, self.included_suffixes)
532 result_lines += self.terminal_display_formatter(dirname, fnames, self.included_suffixes)
533 return '\n'.join(result_lines)
533 return '\n'.join(result_lines)
General Comments 0
You need to be logged in to leave comments. Login now