edit code rule2 in pythainlp.util

run python 2 and python 3

Author: wannaphong

pythainlp/util/__init__.py

@@ -46,23 +46,25 @@ def trigram(token):
     u"๋",
     u"ำ"
 ] # เก็บพวกสระ วรรณยุกต์ที่ซ้ำกันแล้วมีปัญหา
-rule2=[
-    (u"เเ",u"แ"), # เ เ -> แ
-    (u"ํ(t)า",u"\\1ำ"),
-    (u"ํา(t)",u"\\1ำ"),
-    (u"([่-๋])([ัิ-ื])",u"\\2\\1"),
-    (u"([่-๋])([ูุ])", u"\\2\\1"),
-    (u"ำ([่-๋])", u"\\1ำ"),
-    (u"(์)([ัิ-ื])",u"\\2\\1")]
-rule2py2=[
-    (u"เเ",u"แ"),
-    (u"ํ(t)า",u"\1ำ"),
-    (u"ํา(t)",u"\1ำ"),
-    (u"([่-๋])([ัิ-ื])",u"\2\1"),
-    (u"([่-๋])([ูุ])", u"\2\1"),
-    (u"ำ([่-๋])", u"\1ำ"),
-    (u"(์)([ัิ-ื])",u"\2\1")
-] # เก็บพวก พิมพ์ลำดับผิดหรือผิดแป้นแต่กลับแสดงผลถูกต้อง ให้ไปเป็นแป้นที่ถูกต้อง เช่น เ + เ ไปเป็น แ
+if six.PY2:
+	rule2=[
+		(u"เเ",u"แ"),
+		(u"ํ(t)า",u"\1ำ"),
+		(u"ํา(t)",u"\1ำ"),
+		(u"([่-๋])([ัิ-ื])",u"\2\1"),
+		(u"([่-๋])([ูุ])", u"\2\1"),
+		(u"ำ([่-๋])", u"\1ำ"),
+		(u"(์)([ัิ-ื])",u"\2\1")
+	] # เก็บพวก พิมพ์ลำดับผิดหรือผิดแป้นแต่กลับแสดงผลถูกต้อง ให้ไปเป็นแป้นที่ถูกต้อง เช่น เ + เ ไปเป็น แ
+else:
+	rule2=[
+		(u"เเ",u"แ"), # เ เ -> แ
+		(u"ํ(t)า",u"\\1ำ"),
+		(u"ํา(t)",u"\\1ำ"),
+		(u"([่-๋])([ัิ-ื])",u"\\2\\1"),
+		(u"([่-๋])([ูุ])", u"\\2\\1"),
+		(u"ำ([่-๋])", u"\\1ำ"),
+		(u"(์)([ัิ-ื])",u"\\2\\1")]
 def normalize(text):
     """
     จัดการกับข้อความภาษาไทยให้เป็นปกติ
@@ -73,11 +75,11 @@ def normalize(text):
     True
     """
     if six.PY2:
-        for data in rule2py2:
+        for data in rule2:
             text=re.sub(data[0].replace(u"t",u"[่้๊๋]"),data[1],text,re.U)
     else:
         for data in rule2:
             text=re.sub(data[0].replace("t","[่้๊๋]"),data[1],text,re.U)
     for data in list(zip(rule1,rule1)):
         text=re.sub(data[0].replace(u"t",u"[่้๊๋]")+"+",data[1],text,re.U)
-    return text
+    return text