import sys, time

sys.path.insert(0, "./LunaTranslator")
from translator.baidu_ai import TS


class TS1(TS):

    @property
    def srclang(self):
        return "zh"

    @property
    def tgtlang(self):
        return self.xxxxx

    @tgtlang.setter
    def tgtlang(self, _):
        self.xxxxx = _


if __name__ == "__main__":

    import os, json

    f = "zh.json"
    with open("./files/lang/" + f, "r", encoding="utf8") as ff:
        js = ff.read()
        js = json.loads(js)

    xxx = {
        "ru": "ru",
        "en": "en",
        "es": "spa",
        "ko": "kor",
        "fr": "fra",
        "cht": "cht",
        "ja": "jp",
        "vi": "vie",
        "tr": "tr",
        "pl": "pl",
        "uk": "ukr",
        "it": "it",
        "ar": "ara",
        "th": "th",
        "de": "de",
        "sv": "swe",
        "nl": "nl",
        "cs": "cs",
        "pt": "pt",
    }

    needpop = []
    for k in js:
        kk = False
        try:
            k.encode("ascii")
            print(k)
            kk = True
        except:
            pass
        if k not in js or kk:
            needpop.append(k)
    for k in needpop:
        js.pop(k)
    with open(f"./files/lang/" + f, "w", encoding="utf8") as ff:
        ff.write(json.dumps(js, ensure_ascii=False, sort_keys=False, indent=4))
    a = TS1("baiduapi")
    for kk in xxx:
        with open(f"./files/lang/{kk}.json", "r", encoding="utf8") as ff:

            jsen = json.loads(ff.read())

        a.tgtlang = xxx[kk]

        needpop = []
        for k in jsen:
            if k not in js:
                needpop.append(k)
        print(kk, needpop)
        for k in needpop:
            jsen.pop(k)
        with open(f"./files/lang/{kk}.json", "w", encoding="utf8") as ff:
            ff.write(json.dumps(jsen, ensure_ascii=False, sort_keys=False, indent=4))
        for k in js:

            if k not in jsen or jsen[k] == "":
                jsen[k] = list(a.translate(k))[0]
                print(k, jsen[k])
                with open(f"./files/lang/{kk}.json", "w", encoding="utf8") as ff:
                    ff.write(
                        json.dumps(jsen, ensure_ascii=False, sort_keys=False, indent=4)
                    )