Merge pull request #6 from rogermolas/json-support

JSON support

Author: rogermolas

README.md

@@ -27,7 +27,7 @@ csv-localizer use three required commands
 
 | Commands| Descriptions| 
 | :------: |:-------------
-| `-p` | Platforms (e.g ios or android) |
+| `-p` | Platforms (ios, android, json) |
 | `-i` | Input directory, CSV files directory path| 
 | `-0` | Output directory, Generated localizable files path| 
 

csv-localizer

@@ -1,18 +1,22 @@
 #!/usr/bin/env python
 import sys, argparse, logging, os, csv
 
+PLATFORM = None
+IN_PATH = None
+OUT_PATH = None
+LANG_KEYS = None  # static will change later
+
 # Gather our code in a main() function
 def main(args, loglevel):
   logging.basicConfig(format="%(message)s", level=loglevel)
   PLATFORM = args.platform
   IN_PATH = args.input
   OUT_PATH = args.output
-  LANG_KEYS = None  #static will change later
   print '\n'
   logging.info("Start Localizing .... ")
   print '\n'
   logging.info("------------------------------------")
-  
+
   # check source path
   logging.debug("\n")
   logging.debug("Validating source path ...")
@@ -32,7 +36,7 @@ def main(args, loglevel):
     return
   logging.debug("Valid target path, generating output directory ...")
   logging.debug("\n")
-  
+
   # generate output directory
   OUTPUT_DIR = os.path.join(OUT_PATH, "output")
   if not os.path.exists(OUTPUT_DIR):
@@ -42,23 +46,24 @@ def main(args, loglevel):
   else:
     logging.debug("Using output directory: %s" % OUTPUT_DIR)
     logging.debug("\n")
-  
+
   logging.debug("\n")
   if PLATFORM == "ios":
     logging.debug("Platform : %s" % PLATFORM)
   elif PLATFORM == "android":
     logging.debug("Platform : %s" % PLATFORM)
+  elif PLATFORM == "json":
+    logging.debug("Platform : %s" % PLATFORM)
   else:
-    logging.warn("Invalid platform, platform should be ios or android only")
+    logging.warn("Invalid platform, platform should be ios, android, json only")
     logging.debug("\n")
-    logging.error('ERROR LOCALIZING.')
+    logging.error('ERROR LOCALIZING.\n')
     return
 
-  logging.info("Generation output : %s" % OUTPUT_DIR)
+  logging.info("Generated output directory: %s" % OUTPUT_DIR)
   generate_keys(IN_PATH, OUTPUT_DIR, PLATFORM)
   print '\n'
-  logging.info("DONE LOCALIZING.")
-  print '\n'
+  logging.info("DONE LOCALIZING.\n")
 
 def generate_keys(source_path, output, platform):
   base_out_dir = output
@@ -70,7 +75,7 @@ def generate_keys(source_path, output, platform):
       filename, ext = os.path.splitext(f)
       if ext != '.csv':
         continue
-      
+
       fullpath = os.path.join(dirname, f)
 
       with open(fullpath, 'rb') as csvfile:
@@ -80,16 +85,19 @@ def generate_keys(source_path, output, platform):
           if i == 0:
             line.remove(line[0])
             LANG_KEYS = line  # assign new value to key
-            
+
             # iterate each language
             lang_path = ""
             for lang in LANG_KEYS:
               if platform == "ios":
                 lang_path = os.path.join(base_out_dir, "{0}.lproj/".format(lang))
-              
+
               if platform == "android":
                 lang_path = os.path.join(base_out_dir, "values-{0}/".format(lang))
-              
+
+              if platform == "json":
+                lang_path = os.path.join(base_out_dir, "{0}/".format(lang))
+
               # Generate directory per language key
               if not os.path.exists(lang_path):
                 os.makedirs(lang_path)
@@ -98,14 +106,17 @@ def generate_keys(source_path, output, platform):
           full_out_paths = [os.path.join(base_out_dir, "{0}.lproj/".format(langKey) + "Localizable.strings") for langKey in LANG_KEYS]
         if platform == "android":
           full_out_paths = [os.path.join(base_out_dir, "values-{0}/".format(langKey) + "strings.xml") for langKey in LANG_KEYS]
-      
+        if platform == "json":
+          full_out_paths = [os.path.join(base_out_dir, "{0}/".format(langKey) + "{0}_message.json".format(langKey)) for langKey in LANG_KEYS]
+
       allwrites = [open(out_path, 'w') for out_path in full_out_paths]
-  
+
   if platform == "ios":
     start_localize_ios(source_path, allwrites, LANG_KEYS)
   if platform == "android":
     start_localize_android(source_path, allwrites, LANG_KEYS)
-  
+  if platform == "json":
+    start_localize_json(source_path, allwrites, LANG_KEYS)
 
 
 # =========================================================================
@@ -121,15 +132,15 @@ def start_localize_ios(source_path, all_writes, lang_keys):
         continue
 
       fullpath = os.path.join(dirname, f)
-      logging.info("Localizing: %s", filename)
+      logging.info("Localizing: %s to iOS", filename)
 
       with open(fullpath, 'rb') as csvfile:
         [fwrite.write('\n/*  {0}  */\n'.format(filename)) for fwrite in allwrites]
-        
+
         reader = csv.reader(csvfile, delimiter=',')
         iterrows = iter(reader)
         next(iterrows) # skip first line (it is header).
-        
+
         for row in iterrows:
           row_key = row[0].replace(" ", "")
           # comment
