StringsFileUtil.py

#!/usr/bin/python
# -*- coding: UTF-8 -*-

import os
from Log import Log
import codecs
import re
import io


def removeComments(s):
    for x in re.findall(r'("[^\n]*"(?!\\))|(//[^\n]*$|/(?!\\)\*[\s\S]*?\*(?!\\)/)', s, 8):
        s = s.replace(x[1], '')
    return s


class StringsFileUtil:
    'iOS Localizable.strings file util'

    @staticmethod
    def writeToFile(keys, values, directory, name, additional):
        if not os.path.exists(directory):
            os.makedirs(directory)

        Log.info("Creating iOS file:" + directory + name)

        fo = open(directory + "/" + name, "wb")

        for x in range(len(keys)):
            if values[x] is None or values[x] == '':
                Log.error("Key:" + keys[x] +
                          "\'s value is None. Index:" + str(x + 1))
                continue

            key = str(keys[x]).strip()
            value = str(values[x])
            content = "\"" + key + "\" " + "= " + "\"" + value + "\";\n"
            fo.write(content.encode())

        if additional is not None:
            fo.write(additional)

        fo.close()

    @staticmethod
    def getKeysAndValues(path):
        if path is None:
            Log.error('file path is None')
            return

        # 1.Read localizable.strings
        encodings = ['utf-8', 'utf-16']
        for e in encodings:
            try:
                file = codecs.open(path, 'r', encoding=e)
                string = file.read()
                file.close()
            except UnicodeDecodeError:
                print('got unicode error with %s , trying different encoding' % e)
            else:
                break

        # 2.Remove comments
        string = removeComments(string)

        # 3.Split by ";
        localStringList = string.split('\";')
        list = [x.split(' = ') for x in localStringList]

        # 4.Get keys & values
        keys = []
        values = []
        for x in range(len(list)):
            keyValue = list[x]
            if len(keyValue) > 1:
                key = keyValue[0].split('\"')[1]
                value = keyValue[1][1:]
                keys.append(key)
                values.append(value)

        return (keys, values)