Add encoding option to decode unit name

Fixed armoha/euddraft#20
armoha · Jan 27, 2021 · 962c70e · 962c70e
1 parent 6b65433
commit 962c70e
Showing 1 changed file with 14 additions and 6 deletions.
diff --git a/eudplib/core/mapdata/tblformat.py b/eudplib/core/mapdata/tblformat.py
@@ -28,6 +28,14 @@
 from ... import utils as ut
 from ...localize import _
 
+unit_name_encoding = "cp949"
+
+
+def DecodeUnitNameAs(e):
+    global unit_name_encoding
+    " ".encode(e)
+    unit_name_encoding = e
+
 
 def IgnoreColor(s):
     return bytes(filter(lambda x: not (0x01 <= x <= 0x1F or x == 0x7F), s))
@@ -203,10 +211,10 @@ def LoadTBLWithChk(self, content, init_chkt):
                     if nextstring != b"":
                         if j in unitdict:
                             try:
-                                nextstring = (nextstring.decode("cp949")).encode(
-                                    "utf-8"
-                                )
-                            except (UnicodeDecodeError):
+                                nextstring = (
+                                    nextstring.decode(unit_name_encoding)
+                                ).encode("utf-8")
+                            except UnicodeDecodeError:
                                 pass
                         self._emptystring.append((i - 1, nextstring))
                         break
@@ -216,8 +224,8 @@ def LoadTBLWithChk(self, content, init_chkt):
             if string:
                 if i in unitdict:
                     try:
-                        string = (string.decode("cp949")).encode("utf-8")
-                    except (UnicodeDecodeError):
+                        string = (string.decode(unit_name_encoding)).encode("utf-8")
+                    except UnicodeDecodeError:
                         pass
                     unitmap.AddItem(string, unitdict[i])
                     if string != IgnoreColor(string):