From ea057fc8fdd7ead26f2464b6903965a5c2e0524a Mon Sep 17 00:00:00 2001
From: Johan Andersson <j@i19.se>
Date: Sat, 27 Sep 2014 22:53:41 +0200
Subject: [PATCH] mtvnn: download the right number of episodes

this fixes #132
---
 lib/svtplay_dl/service/mtvnn.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/lib/svtplay_dl/service/mtvnn.py b/lib/svtplay_dl/service/mtvnn.py
index 32deceded..c92159a6b 100644
--- a/lib/svtplay_dl/service/mtvnn.py
+++ b/lib/svtplay_dl/service/mtvnn.py
@@ -48,11 +48,18 @@ def find_all_episodes(self, options):
             log.error("Couldn't program id")
             sys.exit(2)
         programid = match.group(1)
-        match = re.findall(r"data-item-id='([^']+)'", self.get_urldata())
+        match = re.findall(r"<li class='(divider playlist-item|playlist-item)'( data-item-id='([^']+)')?", self.get_urldata())
         if not match:
             log.error("Couldn't retrieve episode list")
             sys.exit(2)
+        episodNr = []
+        for i in match:
+            if i[0] == "playlist-item":
+                episodNr.append(i[2])
+            else:
+                break
+
         episodes = []
-        for i in sorted(match):
+        for i in sorted(episodNr):
             episodes.append("http://www.nickelodeon.se/serier/%s-something/videos/%s-something" % (programid, i))
         return episodes