#!python3

import argparse
import subprocess

reservedWords = {
    "函数": "def",
    "删除": "del",
    "跳过": "pass",
    "终止": "break",
    "继续": "continue",
    "返回": "return",
    "从": "from",
    "抛出": "raise",
    "生成": "yield",
    "导入": "import",
    "全局": "global",
    "非局部": "nonlocal",
    "断言": "assert",
    "如果": "if",
    "或如": "elif",
    "否则": "else",
    "当": "while",
    "对于": "for",
    "在": "in",
    "尝试": "try",
    "捕获": "except",
    "最后": "finally",
    "作为": "as",
    "随着": "with",
    "匿名": "lambda",
    "或": "or",
    "与": "and",
    "不": "not",
    "在": "in",
    "空": "None",
    "对": "True",
    "错": "False",
    "自己": "self",
    "类": "class",
    "异步": "async",
    "等待": "await"
}

functionWords = {
    "符": "chr",
    "二进制": "bin",
    "串": "str",
    "八进制": "oct",
    "符值": "ord",
    "十六进制": "hex",
    "元组": "tuple",
    "长": "len",
    "集合": "set",
    "全为真": "all",
    "字典": "dict",
    "任一为真": "any",
    "列表": "list",
    "迭代": "iter",
    "冻结集合": "frozenset",
    "超类": "super",
    "切片": "slice",
    "乘方": "pow",
    "字节": "bytes",
    "排序": "sort",
    "全局字典": "globals",
    "字节数组": "bytearray",
    "局部字典": "locals",
    "属性": "property",
    "对象": "object",
    "删属性": "delattr",
    "变量字典": "vars",
    "取属性": "getattr",
    "可调用": "callable",
    "有属性": "hasattr",
    "内存视图": "memoryview",
    "设属性": "setattr",
    "哈希": "hash",
    "复数": "complex",
    "商余": "divmod",
    "整数": "int",
    "评估": "eval",
    "浮点数": "float",
    "范围": "range",
    "布尔": "bool",
    "表示": "repr",
    "输入": "input",
    "打包": "zip",
    "打印": "print",
    "打开": "open",
    "执行": "exec",
    "排序": "sorted",
    "编译": "compile",
    "反转": "reversed",
    "映射": "map",
    "和": "sum",
    "是实例": "isinstance",
    "枚举": "enumerate",
    "最大值": "max",
    "断点": "breakpoint",
    "最小值": "min",
    "是子类": "issubclass",
    "绝对值": "abs",
    "下一个": "next",
    "类型": "type",
    "筛选": "filter",
    "格式化": "format",
    "静态方法": "staticmethod",
    "舍入": "round",
    "类方法": "classmethod",
    "退出": "exit",
    "帮助": "help",
    "ascii": "ascii",
    "id": "id",
    "dir": "dir"
}

def readfile(filename):
    with open(filename) as raw:
        script = raw.read()
        raw.close()
    return script

def zpyTopy(zhScript):

    for key in functionWords:
        zhScript = zhScript.replace(key, functionWords[key])

    for key in reservedWords:
        zhScript = zhScript.replace(key, reservedWords[key])

    return zhScript

def pyTozpy(zpyScript):

    for key in functionWords:
        zpyScript = zpyScript.replace(functionWords[key], key)

    for key in reservedWords:
        zpyScript = zpyScript.replace(reservedWords[key], key)

    return zpyScript

def showWords(script):
    wordList = script.split(" ")
    for word in wordList:
        print(word)

def savefile(filename, script):
    if ".zpy" in filename:
        filename = filename.replace(".zpy", ".py")
    elif ".py" in filename:
        filename = filename.replace(".py", ".zpy")
    with open("./"+filename, "w") as newFile:
        newFile.write(script)

def execute(script):
    return subprocess.call(['python', '-c', script])

def zpy(filename, **params):

    script = readfile(filename)

    if ".zpy" in filename:
        script = zpyTopy(script)
        execute(script)
    elif ".py" in filename:
        script = pyTozpy(script)
    else:
        print(f"错误: 未知的后缀名 {filename}")

    savefile(filename, script)

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='执行zpy文件')
    parser.add_argument("filename")
    args = parser.parse_args()
    zpy(args.filename)