Added the fix for issue #13 in the original repository.

This commit is contained in:
Wessie 2013-12-18 18:11:23 +01:00
parent 7122e110d1
commit 8a7358d426

View file

@ -15,9 +15,22 @@ def find_files(path, extensions):
yield (p, extension) yield (p, extension)
def read(filename): def read(filename, limit=None):
"""
Reads any file supported by pydub (ffmpeg) and returns the data contained
within.
Can be optionally limited to a certain amount of seconds from the start
of the file by specifying the `limit` parameter. This is the amount of
seconds from the start of the file.
returns: (samplerate, channels)
"""
audiofile = AudioSegment.from_file(filename) audiofile = AudioSegment.from_file(filename)
if limit:
audiofile = audiofile[:limit * 1000]
data = np.fromstring(audiofile._data, np.int16) data = np.fromstring(audiofile._data, np.int16)
channels = [] channels = []