From 23a383407f4f1980e2a3b8c60bc584193cf37225 Mon Sep 17 00:00:00 2001
From: ljmciver <lauren.j.mciver@gmail.com>
Date: Tue, 15 Mar 2022 12:18:21 -0400
Subject: [PATCH] Add catch for files with less then N lines in function
 get_first_n_seq_identifiers

---
 kneaddata/utilities.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/kneaddata/utilities.py b/kneaddata/utilities.py
index b09b47a..29bc19c 100644
--- a/kneaddata/utilities.py
+++ b/kneaddata/utilities.py
@@ -350,7 +350,10 @@ def get_first_n_seq_identifiers(file,n):
     first_seq_identifiers=[]
     # Getting first nth seq identifier
     while(count<n):
-        lines=next(all_lines)
+        try:
+            lines=next(all_lines)
+        except StopIteration:
+            sys.exit("ERROR: {0} has less then {1} lines".format(file,n))
         first_seq_identifiers.append(lines[0])
         count+=1
     return first_seq_identifiers