R/fetch_aqi_sqlite.R

In Gabegit/gmtools: My small tools for efficiency

#' fetch aqi from local sqlite database
#' @param cities: to fetch like c("南昌市", "北京市")
#' @return data.frame
#' @export
#' @examples
#'fetch_aqi_sqlite(c("南昌市", "北京市"))
#'fetch_aqi_sqlite("南昌市")

fetch_aqi_sqlite <- function(cities) {
	# link to the sqlite
	aqidata <- dplyr::src_sqlite("/Users/gabe/liguomin.dropbox/tasks@mycode/2014-srape-city-air-quality-data/code/AQI-from-2014.db")
	aqidat <- dplyr::tbl(aqidata,"AQI")

	if (length(cities) > 1) {
		sdata <- dplyr::filter(aqidat, city %in% cities)
	} else {
		sdata <- dplyr::filter(aqidat, city == cities)
	}

	return(sdata)

}