diff options
Diffstat (limited to 'Lib')
-rw-r--r-- | Lib/collections.py | 14 | ||||
-rw-r--r-- | Lib/test/test_collections.py | 11 |
2 files changed, 23 insertions, 2 deletions
diff --git a/Lib/collections.py b/Lib/collections.py index 0fddb97..8d49cd5 100644 --- a/Lib/collections.py +++ b/Lib/collections.py @@ -16,7 +16,7 @@ from itertools import repeat as _repeat, chain as _chain, starmap as _starmap, i ### namedtuple ################################################################################ -def namedtuple(typename, field_names, verbose=False): +def namedtuple(typename, field_names, verbose=False, rename=False): """Returns a new subclass of tuple with named fields. >>> Point = namedtuple('Point', 'x y') @@ -45,6 +45,16 @@ def namedtuple(typename, field_names, verbose=False): if isinstance(field_names, basestring): field_names = field_names.replace(',', ' ').split() # names separated by whitespace and/or commas field_names = tuple(map(str, field_names)) + if rename: + names = list(field_names) + seen = set() + for i, name in enumerate(names): + if (not all(c.isalnum() or c=='_' for c in name) or _iskeyword(name) + or not name or name[0].isdigit() or name.startswith('_') + or name in seen): + names[i] = '_%d' % (i+1) + seen.add(name) + field_names = tuple(names) for name in (typename,) + field_names: if not all(c.isalnum() or c=='_' for c in name): raise ValueError('Type names and field names can only contain alphanumeric characters and underscores: %r' % name) @@ -54,7 +64,7 @@ def namedtuple(typename, field_names, verbose=False): raise ValueError('Type names and field names cannot start with a number: %r' % name) seen_names = set() for name in field_names: - if name.startswith('_'): + if name.startswith('_') and not rename: raise ValueError('Field names cannot start with an underscore: %r' % name) if name in seen_names: raise ValueError('Encountered duplicate field name: %r' % name) diff --git a/Lib/test/test_collections.py b/Lib/test/test_collections.py index 17ec30c..4901eae 100644 --- a/Lib/test/test_collections.py +++ b/Lib/test/test_collections.py @@ -44,6 +44,17 @@ class TestNamedTuple(unittest.TestCase): self.assertRaises(TypeError, Point._make, [11]) # catch too few args self.assertRaises(TypeError, Point._make, [11, 22, 33]) # catch too many args + def test_name_fixer(self): + for spec, renamed in [ + [('efg', 'g%hi'), ('efg', '_2')], # field with non-alpha char + [('abc', 'class'), ('abc', '_2')], # field has keyword + [('8efg', '9ghi'), ('_1', '_2')], # field starts with digit + [('abc', '_efg'), ('abc', '_2')], # field with leading underscore + [('abc', 'efg', 'efg', 'ghi'), ('abc', 'efg', '_3', 'ghi')], # duplicate field + [('abc', '', 'x'), ('abc', '_2', 'x')], # fieldname is a space + ]: + self.assertEqual(namedtuple('NT', spec, rename=True)._fields, renamed) + def test_instance(self): Point = namedtuple('Point', 'x y') p = Point(11, 22) |