From 5893ffd44bfccf134ede86a7b526062a172f131a Mon Sep 17 00:00:00 2001
From: Johan Andersson <j@i19.se>
Date: Sun, 28 Dec 2014 14:33:25 +0100
Subject: [PATCH] =?UTF-8?q?svtplay:=20support=20for=20=C3=B6ppetarkiv?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

this will only work with those with episodes in the name.

this fixes #140
---
 lib/svtplay_dl/service/svtplay.py | 24 +++++++++++++++---------
 1 file changed, 15 insertions(+), 9 deletions(-)

diff --git a/lib/svtplay_dl/service/svtplay.py b/lib/svtplay_dl/service/svtplay.py
index 0232659..7a5491f 100644
--- a/lib/svtplay_dl/service/svtplay.py
+++ b/lib/svtplay_dl/service/svtplay.py
@@ -148,18 +148,24 @@ def outputfilename(data, filename, raw):
     return output
 
 
-def seasoninfo(line):
-    match = re.search(r'play_video-area-aside__sub-title">([^<]+)<span', line)
+def seasoninfo(data):
+    match = re.search(r'play_video-area-aside__sub-title">([^<]+)<span', data)
     if match:
-        line = re.sub(" +", "", match.group(1)).replace('\n','')
-        match = re.search(r"(\w+(\d+)-)?Avsnitt(\d+)", line)
+        line = match.group(1)
+    else:
+        match = re.search(r'data-title="([^"]+)"', data)
         if match:
-            if match.group(2) is None:
-                season = 1
-            else:
-                season = match.group(2)
-            return "S%sE%s" % (season, match.group(3))
+            line = match.group(1)
         else:
             return None
+
+    line = re.sub(" +", "", match.group(1)).replace('\n','')
+    match = re.search(r"(song(\d+)-)?Avsnitt(\d+)", line)
+    if match:
+        if match.group(2) is None:
+            season = 1
+        else:
+            season = match.group(2)
+        return "S%sE%s" % (season, match.group(3))
     else:
         return None
\ No newline at end of file