def get_subtitles(self):
"""Returns a dict of index -> (language, filename) for available
tracks.
"""
if not self.playbin or not self.supports_subtitles:
return {}
tracks = {}
if self.playbin.get_property("suburi") is None:
# Don't list subtitle tracks that we're getting from an SRT file
for track_index in range(self.playbin.get_property("n-text")):
track_name = self._get_subtitle_track_name(track_index)
if track_name is None:
track_name = _("Track %(tracknumber)d",
{"tracknumber": track_index})
tracks[track_index] = (track_name, None)
files = gather_subtitle_files(self.iteminfo.video_path)
external_track_id = 100
for i, mem in enumerate(files):
track_name = self._get_subtitle_file_name(mem)
if track_name is None:
track_name = _("Subtitle file %(tracknumber)d",
{"tracknumber": i})
tracks[external_track_id + i] = (track_name, mem)
return tracks
def test_subtitles_dir(self):
movie_file, sub_files = self.create_files("foo.mov", [
os.path.join("subtitles", "foo.en.srt"),
os.path.join("subtitles", "foo.fr.srt")
])
self.assertEquals(sub_files,
util.gather_subtitle_files(FilenameType(movie_file)))
def get_subtitles(self):
"""Returns a dict of index -> (language, filename) for available
tracks.
"""
if not self.playbin or not self.supports_subtitles:
return {}
tracks = {}
if self.playbin.get_property("suburi") is None:
# Don't list subtitle tracks that we're getting from an SRT file
for track_index in range(self.playbin.get_property("n-text")):
track_name = self._get_subtitle_track_name(track_index)
if track_name is None:
track_name = _("Track %(tracknumber)d",
{"tracknumber": track_index})
tracks[track_index] = (track_name, None)
files = gather_subtitle_files(self.iteminfo.video_path)
external_track_id = 100
for i, mem in enumerate(files):
track_name = self._get_subtitle_file_name(mem)
if track_name is None:
track_name = _("Subtitle file %(tracknumber)d",
{"tracknumber": i})
tracks[external_track_id + i] = (track_name, mem)
return tracks
def test_lots_of_files(self):
movie_file, sub_files = self.create_files(
"foo.mov", ["foo.en.srt", "blah.ogv", "foo.ogv"])
# weed out the non-srt files so we can test correctly
sub_files = [mem for mem in sub_files if mem.endswith(".srt")]
self.assertEquals(sub_files,
util.gather_subtitle_files(FilenameType(movie_file)))
def test_lots_of_files(self):
movie_file, sub_files = self.create_files(
"foo.mov", ["foo.en.srt", "blah.ogv", "foo.ogv"])
# weed out the non-srt files so we can test correctly
sub_files = [mem for mem in sub_files if mem.endswith(".srt")]
self.assertEquals(
sub_files, util.gather_subtitle_files(FilenameType(movie_file)))
def test_filename_possibilities(self):
movie_file, sub_files = self.create_files(
"foo.mov", ["foo.en.srt", "foo.en.sub", "foo.srt", "foo.sub"])
self.assertEquals(sub_files,
util.gather_subtitle_files(FilenameType(movie_file)))
def test_multiple_files(self):
movie_file, sub_files = self.create_files(
"foo.mov", ["foo.en.srt", "foo.fr.srt", "foo.es.srt"])
self.assertEquals(sub_files,
util.gather_subtitle_files(FilenameType(movie_file)))
def test_no_subtitle_files(self):
movie_file, sub_files = self.create_files("foo.mov")
self.assertEquals(sub_files,
util.gather_subtitle_files(FilenameType(movie_file)))
def test_no_directory(self):
# tests the case where the foofeed directory doesn't exist
movie_file = os.path.join(self.tempdir, "foofeed", "foo.mov")
self.assertEquals([],
util.gather_subtitle_files(FilenameType(movie_file)))
def test_filename_possibilities(self):
movie_file, sub_files = self.create_files(
"foo.mov", ["foo.en.srt", "foo.en.sub", "foo.srt", "foo.sub"])
self.assertEquals(
sub_files, util.gather_subtitle_files(FilenameType(movie_file)))
def test_subtitles_dir(self):
movie_file, sub_files = self.create_files(
"foo.mov", [os.path.join("subtitles", "foo.en.srt"),
os.path.join("subtitles", "foo.fr.srt")])
self.assertEquals(
sub_files, util.gather_subtitle_files(FilenameType(movie_file)))
def test_multiple_files(self):
movie_file, sub_files = self.create_files(
"foo.mov", ["foo.en.srt", "foo.fr.srt", "foo.es.srt"])
self.assertEquals(
sub_files, util.gather_subtitle_files(FilenameType(movie_file)))
def test_no_subtitle_files(self):
movie_file, sub_files = self.create_files("foo.mov")
self.assertEquals(
sub_files, util.gather_subtitle_files(FilenameType(movie_file)))
def test_no_directory(self):
# tests the case where the foofeed directory doesn't exist
movie_file = os.path.join(self.tempdir, "foofeed", "foo.mov")
self.assertEquals(
[], util.gather_subtitle_files(FilenameType(movie_file)))