]> git.phdru.name Git - m_lib.git/blobdiff - m_lib/rus/rus2lat.py
Build(setup.py): Fix trove classifiers
[m_lib.git] / m_lib / rus / rus2lat.py
index e7f731b1c24a7f79431ad7f9d2273db76eb6d5e6..fff0bfd4db2fd0eccd0f605165bb4258e059fde8 100755 (executable)
@@ -2,6 +2,7 @@
 # -*- coding: koi8-r -*-
 
 from __future__ import print_function
+from ..lazy.dict import LazyDictInitFunc
 
 #
 # Rus -> Lat transliteration (koi2lat and win2lat)
@@ -14,6 +15,7 @@ koi2lat_d = {
    "Г": "G",
    "Д": "D",
    "Е": "E",
+   "Ё": "Yo",
    "Ж": "Zh",
    "З": "Z",
    "И": "I",
@@ -46,6 +48,7 @@ koi2lat_d = {
    "г": "g",
    "д": "d",
    "е": "e",
+   "ё": "yo",
    "ж": "zh",
    "з": "z",
    "и": "i",
@@ -77,14 +80,10 @@ koi2lat_d = {
 def make_xxx2lat(encoding="cp1251"):
    d = {}
    for k, v in koi2lat_d.items():
-      if isinstance(k, bytes):
-         k = k.decode("koi8-r")
-      k = k.encode(encoding)
       d[k] = v
    return d
 
 
-from ..lazy.dict import LazyDictInitFunc
 win2lat_d = LazyDictInitFunc(make_xxx2lat, encoding="cp1251")
 
 
@@ -92,12 +91,10 @@ def rus2lat(instr, rus2lat_d = koi2lat_d):
    out = []
    for c in instr:
       c = rus2lat_d.get(c, c)
-      if isinstance(c, bytes):
-         c = c.decode('ascii')
-      elif isinstance(c, int):
+      if isinstance(c, int):
          c = chr(c)
-      out.append(c.encode('ascii'))
-   return b''.join(out)
+      out.append(c)
+   return ''.join(out)
 
 
 koi2lat = rus2lat
@@ -110,6 +107,4 @@ if __name__ == "__main__":
    Test = "Щербаков Игорь Григорьевич. АБВ xyz абв ЬЬЭЮЯ ъьэюя"
    print("Test:", Test)
    print("Тест:", koi2lat(Test))
-   if isinstance(Test, bytes):
-      Test = Test.decode("cp1251")
-   print("Тест:", win2lat(Test.encode("cp1251")))
+   print("Тест:", win2lat(Test))