To a proper random string when no IMDB is found

This commit is contained in:
Ruud
2012-05-19 01:01:09 +02:00
parent 082af9a307
commit 0b034eb0fa
2 changed files with 10 additions and 2 deletions
+5
View File
@@ -2,7 +2,9 @@ from couchpotato.core.logger import CPLog
import hashlib
import os.path
import platform
import random
import re
import string
log = CPLog(__name__)
@@ -117,3 +119,6 @@ def getTitle(library_dict):
log.error('Could not get title for library item: %s' % library_dict)
return None
def randomString(size = 8, chars = string.ascii_uppercase + string.digits):
return ''.join(random.choice(chars) for x in range(size))
@@ -1,6 +1,6 @@
from couchpotato import get_session
from couchpotato.core.event import addEvent, fireEvent
from couchpotato.core.helpers.variable import mergeDicts
from couchpotato.core.helpers.variable import mergeDicts, randomString
from couchpotato.core.logger import CPLog
from couchpotato.core.plugins.base import Plugin
from couchpotato.core.settings.model import Library
@@ -23,7 +23,10 @@ class MovieResultModifier(Plugin):
# Combine on imdb id
for item in results:
imdb = item.get('imdb', 'random-%s' % time.time())
random_string = randomString()
imdb = item.get('imdb', random_string)
imdb = imdb if imdb else random_string
if not temp.get(imdb):
temp[imdb] = self.getLibraryTags(imdb)
order.append(imdb)