# -*- coding: utf-8 -*-

import os

import os.path

import re

import sys

reload(sys)

sys.setdefaultencoding( "utf-8" )

root_path = os.getcwd()+ os.sep

list_name = ""

str_list = []

# full path

def check_file(file):

    dic = os.path.splitext(file)

    file_n = os.path.split(file)

    if dic[1] == ".lua" and file_n[1] != "Language.lua":

        return True

    return False

def is_file(path):

    return os.path.isfile(path)

def replace(match):

    if re.search(u'[\u4e00-\u9fa5]+', match.group(0)) != None:

        global list_name

        if match.group(0) in str_list:

            return "Util" + list_name+".str"+str(str_list.index(match.group(0)))

        else:

            str_list.append(match.group(0))

            return "Util" + list_name+".str"+str(len(str_list)-1)

    else:

        return match.group(0)

def replace_china(path):

    lua_file = open(path)

    file_content = lua_file.read()

    # lua中tabel的名字

    global list_name, str_list

    str_list = []

    list_name = os.path.split(path)[1][:-4].upper()

    try:

        utf_content = file_content.decode("utf8")

    except:

        print path

        return

    re_str=u'"(.*?)"'

    pattern = re.compile(re_str)

    results =  pattern.sub(replace, utf_content)

    if results != utf_content:

        #写入文件

        write_file(path, results)

        create_tabel(path)

def write_file(path, replace_str):

    lua_file = open(path, "w")

    lua_file.write(replace_str)

    lua_file.close()

# 创建lua中的tabel

def create_tabel(path):

    global  list_name

    lua_tabel = open("common\utils\Language.lua", "a")

    lua_tabel.write("\n")

    lua_tabel.write("Util" + list_name + " = {\n")

    for index, china_str in enumerate(str_list):

        lua_tabel.write("    str" + str(index) + " = " + china_str + ",\n")

    lua_tabel.write("}")

    lua_tabel.close()

def walk_dir():

    for root, dir_names, file_names in os.walk(root_path):

        for file_name in file_names:

            full_path = os.path.join(root, file_name)

            if is_file(full_path):

                if check_file(full_path):

                    replace_china(full_path)

            else:

                walk_dir(full_path)

if __name__ == "__main__":

    walk_dir()

    # replace_china(sys.argv[1])