R/sample_tweet.R

In revodavid/rlangtip: What the Package Does (One Line, Title Case)

weight_determination <- function(weighted = F) {
  if (weighted == F) {
    return(NULL)
  } else {
    weighted <- score()
  }
}


sample_tweet <- function(number_of_tweets = 1, holiday = FALSE, categories = NULL, weighted = FALSE) {
  tweets <- read_excel("data_cleaned.xlsx")

  if (length(categories) != 0) {
    tweets <- tweets[(tweets$category %in% categories), ]
  }

  if (!holiday) {
    tweets <- tweets[tweets$category != "holiday", ]
  }

  if (weighted) {
    n <- sample_n(tweets, number_of_tweets, weight = score)
  } else {
    n <- sample_n(tweets, number_of_tweets, weight = NULL)
  }

  return(n)
}