From 11de062c200962e588a7114f85e5f698cdbcd60a Mon Sep 17 00:00:00 2001 From: Jeremy Fincher Date: Tue, 26 Oct 2004 21:09:20 +0000 Subject: [PATCH] Allow isSeparator to be a function. --- src/utils.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/utils.py b/src/utils.py index 98ad51631..9c901b423 100755 --- a/src/utils.py +++ b/src/utils.py @@ -461,9 +461,13 @@ def itersplit(isSeparator, iterable, maxsplit=-1, yieldEmpty=False): """itersplit(isSeparator, iterable, maxsplit=-1, yieldEmpty=False) Splits an iterator based on a predicate isSeparator.""" + if isinstance(isSeparator, basestring): + f = lambda s: s == isSeparator + else: + f = isSeparator acc = [] for element in iterable: - if maxsplit == 0 or not isSeparator(element): + if maxsplit == 0 or not f(element): acc.append(element) else: maxsplit -= 1