R/convertColumnName.R

In tmk-c/myrlib: My R library for daily work

#' Convert column name between column names in DB and R style names
#'
#' @param column column name to be converted
#'
#' @return converted column name
#' @export
convertColumnName <- function(column) {

  # Valid columns
  columns.db <- c("ticker", "date", "adj_open", "adj_high", "adj_low",
                  "adj_close", "adj_volume", "dividend",
                  "open", "high", "low", "close", "volume", "last_updated")

  columns.r <- c("symbol", "date", "adj.open", "adj.high", "adj.low",
                 "adj.close", "adj.volume", "dividend",
                 "open", "high", "low", "close", "volume", "lastupdated")

  if (!column %in% c(columns.db, columns.r)) {
    stop(paste0(column, " is not valid column name."))
  }

  # Columns name maping
  columns.map <- data.table::data.table(
    db.name = columns.db,
    r.name  = columns.r
  )

  if (column %in% columns.db) {
    new.name <- columns.map[db.name == column]$r.name

  } else if (column %in% columns.r) {
    new.name <- columns.map[r.name == column]$db.name

  } else{
    stop(paste0(column, " is not a valid name."))
  }

  return (new.name)
}