m3u: Add playlists.get_items()

This commit is contained in:
Stein Magnus Jodal 2015-03-23 00:28:49 +01:00
parent 4bae9c874c
commit e3f2e368c7
2 changed files with 23 additions and 0 deletions

View File

@ -28,6 +28,12 @@ class M3UPlaylistsProvider(backend.PlaylistsProvider):
for pl in self._playlists.values()]
return sorted(refs, key=operator.attrgetter('name'))
def get_items(self, uri):
playlist = self._playlists.get(uri)
if playlist is None:
return None
return [Ref.track(uri=t.uri, name=t.name) for t in playlist.tracks]
@property
def playlists(self):
return sorted(

View File

@ -278,3 +278,20 @@ class M3UPlaylistsProviderTest(unittest.TestCase):
self.core.playlists.delete('m3u:c.m3u')
check_order(self.core.playlists.as_list(), ['b', 'd'])
def test_get_items_returns_item_refs(self):
track = Track(uri='dummy:a', name='A', length=60000)
playlist = self.core.playlists.create('test')
playlist = self.core.playlists.save(playlist.copy(tracks=[track]))
item_refs = self.core.playlists.get_items(playlist.uri)
self.assertEqual(len(item_refs), 1)
self.assertEqual(item_refs[0].type, 'track')
self.assertEqual(item_refs[0].uri, 'dummy:a')
self.assertEqual(item_refs[0].name, 'A')
def test_get_items_of_unknown_playlist_returns_none(self):
item_refs = self.core.playlists.get_items('dummy:unknown')
self.assertIsNone(item_refs)