R/ONS_EAR_subsistemas.R

In brunomioto/reservatoriosBR: Get Brazilian reservoirs data

#' Dados históricos de Energia Armazenada (EAR) por subsistema segundo a ONS
#'
#' @export
#' @param ano_inicial Ano inicial do registro. Série histórica tem início no ano 2000.
#' @param ano_final Ano final do registro. Pode deixar em branco para retornar a data de registro mais recente.
#'
#' @examples
#' ONS_EAR_subsistemas(ano_inicial=2000, ano_final=format(Sys.Date(), "%Y"))


ONS_EAR_subsistemas <- function(ano_inicial=2000, ano_final=format(Sys.Date(), "%Y")){

  if(ano_inicial < 2000|
     ano_final > format(Sys.Date(), "%Y")){
    usethis::ui_todo("Escolha um ano inicial igual ou maior que 2000 e um ano final igual ou menor que {format(Sys.Date(), '%Y')}")
  }else{

  anos <- seq(ano_inicial,ano_final)

  usethis::ui_info("Buscando dados diários de {ano_inicial} até {ano_final} ...")

  historico <- list()

  for (i in anos) {
    dados_ons <- utils::read.csv(glue::glue("https://ons-dl-prod-opendata.s3.amazonaws.com/dataset/ear_subsistema_di/EAR_DIARIO_SUBSISTEMA_{i}.csv"),sep = ";")
    historico[[i]] <- dados_ons
  }

  historico_ear <- do.call(rbind, historico)

  message("Organizando os dados...")

  historico_ear_clean <- historico_ear %>%
    dplyr::mutate(subsistema = dplyr::recode(id_subsistema,
                                      N = "Norte",
                                      NE = "Nordeste",
                                      SE = "Sudeste / Centro-Oeste",
                                      S = "Sul")) %>%
    dplyr::select(-id_subsistema, -nom_subsistema) %>%
    dplyr::rename(data = ear_data,
                  ear_max_subsistema_mwmes = ear_max_subsistema) %>%
    dplyr::select(data, subsistema, ear_max_subsistema_mwmes, ear_verif_subsistema_mwmes, ear_verif_subsistema_percentual)

  historico_ear_clean$data <- as.Date(historico_ear_clean$data)

  if(nrow(historico_ear_clean) == 0){
    usethis::ui_oops("Não foi possível obter os dados. Verifique se as variáveis estão corretas ou entre em contato!")
  }else{
    usethis::ui_done(motivational_message(3))
    return(dplyr::as_tibble(historico_ear_clean))
    }
  }
}