使用 Python 搜索 Lua 文件中的所有函数调用

Posted 2023-03-28

【中文标题】使用 Python 搜索 Lua 文件中的所有函数调用

【英文标题】：Search for all function calls in a Lua file using Python

【发布时间】：2021-11-30 15:09:07

【问题描述】：

我想使用 python 搜索 Lua 文件中的所有函数调用。 例如，我有这个 Lua 代码：

function displayText (text)
    print(text)
end

sendGreetings = function(sender, reciever) 
    print("Hi " ..reciever.. " greetings from " ..sender.. "!")
end

displayText("Hello, World!")

sendGreetings("Roger", "Michael")

我希望我的 python 代码在该代码中搜索函数调用并返回一个包含函数名称和参数的字典，所以输出应该是这样的：

# function_name: [param1, param2]
"displayText": ["Hello, World!"], "sendGreetings": ["Roger", "Michael"]

我尝试使用正则表达式来实现它，但我遇到了各种各样的问题，而且结果不准确。另外，我不相信 Python 有 Lua 解析器。

【问题讨论】：

Lua 代码中调用的函数是否总是只接受字符串参数？

如果调用的模式很简单，那么你可以使用我的 ltokenp，web.tecgraf.puc-rio.br/~lhf/ftp/lua/#ltokenp

【参考方案1】：

你可以使用luaparser（pip install luaparser）和递归来遍历ast：

import luaparser
from luaparser import ast
class FunCalls:
   def __init__(self):
      self.f_defs, self.f_calls = [], []
   def lua_eval(self, tree):
      #attempts to produce a value for a function parameter. If value is a string or an integer, returns the corresponding Python object. If not, returns a string with the lua code
      if isinstance(tree, (luaparser.astnodes.Number, luaparser.astnodes.String)):
         return tree.n if hasattr(tree, 'n') else tree.s
      return ast.to_lua_source(tree)
   def walk(self, tree):
      to_walk = None
      if isinstance(tree, luaparser.astnodes.Function):
         self.f_defs.append((tree.name.id, [i.id for i in tree.args]))
         to_walk = tree.body
      elif isinstance(tree, luaparser.astnodes.Call):
         self.f_calls.append((tree.func.id, [self.lua_eval(i) for i in tree.args]))
      elif isinstance(tree, luaparser.astnodes.Assign):
         if isinstance(tree.values[0], luaparser.astnodes.AnonymousFunction):
            self.f_defs.append((tree.targets[0].id, [i.id for i in tree.values[0].args]))
      if to_walk is not None:
         for i in ([to_walk] if not isinstance(to_walk, list) else to_walk):
             self.walk(i)
      else:
         for a, b in getattr(tree, '__dict__', ).items():
            if isinstance(b, list) or 'luaparser.astnodes' in str(b):
               for i in ([b] if not isinstance(b, list) else b):
                   self.walk(i)

---

把它们放在一起：

s = """
function displayText (text)
   print(text)
end

sendGreetings = function(sender, reciever) 
   print("Hi " ..reciever.. " greetings from " ..sender.. "!")
end

displayText("Hello, World!")

sendGreetings("Roger", "Michael")
"""
tree = ast.parse(s)
f = FunCalls()
f.walk(tree)
print(dict(f.f_defs)) #the function definitions with signatures
calls = a:b for a, b in f.f_calls if any(j == a for j, _ in f.f_defs)    
print(calls)

输出：

'displayText': ['text'], 'sendGreetings': ['sender', 'reciever']
'displayText': ['Hello, World!'], 'sendGreetings': ['Roger', 'Michael']