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

2018-3-20 2:51:23

收藏：0

阅读：183

评论：1

我正在编写一个名为testmodule.lua的脚本，并想检查该文件是否从另一个脚本导入/需要，或者直接由lua testmodule.lua启动。

如果是直接从命令行启动，我可以进行一些测试或运行主函数，否则只需导出一些函数并不做任何处理。

Python中有一个__name__语句:

if __name__ == '__main__':
    main_entry()

Lua中是否有类似之处？

将一个shell工具写在一个单独的文件中很有用，该文件可以直接运行并被其他Lua脚本导入。但是当某些脚本导入此文件时，我不喜欢调用主函数。

用户1442917

你可以使用以下检查：

if pcall(debug.getlocal, 4, 1) then
  print("在包中")
else
  print("在主脚本中")
end

它检查第四级中第一个变量中是否有任何内容，这将是当前模块的调用者（如果需要），或者在主脚本的情况下为 nothing。

请注意，它不区分 require，dofile，loadfile 或其他类似调用之间的区别。您可能还想查看最近的 Lua 邮件列表中的此线程，其中讨论了检查这些调用的方法以及一些替代方式。

2018-03-20 05:20:56

评论区的留言会收到邮件通知哦~

作者:

用户7012683