Create a dictionary of our trained matrix id's

1 files changed, 8 insertions, 4 deletions

diff --git a/multifasta_to_otu.py b/multifasta_to_otu.py
index e313e4a..ef7f5f6 100755
--- a/multifasta_to_otu.py
+++ b/multifasta_to_otu.py
@@ -76,12 +76,16 @@ def main():
   results = pool.map(quikr_call, fasta_list)
 
   # Create a dictionary and load up our keys
-  
+  records = []
+
+  trained_matrix_headers = open(args.trained_matrix, "rU")
+  for record in SeqIO.parse(trained_matrix_headers, "fasta"):
+    records.append((record.id, 0))
+
+  records = dict(records)
+
 
-  for fasta in fasta_list:
-    fasta_file = open(input_directory + fasta, "rU")
 
-    for record in SeqIO.parse(fasta_file, "fasta") :
 
   # load the keys with values from each fasta result