我有一个名为foo的脚本。R包含另一个脚本other。R，在同一个目录下:
#!/usr/bin/env Rscript
message("Hello")
source("other.R")
但我想让R找到另一个。R，不管当前工作目录是什么。
换句话说，就是foo。R需要知道自己的路径。我该怎么做呢?
当前回答
Steamer25的方法是有效的，但前提是路径中没有空白。在macOS上，至少cmdArgs[match]返回类似/base/some~+~dir~+~带~+~空格/ for /base/some\ dir\带\空格/。
我通过在返回之前用一个简单的空格替换“~+~”来解决这个问题。
thisFile <- function() {
cmdArgs <- commandArgs(trailingOnly = FALSE)
needle <- "--file="
match <- grep(needle, cmdArgs)
if (length(match) > 0) {
# Rscript
path <- cmdArgs[match]
path <- gsub("\\~\\+\\~", " ", path)
return(normalizePath(sub(needle, "", path)))
} else {
# 'source'd via R console
return(normalizePath(sys.frames()[[1]]$ofile))
}
}
显然，你仍然可以像aprstar那样扩展else块。
其他回答
Supressingfire回答的简化版本:
source_local <- function(fname){
argv <- commandArgs(trailingOnly = FALSE)
base_dir <- dirname(substring(argv[grep("--file=", argv)], 8))
source(paste(base_dir, fname, sep="/"))
}
只是在上面的答案的基础上，作为安全检查，您可以添加一个包装器，当sys.frame(1)失败时(如果interactive() == TRUE可能会失败)，或者源脚本不在主脚本所期望的位置时，它会要求用户找到文件。
fun_path = tryCatch(expr =
{file.path(dirname(sys.frame(1)$ofile), "foo.R")},
error = function(e){'foo.R'}
)
if(!file.exists(fun_path))
{
msg = 'Please select "foo.R"'
# ask user to find data
if(Sys.info()[['sysname']] == 'Windows'){#choose.files is only available on Windows
message('\n\n',msg,'\n\n')
Sys.sleep(0.5)#goes too fast for the user to see the message on some computers
fun_path = choose.files(
default = file.path(gsub('\\\\', '/', Sys.getenv('USERPROFILE')),#user
'Documents'),
caption = msg
)
}else{
message('\n\n',msg,'\n\n')
Sys.sleep(0.5)#goes too fast for the user to see the message on some computers
fun_path = file.choose(new=F)
}
}
#source the function
source(file = fun_path,
encoding = 'UTF-8')
当从R控制台“来源”时，我无法得到Suppressingfire的解决方案。 当使用Rscript时，我无法得到hadley的解决方案。
两全其美?
thisFile <- function() {
cmdArgs <- commandArgs(trailingOnly = FALSE)
needle <- "--file="
match <- grep(needle, cmdArgs)
if (length(match) > 0) {
# Rscript
return(normalizePath(sub(needle, "", cmdArgs[match])))
} else {
# 'source'd via R console
return(normalizePath(sys.frames()[[1]]$ofile))
}
}
如果而不是脚本，foo。R，知道它的路径位置，如果你可以改变你的代码，总是从一个公共根引用所有的源路径，那么这些可能会有很大的帮助:
https://github.com/r-lib/rprojroot或https://rprojroot.r-lib.org/ https://here.r-lib.org/
鉴于
/ app /深度/嵌套/ foo。R /应用程序/其他。R
这是可行的
#!/usr/bin/env Rscript
library(here)
source(here("other.R"))
参见https://rprojroot.r-lib.org/了解如何定义项目根。
#!/usr/bin/env Rscript
print("Hello")
# sad workaround but works :(
programDir <- dirname(sys.frame(1)$ofile)
source(paste(programDir,"other.R",sep='/'))
source(paste(programDir,"other-than-other.R",sep='/'))
