lua - 如何确定我的代码是否在 lua 模块中运行？

Question

我正在写一个脚本testmodule.lua并想检查该文件是从另一个脚本导入/需要的还是由 lua testmodule.lua 直接启动的.

如果它是直接通过命令行启动的，我可以做一些测试或运行一个主函数，否则只是导出一些函数而不做任何事情。

Python 有一个 __name__陈述:

if __name__ == '__main__':
    main_entry()

lua中有类似的东西吗？

在单个文件中编写一个 shell util 很有用，它可以直接运行并由其他 lua 脚本导入。但是当一些脚本导入这个文件时，我不喜欢调用 main 函数。

Answer 1

您可以使用以下检查:

if pcall(debug.getlocal, 4, 1) then
  print("in package")
else
  print("in main script")
end

它检查第 4 级的第一个变量中是否有任何内容，这将是当前模块的调用者(如果需要)，或者在主脚本的情况下什么都没有。

请注意，它不区分 require , dofile , loadfile或其他类似的电话。您可能想查看 this recent thread on the Lua mail list讨论了检查这些调用以及一些替代方法。