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):
try:
if type(original) == str:
try:
from couchpotato.environment import Env
return original.decode(Env.get('encoding'))
except UnicodeDecodeError:
raise
elif type(original) is unicode:
if type(original) is unicode:
return original
else:
return unicode(original, *args)
@@ -35,6 +29,15 @@ def toUnicode(original, *args):
ascii_text = str(original).encode('string_escape')
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):
try:
+2 -2
View File
@@ -1,6 +1,6 @@
from couchpotato import get_session
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.logger import CPLog
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
if len(files) == 0:
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:
files.append(os.path.join(root, filename))