如何用R语言基于override.R覆盖base.R的赋值表达式生成new.R？

针对你的需求，用文本处理工具（如sed）处理多行赋值容易出错，直接用R解析代码结构是最可靠的方案，以下是具体实现：

1. 编写处理脚本（保存为merge_assignments.R） #

# 读取文件内容
read_lines <- function(file) {
  readLines(file, warn = FALSE)
}

base_lines <- read_lines("base.R")
override_lines <- read_lines("override.R")

# 解析代码，提取赋值语句的左值与对应行范围
parse_assignments <- function(lines) {
  code <- paste(lines, collapse = "\n")
  ast <- parse(text = code)
  assignments <- list()
  
  for (expr in ast) {
    # 识别 <- 和 = 两种赋值语句
    if (is(expr, "<-") || is(expr, "=")) {
      lhs <- deparse(expr[[1]])
      src_ref <- attr(expr, "srcref")
      assignments[[lhs]] <- list(start = src_ref[1], end = src_ref[3])
    }
  }
  assignments
}

base_assignments <- parse_assignments(base_lines)
override_assignments <- parse_assignments(override_lines)

# 获取override中指定左值的赋值内容
get_override_content <- function(lhs, override_lines, override_assignments) {
  range <- override_assignments[[lhs]]
  override_lines[range$start:range$end]
}

# 生成新文件内容
new_lines <- base_lines
for (lhs in names(override_assignments)) {
  if (lhs %in% names(base_assignments)) {
    # 替换base中对应赋值语句
    base_range <- base_assignments[[lhs]]
    new_lines <- c(
      new_lines[1:(base_range$start - 1)],
      get_override_content(lhs, override_lines, override_assignments),
      new_lines[(base_range$end + 1):length(new_lines)]
    )
  } else {
    # 若base中无对应赋值，追加到文件末尾（可按需修改）
    new_lines <- c(new_lines, get_override_content(lhs, override_lines, override_assignments))
  }
}

# 写入新文件
writeLines(new_lines, "new.R")

2. 执行脚本 #

在终端运行以下命令：

Rscript merge_assignments.R

方案特点 #

• 自动识别单行/多行的lhs <- rhs和lhs = rhs赋值语句
• 精准替换base.R中与override.R左值匹配的赋值内容，保留base中的注释和非赋值代码
• 支持将override中新增的赋值追加到base末尾（可根据需求调整此逻辑）

内容的提问来源于stack exchange，提问作者statquant