R/googledrive.R

In 11defebrero/formularios11F: Procesamiento Formularios Iniciativa 11F

#' Get drive sheet
#'
#' @param file_id ID de Google Drive de la spreadsheet con los datos
#' @param filename Opcional. Nombre del fichero (incluyendo extensión ".csv") para guardar el dataset en disco.
#'
#' @return dataset descargado tal cual está en Google Drive
#'
get_drive_sheet <- function(file_id, filename = NULL) {

  tmpfile <- ifelse(!is.null(filename), filename, paste0(tempfile(), ".csv"))

  googledrive::drive_download(file = googledrive::as_id(file_id),
                              path = tmpfile,
                              type = "csv",
                              overwrite = TRUE)

  data <- readr::read_csv(tmpfile,
                          col_types = readr::cols(.default=readr::col_character()),
                          locale = readr::locale(encoding="UTF-8"),
                          na = c("NA", "no aplica", "")
  )

  if (is.null(filename)) file.remove(tmpfile)

  return(data)
}



#' Upload drive sheet
#'
#' @param dataset dataset a subir
#' @param file_id ID de Google Drive de la spreadsheet con los datos
#'
upload_drive_sheet <- function(dataset, file_id, filename = NULL) {

  tmpfile <- ifelse(!is.null(filename), filename, paste0(tempfile(), ".csv"))

  write.csv(dataset, tmpfile, row.names=FALSE, quote=TRUE, na="")

  googledrive::drive_update(file = googledrive::as_id(file_id),
                            media = tmpfile)

  if (is.null(filename)) file.remove(tmpfile)

  message("\nDone.")
}



#' Upload/update local file to Drive folder
#'
#' @param local_path ruta del fichero a subir
#' @param drive_folder_id ID de Google Drive de la carpeta donde subir el fichero
#'
#' @export
upload_file_to_drive <- function(local_path, drive_folder_id) {

  drive_path <- googledrive::as_id(drive_folder_id)
  googledrive::drive_put(media=local_path, path=drive_path, name=basename(local_path))

  message("\nDone.")
}

#' Get drive json
#'
#' @param file_id ID de config.json
#' @param filename Nombre del fichero temporal que se descarga de drive.
#'
#' @return datos de configuracion, fundamentalmente ids de sheets y formularios en drive
#'
get_drive_json <- function(file_id, filename = NULL) {

  tmpfile <- ifelse(!is.null(filename), filename, tempfile() )

  googledrive::drive_download(file = googledrive::as_id(file_id),
                              path = tmpfile,
                              overwrite = TRUE)

  json_content <- rjson::fromJSON(file = filename)

  file.remove(filename)

  return(json_content)
}