csv <-> rs translator

lang.py

@@ -4,6 +4,8 @@
 
 import os
 import glob
+import sys
+import csv
     
 def get_lang(lang): 
   out = {}    
@@ -23,11 +25,20 @@ def line_split(line):
 
 
 def main():
+  if len(sys.argv) == 1:
+    expand()
+  elif sys.argv[1] == '1':
+    to_csv()
+  else:
+    to_rs(sys.argv[1])
+
+
+def expand():
   for fn in glob.glob('./src/lang/*'): 
     lang = os.path.basename(fn)[:-3] 
     if lang in ['en','cn']: continue  
-    fw = open("%s.rs.gen"%lang, "wb+")
     dict = get_lang(lang)
+    fw = open("%s.rs"%lang, "wt")
     for line in open('./src/lang/cn.rs'):
       line_strip = line.strip()
       if line_strip.startswith('("'):
@@ -36,11 +47,38 @@ def main():
           line = line.replace(v, dict[k])
         else:
           line = line.replace(v, "")
-        fw.write(line.encode())
+        fw.write(line)
       else:
-        fw.write(line.encode())
+        fw.write(line)
     fw.close()
-    os.remove("./src/lang/%s.rs"%lang)
+ 
-    os.rename(fw.name, "./src/lang/%s.rs"%lang)
+
+def to_csv():
+  for fn in glob.glob('./src/lang/*.rs'): 
+    lang = os.path.basename(fn)[:-3] 
+    csvfile = open('./src/lang/%s.csv'%lang, "wt")
+    csvwriter = csv.writer(csvfile)
+    for line in open(fn):
+      line_strip = line.strip()
+      if line_strip.startswith('("'):
+        k, v = line_split(line_strip)
+        csvwriter.writerow([k, v])
+    csvfile.close()
+
+
+def to_rs(lang):
+  csvfile = open('%s.csv'%lang, "rt")
+  fw = open("./src/lang/%s.rs"%lang, "wt")
+  fw.write('''lazy_static::lazy_static! {
+pub static ref T: std::collections::HashMap<&'static str, &'static str> =
+    [
+''')
+  for row in csv.reader(csvfile):
+    fw.write('        ("%s", "%s"),\n'%(row[0].replace('"', '\"'), row[1].replace('"', '\"')))
+  fw.write('''    ].iter().cloned().collect();
+}
+''')
+  fw.close()
+
 
 main()