# HG changeset patch
# User bornea
# Date 1472058003 14400
# Node ID d66c62c6c56f72f1ef7a28e6956dcd82e4e0c5ed
# Parent  55990dea42fbab28228d1cf7e87803d80b6c64f2
Uploaded

diff -r 55990dea42fb -r d66c62c6c56f filter_fasta.py
--- a/filter_fasta.py	Tue Aug 16 20:48:46 2016 -0400
+++ b/filter_fasta.py	Wed Aug 24 13:00:03 2016 -0400
@@ -34,10 +34,13 @@
             break
         cnt += 1
     header = data[header_start]
-    if "Accession Number" in header:
-        prot_start = header.index("Accession Number")
-    elif "Accession" in header:
-        prot_start = header.index("Accession")
+    for i in header:
+        if i == "Accession":
+            prot_start = header.index("Accession")
+        if i == "Accession Number":
+            prot_start = header.index("Accession Number")
+        if i == "Main Accession":
+            prot_start = header.index("Main Accession")
     proteins = []
     for protein in data[header_start:]:
         if len(protein) > prot_start: