addCensusData <- function()

in dbViewR/R/addCensusData.R [20:75]

• 48 lines of code
• 8 McCabe index (conditional complexity)

addCensusData <- function( db = dbViewR::selectFromDB(), 
                           geography = "tract", variables, year,
                           state = "WA", county = "King",
                           source = "decennial",
                           credentials_path = '/home/rstudio/seattle_flu')
  
{
  #get the census api key value from the file
  keyFile <- file(file.path(credentials_path, "census_api_key.txt"), open = 'r') 
  keyValue <- readLines(keyFile, n = 1)
  close(keyFile)
  
  Sys.setenv(CENSUS_API_KEY = keyValue)
  
  if(source == "decennial")
     {
       dataset <- "sf1"
  }
  else if(source =="acs")
  {
    dataset <- "acs5"
  }
  else {
    print('unknown source of census data!')
    return(db)
  }

  varNames <- tidycensus::load_variables(year, dataset, cache = TRUE) 
  varToQuery <- intersect(variables, varNames$name)
  if (length(variables)<length(varToQuery))
    print('some requested variables are not presented in census data')
  
  
  if(source == "decennial")
  {
    censusData <- tidycensus::get_decennial(geography = geography, state = state, county = county, 
                         variables = varToQuery, year = year)
    for (var in varToQuery)
    {
      censusVar <- censusData %>% dplyr::filter(variable == var) %>% dplyr::select(GEOID, !!quo_name(paste(var, year, sep = ".")) := value)
      db$observedData <- dplyr::left_join(db$observedData, censusVar, by = c("residence_census_tract"="GEOID"))
    }
  }
  else if(source =="acs")
  {
    censusData <- tidycensus::get_acs(geography = geography, state = state, county = county, 
                         variables = varToQuery, year = year)
    for (var in varToQuery)
    {
      censusVar <- censusData %>% dplyr::filter(variable == var) %>% dplyr::select(GEOID, !!quo_name(paste(var, year, sep = ".")) := estimate)
      db$observedData <- dplyr::left_join(db$observedData, censusVar, by = c("residence_census_tract"="GEOID"))
    }
  }
  
 return(db)
}