如何检查一个脚本是通过dofile()包含的还是直接在Lua中运行的？

2017-5-23 12:33:22
收藏：0
阅读：144
评论：2

在 Python 中，你可以很容易地检查脚本是否被直接调用（并在调用时执行某些函数，通常用于测试）：

if __name__ == "__main__":
    main()

在 Lua 中，是否有一种方法来检测它是否直接运行（lua foo.lua）或包含在另一个脚本中（dofile（'foo.lua'））。

总是有一种平凡（且丑陋）的方式，在主脚本文件中定义一些虚拟全局变量，然后在 foo.lua 中检查它是否被定义，这样做不太好，如果有更好的方法就更好了。

点赞
用户734069
用户734069

不行。Lua没有办法判断脚本是否直接被调用。

请注意：Lua和Python存在不同的目的。Python专门用于命令行脚本。你编写一个脚本，调用可能用Python、C++或其他语言编写的模块。Python脚本最终控制哪些代码被调用。

Lua首先是一个嵌入式脚本语言。虽然你确实可以使用独立的lua.exe解释器来创建命令行脚本，但这有点笨拙。这种语言的主要目的是嵌入到用C、C++或其他编写的应用程序中。

因此，没有"主"脚本的概念。你如何定义什么是"主"的，什么不是？如果C代码可以随时加载任何脚本，任何目的，其中哪一个是"主要的"？

如果你想定义这个概念，这很简单。只需在你的"主"脚本顶部添加如下内容：

do
  local old_dofile = dofile
  function new_dofile(...)
    if(__is_main) then
      __is_main = __is_main + 1
    else
      __is_main = 1
    end
    old_dofile(...)
    __is_main = __is_main - 1
    if(__is_main == 0) then __is_main = nil end
  end
  dofile = new_dofile
end

当然，这对使用require加载的任何内容都不起作用。你需要为require编写一个版本执行此操作。如果外部代码使用C-API加载函数，它肯定行不通（因此require可能不起作用）。

2012-01-30 15:02:59
用户282536
用户282536

在顶层，您可以检查debug.getinfo(2)是否为空值。

来自 http://www.lua.org/manual/5.1/manual.html#pdf-debug.getinfo

您可以将一个数字作为函数的值，表示给定线程的调用堆栈的函数级别：级别0是当前函数（getinfo本身）；级别1是调用getinfo的函数；依此类推。如果函数是大于活动函数数量的数字，则getinfo返回nil。

2012-01-31 04:54:25

作者:

用户56761

用户56761
推荐文章