From 7689a1e95aa3e8005601ebd921c264776f992fa8 Mon Sep 17 00:00:00 2001 From: Hugo Date: Mon, 1 Jun 2020 14:52:29 +0300 Subject: [PATCH] User.get_now_playing: Add album and cover image to info --- src/pylast/__init__.py | 3 ++- tests/test_network.py | 4 ++++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/src/pylast/__init__.py b/src/pylast/__init__.py index 5ddab60..4befdb7 100644 --- a/src/pylast/__init__.py +++ b/src/pylast/__init__.py @@ -2338,8 +2338,9 @@ class User(_BaseObject, _Chartable): artist = _extract(e, "artist") title = _extract(e, "name") + info = {"album": _extract(e, "album"), "image": _extract_all(e, "image")} - return Track(artist, title, self.network, self.name) + return Track(artist, title, self.network, self.name, info=info) def get_recent_tracks(self, limit=10, cacheable=True, time_from=None, time_to=None): """ diff --git a/tests/test_network.py b/tests/test_network.py index 36188ec..95587e9 100755 --- a/tests/test_network.py +++ b/tests/test_network.py @@ -47,6 +47,10 @@ class TestPyLastNetwork(TestPyLastWithLastFm): self.assertIsNotNone(current_track) self.assertEqual(str(current_track.title).lower(), "test title") self.assertEqual(str(current_track.artist).lower(), "test artist") + self.assertEqual(current_track.info["album"], "Test Album") + + self.assertTrue(len(current_track.info["image"])) + self.assertRegex(current_track.info["image"][pylast.SIZE_LARGE], r"^http.+$") def test_enable_rate_limiting(self): # Arrange