Allow imdb id with int of 4-7

This commit is contained in:
Ruud
2013-10-14 22:05:32 +02:00
parent 0e18dcb8a1
commit 1377b6315c
+2 -2
View File
@@ -137,7 +137,7 @@ def getImdb(txt, check_inside = False, multiple = False):
output.close() output.close()
try: try:
ids = re.findall('(tt\d{7})', txt) ids = re.findall('(tt\d{4,7})', txt)
if multiple: if multiple:
return list(set(ids)) if len(ids) > 0 else [] return list(set(ids)) if len(ids) > 0 else []
return ids[0] return ids[0]
@@ -146,7 +146,7 @@ def getImdb(txt, check_inside = False, multiple = False):
return False return False
def tryInt(s, default=0): def tryInt(s, default = 0):
try: return int(s) try: return int(s)
except: return default except: return default