Nhận tất cả các chức năng nguồn

Trong R, tôi đang sử dụng source() để tải một số chức năng:

source("functions.R")

Có thể lấy danh sách tất cả các chức năng được xác định trong tệp này? Như tên hàm. (Có lẽsource() chính nó bằng cách nào đó có thể trả lại nó?).

Tái bút: Phương án cuối cùng là gọi source()lần thứ hai như local({ source(); })sau đó thực hiện các ls()chức năng bên trong và lọc, nhưng điều đó quá phức tạp - có giải pháp nào dễ dàng và ít vụng về hơn không?

Tôi nghĩ cách tốt nhất sẽ là nguồn tệp vào một môi trường tạm thời. Truy vấn môi trường đó cho tất cả các hàm, sau đó sao chép các giá trị đó vào môi trường cha.

my_source <- function(..., local=NULL) {
  tmp <- new.env(parent=parent.frame())
  source(..., local = tmp)
  funs <- names(tmp)[unlist(eapply(tmp, is.function))]
  for(x in names(tmp)) {
    assign(x, tmp[[x]], envir = parent.frame())
  }
  list(functions=funs)
}

my_source("script.R")

Hơi khó hiểu một chút nhưng bạn có thể nhìn vào những thay đổi trong các đối tượng trước và sau sourcecuộc gọi như thế này.

    # optionally delete all variables
    #rm(list=ls())

    before <- ls()
    cat("f1 <- function(){}\nf2 <- function(){}\n", file = 'define_function.R')
    # defines these
    #f1 <- function() {}
    #f2 <- function() {}
    source('define_function.R')
    after <- ls()

    changed <- setdiff(after, before)
    changed_objects <- mget(changed, inherits = T)
    changed_function <- do.call(rbind, lapply(changed_objects, is.function))
    new_functions <- changed[changed_function]

    new_functions
    # [1] "f1" "f2"

Tôi nghĩ rằng regex này nắm bắt hầu hết mọi loại hàm hợp lệ (toán tử nhị phân, hàm gán) và mọi ký tự hợp lệ trong một tên hàm, nhưng tôi có thể đã bỏ lỡ một trường hợp cạnh.

# lines <- readLines("functions.R")

lines <- c(
  "`%in%` <- function",
  "foo <- function",
  "foo2bar <- function",
  "`%in%`<-function",
  "foo<-function",
  ".foo <-function",
  "foo2bar<-function",
  "`foo2bar<-`<-function",
  "`foo3bar<-`=function",
  "`foo4bar<-` = function",
  "` d d` <- function", 
  "lapply(x, function)"
)
grep("^`?%?[.a-zA-Z][._a-zA-Z0-9 ]+%?(<-`)?`?\\s*(<-|=)\\s*function", lines)
#>  [1]  1  2  3  4  5  6  7  8  9 10
funs <- grep("^`?%?[.a-zA-Z][._a-zA-Z0-9 ]+%?(<-`)?`?\\s*(<-|=)\\s*function", lines, value = TRUE)
gsub("^(`?%?[.a-zA-Z][._a-zA-Z0-9 ]+%?(<-`)?`?).*", "\\1", funs)
#>  [1] "`%in%`"      "foo "        "foo2bar "    "`%in%`"      "foo"        
#>  [6] ".foo "       "foo2bar"     "`foo2bar<-`" "`foo3bar<-`" "`foo4bar<-`"

Nếu đây là tập lệnh của riêng bạn để bạn có quyền kiểm soát cách nó được định dạng thì một quy ước đơn giản sẽ là đủ. Chỉ cần đảm bảo rằng mỗi tên hàm bắt đầu từ ký tự đầu tiên trên dòng của nó và từ đó functioncũng xuất hiện trên dòng đó. Bất kỳ việc sử dụng nào khác của từ functionsẽ xuất hiện trên một dòng bắt đầu bằng khoảng trắng hoặc tab. Sau đó, một giải pháp một dòng là:

sub(" .*", "", grep("^\\S.*function", readLines("myscript.R"), value = TRUE))

Ưu điểm của phương pháp này là

• nó là rất đơn giản . Các quy tắc được nêu đơn giản và chỉ có một dòng mã R đơn giản cần thiết để trích xuất các tên hàm. Regex cũng đơn giản và đối với một tệp hiện có, rất dễ kiểm tra - chỉ cần grep từ functionvà kiểm tra xem mỗi lần xuất hiện được hiển thị có theo quy tắc hay không.

• không cần chạy nguồn. Nó hoàn toàn tĩnh .

• trong nhiều trường hợp, bạn sẽ không cần phải thay đổi tệp nguồn và trong những trường hợp khác sẽ có những thay đổi tối thiểu. Nếu bạn đang viết kịch bản từ đầu với ý nghĩ này thì việc sắp xếp lại càng dễ dàng hơn.

Có nhiều lựa chọn thay thế khác cùng với ý tưởng về các quy ước. bạn có thể có một biểu thức chính quy phức tạp hơn hoặc bạn có thể thêm # FUNCTIONvào cuối dòng đầu tiên của bất kỳ định nghĩa hàm nào nếu bạn đang viết tập lệnh từ đầu và sau đó rút ra cụm từ đó và trích xuất từ ​​đầu tiên trên dòng nhưng dường như gợi ý chính ở đây đặc biệt hấp dẫn do tính đơn giản của nó và các lợi thế khác được liệt kê.

Kiểm tra

# generate test file
cat("f <- function(x) x\nf(23)\n", file = "myscript.R") 

sub(" .*", "", grep("^\\S.*function", readLines("myscript.R"), value = TRUE))
## [1] "f"

Điều này điều chỉnh mã được sử dụng trong bài đăng từ nhận xét của tôi để tìm kiếm một chuỗi mã thông báo (ký hiệu, toán tử gán, sau đó là hàm) và nó sẽ lấy bất kỳ hàm xác định nào. Tôi không chắc liệu nó có mạnh mẽ như câu trả lời của MrFlick không, nhưng đó là một lựa chọn khác:

source2 <- function(file, ...) {
  source(file, ...)
  t_t <- subset(getParseData(parse(file)), terminal == TRUE)
  subset(t_t, token == "SYMBOL" & 
           grepl("ASSIGN", c(tail(token, -1), NA), fixed = TRUE) & 
           c(tail(token, -2), NA, NA) == "FUNCTION")[["text"]]
}