factoring out genID call

DeclareDesign · Oct 20, 2017 · ed14837 · ed14837
1 parent 151a9a9
commit ed14837
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/R/fabricate.R b/R/fabricate.R
@@ -127,13 +127,18 @@ fabricate_data_single_level <- function(data = NULL,
         ", you must provide a single number to N."
       )
     }
-    data <- genID(data.frame(), ifelse(is.null(ID_label), "ID", ID_label), N)
-  } else if (!is.null(data)){
-    if (existing_ID == FALSE) {
-      data <- genID(data, ifelse(is.null(ID_label), "ID", ID_label))
-    }
+    data <- data.frame()
+    existing_ID <- FALSE
+  } else if(!is.null(data)){
+    N <- nrow(data)
+  }
+
+  if (!existing_ID) {
+    if(is.null(ID_label)) ID_label <- "ID"
+    data <- genID(data, ID_label, N)
   }
 
+
   fab(data, options)
 }