json-to-nms-android-xml.py

import argparse
import json
import csv
import sys
import re

assert sys.version_info >= (3, 6), "Python >= 3.6 is required"

# Special replacement class that counts the matches
class ReplaceCounter(object):
    def __init__(self):
        self.idx = 0
    def __call__(self, match):
        self.idx += 1
        return "%{}$s".format(self.idx)

def dict_to_android_xml(d, out_path, base_d):
    with open(out_path, 'w') as out_file:
        out_file.write('<?xml version="1.0" encoding="utf-8"?>\n')
        out_file.write('<resources>\n')
        for key, text in base_d.items():
            if key in d:
                text = d[key]
            key = key.replace('.', '_')
            # We need to increase a counter for every match
            text = re.sub(
                r'\{(\w+)\}',
                ReplaceCounter(),
                text,
            )
            text = text.replace("&", "&amp;")
            text = text.replace('"', '\"')
            text = text.replace("'", "\'")
            text = text.replace("'", "\'")
            text = text.replace("\n", "\\n")
            # This regexp will match any unescaped ' and " also when it appears at
            # the beginning of the string.
            text = re.sub(r'([^\\])\'|^\'', '\g<1>\\\'', text)
            text = re.sub(r'([^\\])\"|^\"', '\g<1>\\\"', text)
            out_file.write('  <string name="{}">{}</string>'.format(key, text))
            out_file.write('\n')
        out_file.write('</resources>\n')

def load_json(in_path):
    with open(in_path) as in_file:
        return json.load(in_file)

def parse_args():
    p = argparse.ArgumentParser(description='translations: CSV to KEYVALUEJSON')
    p.add_argument('--json', metavar='PATH', help='json input path', required=True)
    p.add_argument('--xml', metavar='PATH', help='android XML output path', required=True)
    p.add_argument('--base', metavar='PATH', help='ooni base xml input path', required=True)
    opt = p.parse_args()
    return opt


def main():
    opt = parse_args()
    dict_to_android_xml(load_json(opt.json), opt.xml, load_json(opt.base))
    print("{} written".format(opt.xml))

if __name__ == "__main__":
    main()