From 1ddee2bddcdee1f6afa7992a61b13ce7079711f1 Mon Sep 17 00:00:00 2001
From: Ulthran <ctbushman@gmail.com>
Date: Thu, 6 Jun 2024 14:46:01 -0400
Subject: [PATCH] Allow .fna host extension as well

---
 sbx_mapping.smk | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/sbx_mapping.smk b/sbx_mapping.smk
index 423cab0..a5973f2 100644
--- a/sbx_mapping.smk
+++ b/sbx_mapping.smk
@@ -28,6 +28,9 @@ else:
     if not GenomeFiles:
         GenomeFiles = [f for f in Cfg["sbx_mapping"]["genomes_fp"].glob("*.fa")]
         HOST_FILE_EXT = ".fa"
+    if not GenomeFiles:
+        GenomeFiles = [f for f in Cfg["sbx_mapping"]["genomes_fp"].glob("*.fna")]
+        HOST_FILE_EXT = ".fna"
     GenomeSegments = {
         PurePath(g.name).stem: read_seq_ids(Cfg["sbx_mapping"]["genomes_fp"] / g)
         for g in GenomeFiles