From 20c4028f567f0d4b5df1b80f1b6bea1bc99887b4 Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Mon, 25 Apr 2022 08:08:08 +0300
Subject: `take`: function to select a few items from an iterable

To avoid processing all the items in an iterable, the `take` function
is added in this commit. It realised a limited number (specified at
call time) of items from the iterable given.
---
 quality_control/parsing.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

(limited to 'quality_control')

diff --git a/quality_control/parsing.py b/quality_control/parsing.py
index ac53642..e9bd5f7 100644
--- a/quality_control/parsing.py
+++ b/quality_control/parsing.py
@@ -3,7 +3,7 @@
 import csv
 from enum import Enum
 from functools import reduce
-from typing import Iterator, Generator
+from typing import Iterable, Generator
 
 import quality_control.average as avg
 import quality_control.standard_error as se
@@ -124,3 +124,15 @@ def parse_errors(filepath: str, filetype: FileType, strains: list,
     return (
         error for error in __errors(filepath, filetype, strains, seek_pos)
         if error is not None)
+
+def take(iterable: Iterable, num: int) -> list:
+    """Take at most `num` items from `iterable`."""
+    iterator = iter(iterable)
+    items = []
+    try:
+        for i in range(0, num):
+            items.append(next(iterator))
+
+        return items
+    except StopIteration:
+        return items
-- 
cgit v1.2.3