@@ -141,7 +152,8 @@ def start_localize_ios(source_path, all_writes, lang_keys):
           if any([value == "" for value in row_values]):
             [fwrite.write('\n') for idx, fwrite in enumerate(allwrites)]
           else:
-            [fwrite.write('"{key}" = "{lang}";\n'.format(key=row_key, lang=row_values[idx])) for idx, fwrite in enumerate(allwrites)]
+            [fwrite.write('"{key}" = "{lang}";\n'.format(key=row_key, lang=row_values[idx]))
+             for idx, fwrite in enumerate(allwrites)]
   [fwrite.close() for fwrite in allwrites]
 
 
@@ -153,23 +165,23 @@ def start_localize_android(source_path, all_writes, lang_keys):
 
   [fwrite.write('<?xml version="1.0" encoding="utf-8"?>\n') for fwrite in allwrites]
   [fwrite.write('<resources>') for fwrite in allwrites]
-  
+
   for dirname, dirnames, filenames in os.walk(source_path):
     for f in filenames:
       filename, ext = os.path.splitext(f)
       if ext != '.csv':
         continue
-      
+
       fullpath = os.path.join(dirname, f)
-      logging.info("Localizing: %s", filename)  
+      logging.info("Localizing: %s to Android", filename)
 
       with open(fullpath, 'rb') as csvfile:
         [fwrite.write('\n<!-- {0} -->\n'.format(filename)) for fwrite in allwrites]
-        
+
         reader = csv.reader(csvfile, delimiter=',')
         iterrows = iter(reader)
         next(iterrows) # skip first line (it is header).
-        
+
         for row in iterrows:
           row_key = row[0].replace(" ", "")
           # comment
@@ -181,26 +193,75 @@ def start_localize_android(source_path, all_writes, lang_keys):
           if any([value == "" for value in row_values]):
             [fwrite.write('\n') for idx, fwrite in enumerate(allwrites)]
           else:
-            [fwrite.write('\t<string name="{key}">{lang}</string>\n'.format(key=row_key, lang=row_values[idx])) for idx, fwrite in enumerate(allwrites)]
+            [fwrite.write('\t<string name="{key}">{lang}</string>\n'.format(key=row_key, lang=row_values[idx]))
+             for idx, fwrite in enumerate(allwrites)]
         [fwrite.write('</resources>') for fwrite in allwrites]
   [fwrite.close() for fwrite in allwrites]
 
 
 # =========================================================================
-# +++++++ Standard boilerplate to call the main() function to begin +++++++
+# ++++++++++++++++++++++++++++++ iOS ++++++++++++++++++++++++++++++++++++++
 # =========================================================================
+def start_localize_json(source_path, all_writes, lang_keys):
+  allwrites = all_writes
+  [fwrite.write('{') for fwrite in allwrites]
+
+  for dirname, dirnames, filenames in os.walk(source_path):
+    for f in filenames:
+      filename, ext = os.path.splitext(f)
+      if ext != '.csv':
+        continue
+
+      fullpath = os.path.join(dirname, f)
+      logging.info("Localizing: %s to JSON", filename)
+
+      current_row = 0
+
+      with open(fullpath, 'rb') as csvfile:
+        [fwrite.write('\n/*  {0}  */\n'.format(filename)) for fwrite in allwrites]
+
+        reader = csv.reader(csvfile, delimiter=',')
+        iterrows = iter(reader)
+        next(iterrows) # skip first line (it is header).
+        list_data = list(iterrows)
+        
+        for row in list_data:
+          row_key = row[0].replace(" ", "")
+          # comment
+          if row_key[:2] == '//':
+            continue
+          row_values = [row[i+1] for i in range(len(lang_keys))]
+          separator = ", \n"
 
-parser = argparse.ArgumentParser(description = "Locatization commands")
-parser.add_argument("-p",help="Specify Platform (iOS, Android)" ,dest="platform", type=str, required=True)
-parser.add_argument("-i",help="Input source, CSV file path" ,dest="input", type=str, required=True)
-parser.add_argument("-o",help="Generated output path for localizable files" ,dest="output", type=str, required=True)
+          # check if last row then remove ';'
+          if current_row == len(list_data) - 1:
+            separator = "\n"
+          current_row += 1 # iterate row count
+
+          # if any row is empty, skip it!
+          if any([value == "" for value in row_values]):
+            [fwrite.write('\n') for idx, fwrite in enumerate(allwrites)]
+          else:
+            [fwrite.write('"{key}": "{lang}"{separator}'.format(key=row_key, lang=row_values[idx], separator=separator))
+             for idx, fwrite in enumerate(allwrites)]
+      [fwrite.write('}') for fwrite in allwrites]
+  [fwrite.close() for fwrite in allwrites]
+
+
+# =========================================================================
+# +++++++ Standard boilerplate to call the main() function to begin +++++++
+# =========================================================================
+parser = argparse.ArgumentParser(description="Locatization commands")
+parser.add_argument("-p", help="Specify Platform (iOS, Android)", dest="platform", type=str, required=True)
+parser.add_argument("-i", help="Input source, CSV file path", dest="input", type=str, required=True)
+parser.add_argument("-o", help="Generated output path for localizable files", dest="output", type=str, required=True)
 
 parser.add_argument("-v",
                     "--verbose",
                     help="increase output verbosity",
                     action="store_true")
 args = parser.parse_args()
-  
+
 # Setup logging
 if args.verbose:
   loglevel = logging.DEBUG