3

这个问题类似于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：更新的问题明确指出不能修改作为源的脚本（假设它们不受您的控制）。

4

4 回答 4

2

您可以使用trace在函数中注入代码，因此您可以强制所有source调用 set 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即使有错误。

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

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

或者可能

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

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

于 2021-10-02T14:52:48.800 回答
1

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

重新定义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)
于 2021-10-04T10:05:27.590 回答
1

保护自己免受无法控制的代码副作用的最佳方法是隔离。您可以使用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.

于 2021-10-06T13:17:05.820 回答
0

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

文件 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>
于 2021-09-30T12:17:33.943 回答