R/output.R

In dylanrussellmd/nsqipr: Interact with NSQIP data via R

output <- function(df, file, filename, csv, rds, longtables, progbar) {
  output_rds(df, file, filename, rds, progbar)
  output_csv(df, file, filename, csv, progbar)
  output_longtables(longtables, file, filename, csv, rds, progbar)
}

output_rds <- function(df, file, filename, rds, progbar) {
  if(rds) {
    tick(progbar, "writing main .rds for", filename)
    rds_path <-  fs::path(fs::path_dir(file), "rds")
    fs::dir_create(rds_path)
    saveRDS(df, file = paste(fs::path(rds_path, fs::path_ext_remove(filename)), "_clean.rds", sep =""))
  }
}

output_csv <- function(df, file, filename, csv, progbar) {
  if(!is.na(csv)) {
    csv_path <- fs::path(fs::path_dir(file), "csv")
    fs::dir_create(csv_path)
    if(csv == "indiv" | csv == "both") {
      tick(progbar, "writing main .csv for", filename)
      data.table::fwrite(df, file = paste(fs::path(csv_path, fs::path_ext_remove(filename)), "_clean.csv", sep =""), showProgress = FALSE)
    }
    if(csv == "append" | csv == "both") {
      tick(progbar, "appending to full main .csv:", filename)
      data.table::fwrite(df, file = paste(fs::path(csv_path, basename(dirname(file))), "_full_clean.csv", sep = ""), showProgress = FALSE, append = TRUE)
    }
  }
}

output_longtables <- function(longtables, file, filename, csv, rds, progbar) {
  tick(progbar, "writing supplementary tables for", filename)
  lapply(longtables, function(x) {
    if(isFullDT(x)) {
      name <- names(x)[[3]]
      if(!is.na(csv)) {
        csv_path <- fs::path(fs::path_dir(file), "csv")
        if(csv == "indiv" | csv == "both") {
          paste(fs::path(csv_path, filename), "_", name ,".csv", sep ="")
          data.table::fwrite(x, file = paste(fs::path(csv_path, fs::path_ext_remove(filename)), "_", name ,".csv", sep =""), showProgress = FALSE)
        }
        if(csv == "append" | csv == "both") {
          data.table::fwrite(x, file = paste(fs::path(csv_path, basename(dirname(file))), "_full_", name ,".csv", sep = ""), showProgress = FALSE, append = TRUE)
        }
      }
      if(rds) {
        rds_path <-  fs::path(fs::path_dir(file), "rds")
        saveRDS(x, file = paste(fs::path(rds_path, fs::path_ext_remove(filename)), "_", name ,".rds", sep =""))
      }
    }
  })
}