R/pbiAddRowsToTable.R

In olfrost/poweRbi: A httr wrapper for the Power BI REST API

#' Add rows to a table.
#' @description Assumes that the schema of the data frame matches the table, it will try and insert the rows.
#' @param df The data frame you want to insert.
#' @importFrom httr add_headers POST content
#' @importFrom RJSONIO toJSON
#' @param guid Dataset Id.
#' @param tableName The friendly name of the table. The URL will be encoded.
#' @param truncate When TRUE, calls pbiTruncateDataset to clean the table before loading.
#' @examples
#' \dontrun{pbiAddRowsToTable(iris, guid = "abc", tableName = "Iris")}

pbiAddRowsToTable <- function(df, guid, tableName, truncate = FALSE){

  # Check access token.
  if(as.numeric(Sys.time()) > .pkgenv[["expires"]]){
    # Refresh the access token with an internal method.
    .pkgenv[["token"]] <- ._pbiRefresh()
  }

  # Construct URL for the POST request.
  url = URLencode(paste0("https://api.powerbi.com/v1.0/myorg/datasets/", guid, "/tables/", tableName, "/rows"))

  # Truncate the dataset before inserting if that's what the user wants.
  if(truncate) {
    stop("Truncating rows...")
    pbiTruncateDataset(guid, tableName)
  }

  # Convert all rows from the data frame into JSON.
  rows = paste0(apply(df, 1, RJSONIO::toJSON), collapse = ",")
  rows = paste0("{\"rows\":[", rows, "]}")

  # Make the request.
  l = httr::content(httr::POST(
    url = url,
    httr::add_headers(
      Authorization = paste("Bearer", .pkgenv[["token"]]),
      `Content-Type` = "application/json"
    ),
    body = rows))

  # Two-tiered IF statement for this particular statement.

  if(!is.raw(l)){

  # Error handling.
  if(exists("error", where = l)) {
    stop(paste(guid, "and", tableName, "produced an error message:", l$error$message, "."))
    return(NULL)
  }}

  cat(paste(nrow(df), "rows inserted into", tableName, "successfully!"))

}