R/utils/saving_utils.R

In jacobkap/crime_data: Clean FBI data

library(haven)
save_files <- function(data,
                       year,
                       file_name,
                       save_name,
                       rds_only = FALSE) {
  if (any(nchar(names(data)) > 31)) {
    print(names(data)[nchar(names(data)) > 31])
  }

  data <-
    data %>%
    dplyr::mutate_if(is.Date, as.character)
  data <- ungroup(data)
  data <- as.data.frame(data)


  saveRDS(data, file =  paste0(save_name, year, ".rds"))
  if (!rds_only) {
    write_dta(data, path = paste0(save_name, year, ".dta"))
  }
}

save_as_zip <- function(file_name, pattern = NULL) {
  file_ext  <- c("rds", "dta")
  all_files <- list.files()


  all_file_extentions <- gsub(".*\\.", "", all_files)
  file_ext <- file_ext[file_ext %in% unique(all_file_extentions)]
  if (!is.null(pattern)) {
    sps_files <- all_files[grep("maltz|manual|sps$|record description",
                                all_files, ignore.case = TRUE)]
    all_files <- list.files(pattern = pattern)
    all_files <- c(sps_files, all_files)
  }

  if (any(grepl(".zip$", all_files))) {
    all_files <- all_files[-grep(".zip$", all_files)]
  }

  codebooks <- list.files(pattern = "maltz|manual|codebook|pdf$|sps$|doc$|docx$|txt$")
  for (i in seq_along(file_ext)) {
    zip_files <- all_files[grep(file_ext[i], all_files)]
    zip_files <- c(zip_files, codebooks)
    zip_files <- sort(zip_files)

    zip::zipr(zipfile = paste0(file_name,
                               file_ext[i], ".zip"),
              files = zip_files)
  }
}

save_raw_as_zip <- function(file_name) {
  all_files <- list.files()
  zip::zip(zipfile = paste0(file_name, ".zip"),
           files = all_files)
}