R/airports.R

In jrnold/groundcontrol: Create a Custom Dataset of On-time Flights

download_airports <- function(data_dir, raw_dir) {
  cache_dir <- raw_dir
  airports_url <-
    paste0("https://raw.githubusercontent.com/jpatokal/",
           "openflights/master/data/airports.dat")
  dst <- file.path(cache_dir, "airports.dat")
  if (!file.exists(dst)) {
    download.file(airports_url, dst)
  }

  col_types <-
    cols(
      id = col_integer(),
      name = col_character(),
      city = col_character(),
      country = col_character(),
      faa = col_character(),
      icao = col_character(),
      lat = col_double(),
      lon = col_double(),
      alt = col_integer(),
      tz = col_double(),
      dst = col_character(),
      tzone = col_character()
    )
  raw <- read_csv(dst,
                  col_names = c("id", "name", "city", "country", "faa", "icao",
                                "lat", "lon", "alt", "tz", "dst", "tzone"),
                  col_types = col_types)

  airports <-
    raw %>%
    filter_(~country == "United States", ~faa != "") %>%
    select_(~faa, ~name, ~lat, ~lon, ~alt, ~tz, ~dst, ~tzone) %>%
    group_by_(~faa) %>% slice(1) %>% ungroup() # take first if duplicated

  save_csv(airports, file.path(raw_dir, "airports.csv"))
  save_rda(airports, file = file.path(data_dir, "airports.rda"),
           compress = "bzip2")
  airports
}