Track new host file ext

sunbeam-labs · Jun 5, 2024 · 179092b · 179092b
1 parent 684336d
commit 179092b
Showing 1 changed file with 24 additions and 26 deletions.
diff --git a/sbx_mapping.smk b/sbx_mapping.smk
@@ -15,29 +15,26 @@ def get_mapping_path() -> Path:
 
 
 SBX_MAPPING_VERSION = open(get_mapping_path() / "VERSION").read().strip()
-
-try:
-    GenomeFiles
-    GenomeSegments
-except NameError:
-    sys.stderr.write("sbx_mapping::INFO Collecting target genomes... ")
-    if (
-        Cfg["sbx_mapping"]["genomes_fp"] == Cfg["all"]["root"]
-        or not Cfg["sbx_mapping"]["genomes_fp"]
-    ):
-        GenomeFiles = []
-        GenomeSegments = {}
-    else:
-        GenomeFiles = [f for f in Cfg["sbx_mapping"]["genomes_fp"].glob("*.fasta")]
-        if not GenomeFiles:
-            GenomeFiles = [f for f in Cfg["sbx_mapping"]["genomes_fp"].glob("*.fa")]
-        GenomeSegments = {
-            PurePath(g.name).stem: read_seq_ids(Cfg["sbx_mapping"]["genomes_fp"] / g)
-            for g in GenomeFiles
-        }
-        GenomeFiles = {PurePath(g.name).stem: g for g in GenomeFiles}
-    sys.stderr.write("done.\n")
-    sys.stderr.write(f"sbx_mapping::INFO Genome files found: {str(GenomeFiles)}\n")
+HOST_FILE_EXT = ".fasta"
+sys.stderr.write("sbx_mapping::INFO Collecting target genomes... ")
+if (
+    Cfg["sbx_mapping"]["genomes_fp"] == Cfg["all"]["root"]
+    or not Cfg["sbx_mapping"]["genomes_fp"]
+):
+    GenomeFiles = []
+    GenomeSegments = {}
+else:
+    GenomeFiles = [f for f in Cfg["sbx_mapping"]["genomes_fp"].glob("*.fasta")]
+    if not GenomeFiles:
+        GenomeFiles = [f for f in Cfg["sbx_mapping"]["genomes_fp"].glob("*.fa")]
+        HOST_FILE_EXT = ".fa"
+    GenomeSegments = {
+        PurePath(g.name).stem: read_seq_ids(Cfg["sbx_mapping"]["genomes_fp"] / g)
+        for g in GenomeFiles
+    }
+    GenomeFiles = {PurePath(g.name).stem: g for g in GenomeFiles}
+sys.stderr.write("done.\n")
+sys.stderr.write(f"sbx_mapping::INFO Genome files found: {str(GenomeFiles)}\n")
 
 
 try:
@@ -83,10 +80,10 @@ rule all_mapping:
 
 rule build_genome_index:
     input:
-        Cfg["sbx_mapping"]["genomes_fp"] / "{genome}.fasta",
+        Cfg["sbx_mapping"]["genomes_fp"] / ("{genome}" + HOST_FILE_EXT),
     output:
         [
-            Cfg["sbx_mapping"]["genomes_fp"] / ("{genome}.fasta." + ext)
+            Cfg["sbx_mapping"]["genomes_fp"] / ("{genome}" + HOST_FILE_EXT + "." + ext)
             for ext in ["amb", "ann", "bwt", "pac", "sa"]
         ],
     benchmark:
@@ -113,6 +110,7 @@ rule align_to_genome:
         LOG_FP / "align_to_genome_{genome}_{sample}.log",
     params:
         index_fp=Cfg["sbx_mapping"]["genomes_fp"],
+        host_file_ext=HOST_FILE_EXT,
     threads: 4
     conda:
         "envs/sbx_mapping_env.yml"
@@ -121,7 +119,7 @@ rule align_to_genome:
     shell:
         """
         bwa mem -M -t {threads} \
-        {params.index_fp}/{wildcards.genome}.fasta \
+        {params.index_fp}/{wildcards.genome}{params.host_file_ext} \
         {input.reads} -o {output} \
         2>&1 | tee {log}
         """