写一个python脚本,来格式化lua代码

可以使用Python中的`luaparser`库来格式化Lua代码。首先需要使用`pip`命令安装该库：

pip install luaparser

然后可以编写以下Python脚本：

from luaparser import ast
from luaparser.pretty import LuaPrettyPrinter

def format_lua_code(lua_code):
    # 解析AST
    tree = ast.parse(lua_code)

    # 格式化代码
    formatter = LuaPrettyPrinter(indent='  ')
    formatted_code = formatter.pretty(tree)

    return formatted_code

在上面的脚本中，`format_lua_code`函数接受一个字符串参数`lua_code`，表示要格式化的Lua代码。函数内部使用`ast.parse()`函数将Lua代码解析为一个AST（抽象语法树）对象，然后使用`LuaPrettyPrinter`类来格式化AST并生成格式化后的代码。其中，`indent`参数指定缩进字符，这里使用两个空格作为缩进。

使用以上方法，可以将一个Lua脚本字符串格式化为易于阅读的形式。例如：

lua_code = """
function add(a, b)
    return a + b
end

-- test
print(add(1, 2))
"""
formatted_code = format_lua_code(lua_code)
print(formatted_code)

输出结果为：

function add(a, b)
  return a + b
end

-- test
print(add(1, 2))