Отримати всі джерела функцій

У R я використовую source()для завантаження деяких функцій:

source("functions.R")

Чи можливо отримати список усіх функцій, визначених у цьому файлі? Як імена функцій. (Може, source()сама може якось це повернути?).

PS: Останнім засобом було б зателефонувати source()вдруге на кшталт, local({ source(); })а потім виконувати функції ls()всередині та фільтрувати, але це занадто складно - чи є простіше та менш незграбне рішення?

Я думаю, що найкращим способом було б передати файл у тимчасове середовище. Перевірте це середовище для всіх функцій, а потім скопіюйте ці значення в батьківське середовище.

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

Це трохи незграбно, але ви можете подивитися на зміни в об'єктах до і після sourceдзвінка, як це.

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

Я думаю, що цей регулярний вираз охоплює майже кожен дійсний тип функції (двійковий оператор, функції присвоєння) та кожен дійсний символ у імені функції, але, можливо, я пропустив кращий регістр.

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

Якщо це ваш власний сценарій, щоб у вас був контроль над тим, як він відформатований, буде достатньо простої конвенції. Просто переконайтеся, що ім’я кожної функції починається з першого символу на його рядку, і що слово functionтакож з’являється в цьому рядку. Будь-яке інше вживання слова functionповинно з’являтися на рядку, який починається з пробілу чи вкладки. Тоді однолінійне рішення:

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

Переваги такого підходу полягають у тому, що

• це дуже просто . Правила викладені просто, і для вилучення імен функцій потрібен лише один простий рядок коду R. Regex також простий, і для вже наявного файлу його дуже легко перевірити - просто приберіть слово functionі перевірте, чи відповідає кожне відображене виникнення за правилом.

• не потрібно запускати джерело. Це повністю статично .

• у багатьох випадках вам взагалі не потрібно буде змінювати вихідний файл, а в інших - мінімальні зміни. Якщо ви пишете сценарій з нуля, маючи це на увазі, це ще простіше домовитись.

Існує багато інших альтернатив уздовж ідеї конвенцій. у вас може бути більш складний регулярний вираз або ви можете додати # FUNCTIONв кінці першого рядка будь-якого визначення функції, якщо ви пишете сценарій з нуля, а потім виграбуєте цю фразу і витягуєте перше слово в рядку, але головна пропозиція тут здається особливо привабливий через свою простоту та інші перелічені переваги.

Тест

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

Це адаптує код, використаний у публікації з мого коментаря, для пошуку послідовності лексем (символ, оператор призначення, потім функція), і він повинен захоплювати будь-які визначені функції. Я не впевнений, чи надійний він як відповідь MrFlick, але це інший варіант:

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