diff scripts/pogs.py @ 10:1a728cb1da31 draft

planemo upload for repository https://github.com/abims-sbr/adaptsearch commit b7a3030ea134b5dfad89b1a869db659d72d1145c
author abims-sbr
date Wed, 28 Feb 2018 10:37:38 -0500
parents b19ed7395dcc
children
line wrap: on
line diff
--- a/scripts/pogs.py	Wed Feb 28 06:04:15 2018 -0500
+++ b/scripts/pogs.py	Wed Feb 28 10:37:38 2018 -0500
@@ -11,7 +11,7 @@
 BETA VERSION
 """
 
-import os, argparse
+import os, argparse, itertools
 import numpy as np
 import pandas as pd
 
@@ -59,9 +59,7 @@
     def getPairwiseCouple(pairwiseFile):        
         list_pairwises_2sp = []
         with open(pairwiseFile, "r") as file:
-            while (1):  # Ugly !
-                name, sequence, name2, sequence2 = file.readline(), file.readline(), file.readline(), file.readline()
-                if not name: break # Use assert ?
+            for name, sequence, name2, sequence2 in itertools.izip_longest(*[file]*4):            
                 # One locus every two lines (one pairwise couple = 4 lines) : header + sequence
                 locus1 = Locus(name, sequence)
                 locus2 = Locus(name2, sequence2)