fix: #45 fetch all videos of a playlist

Author: iamtalhaasghar

mps_youtube/commands/local_playlist.py

@@ -104,7 +104,7 @@ def save_last():
 
         # save using artist name in postion 1
         if g.model:
-            saveas = g.pafy_pls[g.selected_pafy_pls_id][0]['info']['title']
+            saveas = g.pafy_pls[g.selected_pafy_pls_id][0].info['info']['title']
             saveas = re.sub(r"[^-\w]", "-", saveas, flags=re.UNICODE)
 
         # loop to find next available name
@@ -153,7 +153,7 @@ def open_save_view(action, name):
             g.content = content.generate_songlist_display()
 
         else:
-            g.userpl[name] = Playlist(name, [Video(i['id'], i['title'], parse_video_length(i['duration'])) for i in g.pafy_pls[g.selected_pafy_pls_id][0]['videos']])
+            g.userpl[name] = Playlist(name, [Video(i['id'], i['title'], parse_video_length(i['duration'])) for i in g.pafy_pls[g.selected_pafy_pls_id][0].videos])
             g.message = util.F('pl saved') % name
             playlists.save()
             g.content = content.generate_songlist_display()

mps_youtube/commands/songlist.py

@@ -84,15 +84,15 @@ def plist(parturl):
     else:
         util.dbg("%sFetching playlist using pafy%s", c.y, c.w)
         ytpl = pafy.get_playlist(parturl)
-        plitems = util.IterSlicer(ytpl['videos'])
+        plitems = util.IterSlicer(ytpl.videos)
         g.pafy_pls[parturl] = (ytpl, plitems)
 
     def pl_seg(s, e):
         return [Video(i['id'], i['title'], util.parse_video_length(i['duration'])) for i in plitems[s:e]]
 
-    msg = "Showing YouTube playlist %s" % (c.y + ytpl['info']['title'] + c.w)
+    msg = "Showing YouTube playlist %s" % (c.y + ytpl.info['info']['title'] + c.w)
     loadmsg = "Retrieving YouTube playlist"
-    paginatesongs(pl_seg, length=len(ytpl['videos']), msg=msg, loadmsg=loadmsg)
+    paginatesongs(pl_seg, length=len(ytpl.videos), msg=msg, loadmsg=loadmsg)
 
 
 @command(r'(rm|add)\s*(-?\d[-,\d\s]{,250})', 'rm', 'add')

mps_youtube/pafy.py

@@ -90,7 +90,9 @@ def get_playlist(playlist_id):
     Get all videos of a playlist identified by playlist_id
     '''
 
-    playlist = Playlist.get('https://www.youtube.com/playlist?list=%s' % playlist_id)
+    playlist = Playlist('https://www.youtube.com/playlist?list=%s' % playlist_id)
+    while playlist.hasMoreVideos:
+        playlist.getNextVideos()
     return playlist
 
 def get_video_title_suggestions(query):