R/cleanup-annotate.R

In nyuglobalties/blueprintr: Automagically Document and Test Datasets Using Targets Or Drake

has_annotation_cleanup <- function(bp) {
  isTRUE(bp$annotate)
}

annotate_variables <- function(df, bp, meta) {
  overwrite <- bp$annotate_overwrite

  for (vn in names(df)) {
    df[[vn]] <- annotate_variable(df[[vn]], vn, meta, overwrite)
  }

  df
}

annotate_variable <- function(x, varname, meta, overwrite) {
  meta <- tidytable::filter(meta, .data$name == varname)
  fields <- setdiff(names(meta), c("name", "type", "dropped"))
  fields <- fields[!grepl("^\\.", fields)]

  if (using_improved_annotations()) {
    overwrite <- TRUE
  }

  for (f in fields) {
    x <- add_annotation(x, f, meta[[f]], overwrite)

    if (using_improved_annotations() && has_super_annotation(x, f)) {
      x <- set_annotation(x, f, super_annotation(x, f))
      x <- remove_super_annotation(x, f)
    }
  }

  x
}