รับฟังก์ชั่นที่มาทั้งหมด

ใน 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"

ฉันคิดว่า regex นี้จับเกือบทุกประเภทของฟังก์ชั่นที่ถูกต้อง (ตัวดำเนินการไบนารีฟังก์ชั่นการมอบหมาย) และทุกตัวละครที่ถูกต้องในชื่อฟังก์ชั่น แต่ฉันอาจจะพลาดกรณีขอบ

# 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และตรวจสอบว่าแต่ละเหตุการณ์ที่ปรากฏตามกฎ

• ไม่จำเป็นต้องเรียกใช้แหล่งที่มา มันเป็นทั้งแบบคงที่

• ในหลายกรณีคุณไม่จำเป็นต้องเปลี่ยนไฟล์ต้นฉบับเลยและในกรณีอื่น ๆ จะมีการเปลี่ยนแปลงเล็กน้อย หากคุณกำลังเขียนสคริปต์ตั้งแต่เริ่มต้นโดยคำนึงถึงสิ่งนี้มันจะง่ายกว่าในการจัดการ

มีทางเลือกอื่น ๆ อีกมากมายตามแนวความคิดของการประชุม คุณอาจมี regex ที่ซับซ้อนมากขึ้นหรือคุณสามารถเพิ่ม# FUNCTIONที่ท้ายบรรทัดแรกของนิยามฟังก์ชันใด ๆ หากคุณกำลังเขียนสคริปต์ตั้งแต่เริ่มต้นและจากนั้น grep out วลีนั้นและแยกคำแรกในบรรทัด แต่คำแนะนำหลักที่นี่ดูเหมือน น่าสนใจเป็นพิเศษเนื่องจากความเรียบง่ายและข้อดีอื่น ๆ ที่ระบุไว้

ทดสอบ

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