Make ncbi_download_genome() robust to multiple links.

stitam · Aug 22, 2024 · 20ecde8 · 20ecde8
1 parent 2237883
commit 20ecde8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/R/ncbi_download_genome.R b/R/ncbi_download_genome.R
@@ -60,13 +60,13 @@ ncbi_download_genome <- function(query,
       stop("'ncbi_uid' object must contain NCBI Assembly UIDs.")
     }
   } else if ("ncbi_uid_link" %in% class(query)) {
-    if (names(query)[2] == "assembly") {
+    if (names(query)[length(names(query))] == "assembly") {
       assembly_uid <- unique(query$assembly)
     } else {
       stop("'ncbi_uid_link' object must contain links to NCBI Assembly UIDs.")
     }
   } else if ("ncbi_link" %in% class(query)) {
-    if (names(query)[2] == "assembly") {
+    if (names(query)[length(names(query))] == "assembly") {
       assembly_uid <- ncbi_get_uid(query$assembly, db = "assembly")$uid
     } else {
       stop("'ncbi_link' object must contain links to NCBI Assembly IDs.")