Some more encoding

This commit is contained in:
Ruud
2012-01-14 21:09:26 +01:00
parent 199c17e2c1
commit f4a4bca7a9
2 changed files with 12 additions and 9 deletions
+10 -7
View File
@@ -20,13 +20,7 @@ def simplifyString(original):
def toUnicode(original, *args): def toUnicode(original, *args):
try: try:
if type(original) == str: if type(original) is unicode:
try:
from couchpotato.environment import Env
return original.decode(Env.get('encoding'))
except UnicodeDecodeError:
raise
elif type(original) is unicode:
return original return original
else: else:
return unicode(original, *args) return unicode(original, *args)
@@ -35,6 +29,15 @@ def toUnicode(original, *args):
ascii_text = str(original).encode('string_escape') ascii_text = str(original).encode('string_escape')
return unicode(ascii_text) return unicode(ascii_text)
def ek(original, *args):
if type(original) in [str, unicode]:
try:
from couchpotato.environment import Env
return original.decode(Env.get('encoding'))
except UnicodeDecodeError:
return toUnicode(original, *args)
return original
def isInt(value): def isInt(value):
try: try:
+2 -2
View File
@@ -1,6 +1,6 @@
from couchpotato import get_session from couchpotato import get_session
from couchpotato.core.event import fireEvent, addEvent from couchpotato.core.event import fireEvent, addEvent
from couchpotato.core.helpers.encoding import toUnicode, simplifyString from couchpotato.core.helpers.encoding import toUnicode, simplifyString, ek
from couchpotato.core.helpers.variable import getExt, getImdb, tryInt from couchpotato.core.helpers.variable import getExt, getImdb, tryInt
from couchpotato.core.logger import CPLog from couchpotato.core.logger import CPLog
from couchpotato.core.plugins.base import Plugin from couchpotato.core.plugins.base import Plugin
@@ -145,7 +145,7 @@ class Scanner(Plugin):
# Scan all files of the folder if no files are set # Scan all files of the folder if no files are set
if len(files) == 0: if len(files) == 0:
files = [] files = []
for root, dirs, walk_files in os.walk(folder): for root, dirs, walk_files in os.walk(ek(folder)):
for filename in walk_files: for filename in walk_files:
files.append(os.path.join(root, filename)) files.append(os.path.join(root, filename))