From b2dadbfb614395d2c2827b43c91cc2f798f990b9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ren=C3=A9=20=27Necoro=27=20Neumann?= Date: Fri, 14 Aug 2009 16:16:49 +0200 Subject: Allow to pass the number of elements in a vector --- portato/eix/parser.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) (limited to 'portato/eix') diff --git a/portato/eix/parser.py b/portato/eix/parser.py index ebf2566..ecfb74c 100644 --- a/portato/eix/parser.py +++ b/portato/eix/parser.py @@ -63,8 +63,9 @@ def number (file, skip = False): return value -def vector (file, get_type, skip = False): - nelems = number(file) +def vector (file, get_type, skip = False, nelems = None): + if nelems is None: + nelems = number(file) if skip: for i in range(nelems): @@ -72,8 +73,11 @@ def vector (file, get_type, skip = False): else: return [get_type(file) for i in range(nelems)] -def typed_vector(type): - return partial(vector, get_type = type) +def typed_vector(type, nelems = None): + if nelems is None: + return partial(vector, get_type = type) + else: + return partial(vector, get_type = type, nelems = nelems) def string (file, skip = False): nelems = number(file) -- cgit v1.2.3