Dapatkan semua fungsi bersumber

Di R, saya menggunakan source()untuk memuat beberapa fungsi:

source("functions.R")

Apakah mungkin untuk mendapatkan daftar semua fungsi yang didefinisikan dalam file ini? Sebagai nama fungsi. (Mungkin source()sendiri entah bagaimana dapat mengembalikannya?).

PS: Cara terakhir adalah memanggil source()kedua kalinya seperti local({ source(); })dan kemudian melakukan ls()fungsi-fungsi di dalam dan menyaring, tapi itu terlalu rumit - apakah ada solusi yang lebih mudah dan kurang canggung?

Saya pikir cara terbaik adalah dengan sumber file ke lingkungan sementara. Permintaan lingkungan itu untuk semua fungsi, lalu salin nilai-nilai itu ke lingkungan induk.

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")

Agak kikuk tetapi Anda bisa melihat perubahan pada objek sebelum dan sesudah sourcepanggilan seperti ini.

    # 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"

Saya pikir regex ini menangkap hampir setiap jenis fungsi yang valid (operator biner, fungsi penugasan) dan setiap karakter yang valid dalam nama fungsi, tetapi saya mungkin telah melewatkan case edge.

# 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<-`"

Jika ini adalah skrip Anda sendiri sehingga Anda memiliki kendali atas cara memformatnya, konvensi sederhana akan cukup. Pastikan saja setiap nama fungsi dimulai pada karakter pertama pada barisnya dan bahwa kata tersebut functionjuga muncul pada baris tersebut. Penggunaan kata lain functionakan muncul pada baris yang dimulai dengan spasi atau tab. Maka solusi satu baris adalah:

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

Keuntungan dari pendekatan ini adalah bahwa

• ini sangat sederhana . Aturan hanya dinyatakan dan hanya ada satu baris sederhana kode R yang diperlukan untuk mengekstrak nama fungsi. Regex juga sederhana dan untuk file yang sudah ada sangat mudah untuk memeriksa - cukup ambil kata functiondan periksa apakah setiap kemunculan yang ditampilkan mengikuti aturan.

• tidak perlu menjalankan sumbernya. Itu sepenuhnya statis .

• dalam banyak kasus Anda tidak perlu mengubah file sumber sama sekali dan dalam kasus lain akan ada perubahan minimal. Jika Anda menulis skrip dari awal dengan mengingat hal ini, bahkan lebih mudah untuk mengaturnya.

Ada banyak alternatif lain di sepanjang gagasan konvensi. Anda dapat memiliki regex yang lebih canggih atau Anda dapat menambahkan # FUNCTIONdi akhir baris pertama dari definisi fungsi apa pun jika Anda menulis skrip dari awal dan kemudian mengeluarkan frasa itu dan mengekstrak kata pertama pada baris tersebut tetapi saran utama di sini tampaknya sangat menarik karena kesederhanaannya dan kelebihan lainnya yang tercantum.

Uji

# 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"

Ini mengadaptasi kode yang digunakan dalam posting dari komentar saya untuk mencari urutan token (simbol, operator penugasan, kemudian berfungsi), dan harus mengambil fungsi yang telah ditentukan. Saya tidak yakin apakah itu kuat seperti jawaban MrFlick, tetapi itu adalah pilihan lain:

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"]]
}