Extract function to get fasta chunk name.

griffithlab · Nov 15, 2016 · fc2dd34 · fc2dd34
1 parent a639088
commit fc2dd34
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/pvacseq/lib/pipeline.py b/pvacseq/lib/pipeline.py
@@ -154,9 +154,12 @@ def generate_fasta(self):
 
     def split_fasta_file_path(self, split_start, split_end):
         basename    = os.path.join(self.tmp_dir, self.sample_name + "_" + str(self.peptide_sequence_length) + ".fa.split")
-        fasta_chunk = "%d-%d" % (split_start*2-1, split_end*2)
+        fasta_chunk = self.split_fasta_chunk(split_start, split_end)
         return "%s_%s" % (basename, fasta_chunk)
 
+    def split_fasta_chunk(self, split_start, split_end):
+        return "%d-%d" % (split_start*2-1, split_end*2)
+
     @abstractmethod
     def call_iedb_and_parse_outputs(self, chunks):
         pass
@@ -310,7 +313,7 @@ def __init__(self, **kwargs):
     def generate_fasta(self, chunks):
         status_message("Generating Variant Peptide FASTA and Key Files")
         for (split_start, split_end) in chunks:
-            fasta_chunk = "%d-%d" % (split_start*2-1, split_end*2)
+            fasta_chunk = self.split_fasta_chunk(split_start, split_end)
             split_tsv_file_path       = self.split_tsv_file_path(split_start, split_end)
             split_fasta_file_path     = self.split_fasta_file_path(split_start, split_end)
             if os.path.exists(split_fasta_file_path):
@@ -333,7 +336,7 @@ def generate_fasta(self, chunks):
     def call_iedb_and_parse_outputs(self, chunks):
         split_parsed_output_files = []
         for (split_start, split_end) in chunks:
-            fasta_chunk = "%d-%d" % (split_start*2-1, split_end*2)
+            fasta_chunk = self.split_fasta_chunk(split_start, split_end)
             for a in self.alleles:
                 for epl in self.epitope_lengths:
                     split_fasta_file_path = self.split_fasta_file_path(split_start, split_end)
@@ -402,7 +405,7 @@ def __init__(self, **kwargs):
     def generate_fasta(self, chunks):
         status_message("Generating Variant Peptide FASTA and Key Files")
         for (split_start, split_end) in chunks:
-            fasta_chunk = "%d-%d" % (split_start*2-1, split_end*2)
+            fasta_chunk = self.split_fasta_chunk(split_start, split_end)
             split_tsv_file_path       = self.split_tsv_file_path(split_start, split_end)
             split_fasta_file_path     = self.split_fasta_file_path(split_start, split_end)
             if os.path.exists(split_fasta_file_path):
@@ -425,7 +428,7 @@ def generate_fasta(self, chunks):
     def call_iedb_and_parse_outputs(self, chunks):
         split_parsed_output_files = []
         for (split_start, split_end) in chunks:
-            fasta_chunk = "%d-%d" % (split_start*2-1, split_end*2)
+            fasta_chunk = self.split_fasta_chunk(split_start, split_end)
             for a in self.alleles:
                 split_fasta_file_path = self.split_fasta_file_path(split_start, split_end)
                 split_iedb_output_files = []