R/aa-utils.R

In CBDRH/covidrecon: Pulls and Analyses COVID data

str_remove_x <- function(x) stringr::str_remove_all(x, "x")

covid_jhu_clean_date <- function(x) lubridate::mdy(str_remove_x(x))

add_continent <- function(x, country_source){
  x %>%
  dplyr::mutate(
    continent = countrycode::countrycode(
      sourcevar = {{ country_source }},
      origin = "country.name",
      destination = "continent",
      warn = FALSE
    )
  )
}

glue_date <- function(date){
  glue::glue("https://www.ecdc.europa.eu/sites/default/files/documents/COVID-19-geographic-disbtribution-worldwide-{date}.xlsx")
}

discard_null <- function(x) purrr::discard(x, is.null)

pluck_result <- function(x) purrr::pluck(x, "result")

inherits_data_frames <- function(x) purrr::map_lgl(x, inherits, "data.frame")

create_title_date <- function(covid_data){
  date <- format(max(covid_data$date), "%d %B %Y")
  glue::glue("COVID-19 epidemic trajectories up to {date}")
}


#' Difference in days between two dates
#'
#' @param x first date
#' @param y second date
#'
#' @return numeric number of days
#' @export
diff_days <- function(x, y){
  as.numeric(difftime(x, y, units = "days"))
}

underscore_to_space <- function(x){
  stringr::str_replace_all(string = x,
                           pattern = "_",
                           replacement = " ")
}

filter_last_country_date <- function(covid_data){
  covid_data %>%
  dplyr::group_by(geo_id) %>%
  dplyr::filter(date == dplyr::last(date))
}