| OLD | NEW |
| (Empty) |
| 1 #!/usr/bin/env python | |
| 2 # | |
| 3 # $Id: compat.py 712 2010-10-20 15:25:57Z g.rodola $ | |
| 4 # | |
| 5 | |
| 6 """Module which provides compatibility with older Python versions.""" | |
| 7 | |
| 8 from operator import itemgetter as _itemgetter | |
| 9 from keyword import iskeyword as _iskeyword | |
| 10 import sys as _sys | |
| 11 | |
| 12 | |
| 13 def namedtuple(typename, field_names, verbose=False, rename=False): | |
| 14 """A collections.namedtuple implementation written in Python | |
| 15 to support Python versions < 2.6. | |
| 16 | |
| 17 Taken from: http://code.activestate.com/recipes/500261/ | |
| 18 """ | |
| 19 # Parse and validate the field names. Validation serves two | |
| 20 # purposes, generating informative error messages and preventing | |
| 21 # template injection attacks. | |
| 22 if isinstance(field_names, basestring): | |
| 23 # names separated by whitespace and/or commas | |
| 24 field_names = field_names.replace(',', ' ').split() | |
| 25 field_names = tuple(map(str, field_names)) | |
| 26 if rename: | |
| 27 names = list(field_names) | |
| 28 seen = set() | |
| 29 for i, name in enumerate(names): | |
| 30 if (not min(c.isalnum() or c=='_' for c in name) or _iskeyword(name) | |
| 31 or not name or name[0].isdigit() or name.startswith('_') | |
| 32 or name in seen): | |
| 33 names[i] = '_%d' % i | |
| 34 seen.add(name) | |
| 35 field_names = tuple(names) | |
| 36 for name in (typename,) + field_names: | |
| 37 if not min(c.isalnum() or c=='_' for c in name): | |
| 38 raise ValueError('Type names and field names can only contain ' \ | |
| 39 'alphanumeric characters and underscores: %r' | |
| 40 % name) | |
| 41 if _iskeyword(name): | |
| 42 raise ValueError('Type names and field names cannot be a keyword: %r
' \ | |
| 43 % name) | |
| 44 if name[0].isdigit(): | |
| 45 raise ValueError('Type names and field names cannot start with a ' \ | |
| 46 'number: %r' % name) | |
| 47 seen_names = set() | |
| 48 for name in field_names: | |
| 49 if name.startswith('_') and not rename: | |
| 50 raise ValueError('Field names cannot start with an underscore: %r' | |
| 51 % name) | |
| 52 if name in seen_names: | |
| 53 raise ValueError('Encountered duplicate field name: %r' % name) | |
| 54 seen_names.add(name) | |
| 55 | |
| 56 # Create and fill-in the class template | |
| 57 numfields = len(field_names) | |
| 58 # tuple repr without parens or quotes | |
| 59 argtxt = repr(field_names).replace("'", "")[1:-1] | |
| 60 reprtxt = ', '.join('%s=%%r' % name for name in field_names) | |
| 61 template = '''class %(typename)s(tuple): | |
| 62 '%(typename)s(%(argtxt)s)' \n | |
| 63 __slots__ = () \n | |
| 64 _fields = %(field_names)r \n | |
| 65 def __new__(_cls, %(argtxt)s): | |
| 66 return _tuple.__new__(_cls, (%(argtxt)s)) \n | |
| 67 @classmethod | |
| 68 def _make(cls, iterable, new=tuple.__new__, len=len): | |
| 69 'Make a new %(typename)s object from a sequence or iterable' | |
| 70 result = new(cls, iterable) | |
| 71 if len(result) != %(numfields)d: | |
| 72 raise TypeError('Expected %(numfields)d arguments, got %%d' %% l
en(result)) | |
| 73 return result \n | |
| 74 def __repr__(self): | |
| 75 return '%(typename)s(%(reprtxt)s)' %% self \n | |
| 76 def _asdict(self): | |
| 77 'Return a new dict which maps field names to their values' | |
| 78 return dict(zip(self._fields, self)) \n | |
| 79 def _replace(_self, **kwds): | |
| 80 'Return a new %(typename)s object replacing specified fields with ne
w values' | |
| 81 result = _self._make(map(kwds.pop, %(field_names)r, _self)) | |
| 82 if kwds: | |
| 83 raise ValueError('Got unexpected field names: %%r' %% kwds.keys(
)) | |
| 84 return result \n | |
| 85 def __getnewargs__(self): | |
| 86 return tuple(self) \n\n''' % locals() | |
| 87 for i, name in enumerate(field_names): | |
| 88 template += ' %s = _property(_itemgetter(%d))\n' % (name, i) | |
| 89 if verbose: | |
| 90 print template | |
| 91 | |
| 92 # Execute the template string in a temporary namespace | |
| 93 namespace = dict(_itemgetter=_itemgetter, __name__='namedtuple_%s' % typenam
e, | |
| 94 _property=property, _tuple=tuple) | |
| 95 try: | |
| 96 exec template in namespace | |
| 97 except SyntaxError, e: | |
| 98 raise SyntaxError(e.message + ':\n' + template) | |
| 99 result = namespace[typename] | |
| 100 | |
| 101 # For pickling to work, the __module__ variable needs to be set | |
| 102 # to the frame where the named tuple is created. Bypass this | |
| 103 # step in enviroments where sys._getframe is not defined (Jython | |
| 104 # for example) or sys._getframe is not defined for arguments | |
| 105 # greater than 0 (IronPython). | |
| 106 try: | |
| 107 result.__module__ = _sys._getframe(1).f_globals.get('__name__', '__main_
_') | |
| 108 except (AttributeError, ValueError): | |
| 109 pass | |
| 110 | |
| 111 return result | |
| OLD | NEW |