Merge pull request #359 from USEPA/ref-internal-files

add provider option to DR

R/DataDiscoveryRetrieval.R

@@ -43,6 +43,7 @@
 #' @param statecode FIPS state alpha code that identifies a state (e.g. statecode = "DE" for Delaware)
 #' @param organization A string of letters and/or numbers (some additional characters also possible) used to signify an organization with data in the Water Quality Portal
 #' @param project A string of letters and/or numbers (some additional characters also possible) used to signify a project with data in the Water Quality Portal
+#' @param providers Leave blank to include all, or specify "STEWARDS", "STORET" (i.e., WQX), and/or "NWIS".
 #' @param applyautoclean Logical, defaults to TRUE. Applies TADA_AutoClean function on the returned data profile.
 #'
 #' @return TADA-compatible dataframe
@@ -133,6 +134,16 @@
 #'   "WNENVDPT_WQX",
 #'   "PUEBLO_POJOAQUE"
 #' ))
+#' 
+#' # query only NWIS data for a 10 year period in CT
+#' tada10 = TADA_DataRetrieval(
+#' startDate = "2013-01-01", 
+#' endDate = "2022-12-31", 
+#' sampleMedia = c("Water", "water"),
+#' statecode = "CT", # consider downloading only 1 state at a time
+#' providers = "NWIS",
+#' applyautoclean = FALSE
+#' )
 #' }
 #'
 TADA_DataRetrieval <- function(startDate = "null",
@@ -147,6 +158,7 @@ TADA_DataRetrieval <- function(startDate = "null",
                                statecode = "null",
                                organization = "null",
                                project = "null",
+                               providers = "null",
                                applyautoclean = TRUE) {
   # Set query parameters
   WQPquery <- list()
@@ -215,6 +227,12 @@ TADA_DataRetrieval <- function(startDate = "null",
   } else if (project != "null") {
     WQPquery <- c(WQPquery, project = project)
   }
+
+  if (length(providers) > 1) {
+    WQPquery <- c(WQPquery, providers = list(providers))
+  } else if (providers != "null") {
+    WQPquery <- c(WQPquery, providers = providers)
+  }
 
   if (length(organization) > 1) {
     WQPquery <- c(WQPquery, organization = list(organization))