fix unicode, not str error in utils.py

taishi-i · Jun 19, 2020 · edf3427 · edf3427
1 parent af900a3
commit edf3427
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/nagisa/utils.py b/nagisa/utils.py
@@ -12,7 +12,10 @@
 from six.moves import cPickle
 
 
-if sys.version_info.major == 2:
+if sys.version_info.major == 3:
+    PY_3 = True
+else:
+    PY_3 = False
     reload(sys)
     sys.setdefaultencoding('utf-8')
 
@@ -31,7 +34,7 @@
 
 
 def utf8rstrip(text):
-    if type(text) != str:
+    if PY_3 is False:
         return unicode(text.rstrip(), 'utf-8')
     else:
         return text.rstrip()
@@ -49,7 +52,7 @@ def preprocess(text):
 
 
 def preprocess_without_rstrip(text):
-    if type(text) != str:
+    if PY_3 is False:
         text = unicode(text, 'utf-8')
     text = normalize(text)
     text = text.replace(' ', '　')