diff --git a/scrapeit/imdb.py b/scrapeit/imdb.py index b4fb0dc..80b3809 100644 --- a/scrapeit/imdb.py +++ b/scrapeit/imdb.py @@ -188,7 +188,14 @@ class IMDb: title = title[1:-1] elif title.startswith('"') and title.find('"',1) > 0 and \ title.find('"',1) == title.rfind('"'): - title = title[1:title.rfind('"')] + ':' + title[title.rfind('"')+1:] + se = re.compile("Season (\d*), Episode (\d*)\)").findall(data) + if se: + se = se[0] + print se + se = ' (S%02dE%02d)' % (int(se[0]), int(se[1])) + title = title[1:title.rfind('"')] + se + title[title.rfind('"')+1:] + else: + title = title[1:title.rfind('"')] + ':' + title[title.rfind('"')+1:] return title def parseYear(self):