R/step_other.R

In orbital: Predict with 'tidymodels' Workflows in Databases

#' @export
orbital.step_other <- function(x, all_vars, ...) {
	objects <- x$objects

	objects <- objects[names(objects) %in% all_vars]

	if (length(objects) == 0) {
		return(NULL)
	}

	out <- character()
	for (col in names(objects)) {
		if (!objects[[col]]$collapse) {
			next
		}
		levels <- objects[[col]]$keep
		levels <- glue::glue("\"{levels}\"")
		levels <- paste(levels, collapse = ", ")
		levels <- glue::glue("c({levels})")
		out[[col]] <- glue::glue(
			"dplyr::if_else(is.na({col}), NA, dplyr::if_else({col} %in% {levels}, {col}, \"{objects[[col]]$other}\"))"
		)
	}
	out
}