从 .GlobalEnv 中分离脚本：源脚本的源脚本

Question

这个问题类似于Source script to separate environment in R, not the global environment，但有一个关键的转折。

考虑一个引用另一个脚本的脚本：

# main.R
source("funs.R")
x <- 1

# funs.R
hello <- function() {message("Hi")}

我想获取脚本main.R 并将所有内容保存在“本地”环境中，例如env <- new.env()。通常，可以调用source("main.R", local = env) 并期望一切都在env 环境中。但是，这里不是这样：x 是env 的一部分，但函数hello 不是！它位于.GlobalEnv。

问题：如何在 R 中将脚本提供给单独的环境，即使该脚本本身提供其他脚本，并且不修改其他正在提供的脚本？

感谢您的帮助，如果我能澄清任何事情，请告诉我。

编辑 1：更新问题，明确指出不能修改作为源的脚本（假设它们不受您的控制）。

Answer 1

您可以使用trace 在函数中注入代码， 所以你可以强制所有source 调用设置local = TRUE。 如果local 是FALSE，我在这里只覆盖它，以防任何对source 的嵌套调用实际上由于它们自己的特殊逻辑而将其设置为其他环境。

env <- new.env()

# use !isTRUE if you want to support older R versions (<3.5.0)
tracer <- quote(
  if (isFALSE(local)) {
    local <- TRUE
  }
)

trace(source, tracer, print = FALSE, where = .GlobalEnv)

# if you're doing this inside a function, uncomment the next line
#on.exit(untrace(source, where = .GlobalEnv))

source("main.R", local = env)

如代码中所述， 如果您将此逻辑包装在一个函数中， 考虑使用on.exit 来确保您untrace 即使有错误。

编辑：如 cmets 中所述， 如果您将加载的某些脚本假设有 1 个（全局）环境，一切都结束了，这可能会出现问题。 我想您可以将跟踪器更改为类似

tracer <- quote(
  if (missing(local)) {
    local <- TRUE
  }
)

或许

tracer <- quote(
  if (isFALSE(local)) {
    # fetch the specific environment you created
    local <- get("env", .GlobalEnv)
  }
)

前者假设如果脚本根本没有指定local， 它不关心哪个环境最终容纳了所有东西。 后者假定未指定 local 或将其设置为 FALSE 的 source 调用希望所有内容都在 1 个环境中结束， 并修改逻辑以使用您的环境而不是全局环境。

Answer 2

保护自己免受无法控制的代码副作用的最佳方法是隔离。您可以使用callr 轻松执行在单独的 R 会话中隔离的脚本：

使用环境：

env <- new.env()
env <- as.environment(callr::r(function(env) {
    list2env(env, .GlobalEnv)
    source("main.R")
    as.list(.GlobalEnv)
}, args = list(as.list(env))))
env
#> <environment: 0x0000000018124878>
env$hello()
#> Hi

坚持列表的更简单版本：

params <- list()
results <- callr::r(function(params) {
    list2env(params, .GlobalEnv)
    source("main.R")
    as.list(.GlobalEnv)
}, args = list(params))
results
#> $x
#> [1] 1
#> 
#> $hello
#> function () 
#> {
#>   message("Hi")
#> }
results$hello()
#> Hi

param 部分仅在您确实需要提供输入脚本时才需要（不用于您的示例）。 显然，这不适用于开放连接和类似的东西。在这种情况下，您可能需要查看callr::r_session。

Answer 3

免责声明：非常丑陋且有潜在危险，但无论如何。

重新定义source:

env<-new.env()
source<-function(...) base::source(..., local = env)
source("main.R")
#just remove your redefinition when you don't need it
rm(source)

Answer 4

在所有嵌套的source() 函数中使用参数local = TRUE。

文件 3：

# last.R
y <- 2

文件 2：

# funs.R
source("last.R", local = TRUE)
hello <- function() {message("Hi")}

文件 1：

# main.R
source("funs.R", local = TRUE)
x <- 1

采购文件 1：

env <- new.env()
source("main.R", local = env)

检查环境中的内容：

as.list(env)

$x
[1] 1

$y
[1] 2

$hello
function() {message("Hi")}
<environment: 0x000001d9dff719b8>