Code cleanups
This commit is contained in:
parent
1eda5c9a83
commit
491f7070f1
|
@ -601,47 +601,49 @@ Example:
|
||||||
|
|
||||||
>>> Point = namedtuple('Point', ['x', 'y'], verbose=True)
|
>>> Point = namedtuple('Point', ['x', 'y'], verbose=True)
|
||||||
class Point(tuple):
|
class Point(tuple):
|
||||||
'Point(x, y)'
|
'Point(x, y)'
|
||||||
<BLANKLINE>
|
<BLANKLINE>
|
||||||
__slots__ = ()
|
__slots__ = ()
|
||||||
<BLANKLINE>
|
<BLANKLINE>
|
||||||
_fields = ('x', 'y')
|
_fields = ('x', 'y')
|
||||||
<BLANKLINE>
|
<BLANKLINE>
|
||||||
def __new__(_cls, x, y):
|
def __new__(_cls, x, y):
|
||||||
'Create a new instance of Point(x, y)'
|
'Create a new instance of Point(x, y)'
|
||||||
return _tuple.__new__(_cls, (x, y))
|
return _tuple.__new__(_cls, (x, y))
|
||||||
<BLANKLINE>
|
<BLANKLINE>
|
||||||
@classmethod
|
@classmethod
|
||||||
def _make(cls, iterable, new=tuple.__new__, len=len):
|
def _make(cls, iterable, new=tuple.__new__, len=len):
|
||||||
'Make a new Point object from a sequence or iterable'
|
'Make a new Point object from a sequence or iterable'
|
||||||
result = new(cls, iterable)
|
result = new(cls, iterable)
|
||||||
if len(result) != 2:
|
if len(result) != 2:
|
||||||
raise TypeError('Expected 2 arguments, got %d' % len(result))
|
raise TypeError('Expected 2 arguments, got %d' % len(result))
|
||||||
return result
|
return result
|
||||||
<BLANKLINE>
|
<BLANKLINE>
|
||||||
def __repr__(self):
|
def __repr__(self):
|
||||||
'Return a nicely formatted representation string'
|
'Return a nicely formatted representation string'
|
||||||
return 'Point(x=%r, y=%r)' % self
|
return 'Point(x=%r, y=%r)' % self
|
||||||
<BLANKLINE>
|
<BLANKLINE>
|
||||||
def _asdict(self):
|
def _asdict(self):
|
||||||
'Return a new OrderedDict which maps field names to their values'
|
'Return a new OrderedDict which maps field names to their values'
|
||||||
return OrderedDict(zip(self._fields, self))
|
return OrderedDict(zip(self._fields, self))
|
||||||
<BLANKLINE>
|
<BLANKLINE>
|
||||||
__dict__ = property(_asdict)
|
__dict__ = property(_asdict)
|
||||||
<BLANKLINE>
|
<BLANKLINE>
|
||||||
def _replace(_self, **kwds):
|
def _replace(_self, **kwds):
|
||||||
'Return a new Point object replacing specified fields with new values'
|
'Return a new Point object replacing specified fields with new values'
|
||||||
result = _self._make(map(kwds.pop, ('x', 'y'), _self))
|
result = _self._make(map(kwds.pop, ('x', 'y'), _self))
|
||||||
if kwds:
|
if kwds:
|
||||||
raise ValueError('Got unexpected field names: %r' % kwds.keys())
|
raise ValueError('Got unexpected field names: %r' % kwds.keys())
|
||||||
return result
|
return result
|
||||||
<BLANKLINE>
|
<BLANKLINE>
|
||||||
def __getnewargs__(self):
|
def __getnewargs__(self):
|
||||||
'Return self as a plain tuple. Used by copy and pickle.'
|
'Return self as a plain tuple. Used by copy and pickle.'
|
||||||
return tuple(self)
|
return tuple(self)
|
||||||
|
<BLANKLINE>
|
||||||
|
x = _property(_itemgetter(0), doc='Alias for field number 0')
|
||||||
|
<BLANKLINE>
|
||||||
|
y = _property(_itemgetter(1), doc='Alias for field number 1')
|
||||||
<BLANKLINE>
|
<BLANKLINE>
|
||||||
x = _property(_itemgetter(0), doc='Alias for field number 0')
|
|
||||||
y = _property(_itemgetter(1), doc='Alias for field number 1')
|
|
||||||
|
|
||||||
>>> p = Point(11, y=22) # instantiate with positional or keyword arguments
|
>>> p = Point(11, y=22) # instantiate with positional or keyword arguments
|
||||||
>>> p[0] + p[1] # indexable like the plain tuple (11, 22)
|
>>> p[0] + p[1] # indexable like the plain tuple (11, 22)
|
||||||
|
|
|
@ -234,10 +234,60 @@ class OrderedDict(dict):
|
||||||
### namedtuple
|
### namedtuple
|
||||||
################################################################################
|
################################################################################
|
||||||
|
|
||||||
|
_class_template = '''\
|
||||||
|
class {typename}(tuple):
|
||||||
|
'{typename}({arg_list})'
|
||||||
|
|
||||||
|
__slots__ = ()
|
||||||
|
|
||||||
|
_fields = {field_names!r}
|
||||||
|
|
||||||
|
def __new__(_cls, {arg_list}):
|
||||||
|
'Create new instance of {typename}({arg_list})'
|
||||||
|
return _tuple.__new__(_cls, ({arg_list}))
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def _make(cls, iterable, new=tuple.__new__, len=len):
|
||||||
|
'Make a new {typename} object from a sequence or iterable'
|
||||||
|
result = new(cls, iterable)
|
||||||
|
if len(result) != {num_fields:d}:
|
||||||
|
raise TypeError('Expected {num_fields:d} arguments, got %d' % len(result))
|
||||||
|
return result
|
||||||
|
|
||||||
|
def __repr__(self):
|
||||||
|
'Return a nicely formatted representation string'
|
||||||
|
return '{typename}({repr_fmt})' % self
|
||||||
|
|
||||||
|
def _asdict(self):
|
||||||
|
'Return a new OrderedDict which maps field names to their values'
|
||||||
|
return OrderedDict(zip(self._fields, self))
|
||||||
|
|
||||||
|
__dict__ = property(_asdict)
|
||||||
|
|
||||||
|
def _replace(_self, **kwds):
|
||||||
|
'Return a new {typename} object replacing specified fields with new values'
|
||||||
|
result = _self._make(map(kwds.pop, {field_names!r}, _self))
|
||||||
|
if kwds:
|
||||||
|
raise ValueError('Got unexpected field names: %r' % kwds.keys())
|
||||||
|
return result
|
||||||
|
|
||||||
|
def __getnewargs__(self):
|
||||||
|
'Return self as a plain tuple. Used by copy and pickle.'
|
||||||
|
return tuple(self)
|
||||||
|
|
||||||
|
{field_defs}
|
||||||
|
'''
|
||||||
|
|
||||||
|
_repr_template = '{name}=%r'
|
||||||
|
|
||||||
|
_field_template = '''\
|
||||||
|
{name} = _property(_itemgetter({index:d}), doc='Alias for field number {index:d}')
|
||||||
|
'''
|
||||||
|
|
||||||
def namedtuple(typename, field_names, verbose=False, rename=False):
|
def namedtuple(typename, field_names, verbose=False, rename=False):
|
||||||
"""Returns a new subclass of tuple with named fields.
|
"""Returns a new subclass of tuple with named fields.
|
||||||
|
|
||||||
>>> Point = namedtuple('Point', 'x y')
|
>>> Point = namedtuple('Point', ['x', 'y'])
|
||||||
>>> Point.__doc__ # docstring for the new class
|
>>> Point.__doc__ # docstring for the new class
|
||||||
'Point(x, y)'
|
'Point(x, y)'
|
||||||
>>> p = Point(11, y=22) # instantiate with positional args or keywords
|
>>> p = Point(11, y=22) # instantiate with positional args or keywords
|
||||||
|
@ -267,8 +317,11 @@ def namedtuple(typename, field_names, verbose=False, rename=False):
|
||||||
names = list(field_names)
|
names = list(field_names)
|
||||||
seen = set()
|
seen = set()
|
||||||
for i, name in enumerate(names):
|
for i, name in enumerate(names):
|
||||||
if (not all(c.isalnum() or c=='_' for c in name) or _iskeyword(name)
|
if (not all(c.isalnum() or c=='_' for c in name)
|
||||||
or not name or name[0].isdigit() or name.startswith('_')
|
or _iskeyword(name)
|
||||||
|
or not name
|
||||||
|
or name[0].isdigit()
|
||||||
|
or name.startswith('_')
|
||||||
or name in seen):
|
or name in seen):
|
||||||
names[i] = '_%d' % i
|
names[i] = '_%d' % i
|
||||||
seen.add(name)
|
seen.add(name)
|
||||||
|
@ -280,60 +333,34 @@ def namedtuple(typename, field_names, verbose=False, rename=False):
|
||||||
raise ValueError('Type names and field names cannot be a keyword: %r' % name)
|
raise ValueError('Type names and field names cannot be a keyword: %r' % name)
|
||||||
if name[0].isdigit():
|
if name[0].isdigit():
|
||||||
raise ValueError('Type names and field names cannot start with a number: %r' % name)
|
raise ValueError('Type names and field names cannot start with a number: %r' % name)
|
||||||
seen_names = set()
|
seen = set()
|
||||||
for name in field_names:
|
for name in field_names:
|
||||||
if name.startswith('_') and not rename:
|
if name.startswith('_') and not rename:
|
||||||
raise ValueError('Field names cannot start with an underscore: %r' % name)
|
raise ValueError('Field names cannot start with an underscore: %r' % name)
|
||||||
if name in seen_names:
|
if name in seen:
|
||||||
raise ValueError('Encountered duplicate field name: %r' % name)
|
raise ValueError('Encountered duplicate field name: %r' % name)
|
||||||
seen_names.add(name)
|
seen.add(name)
|
||||||
|
|
||||||
# Create and fill-in the class template
|
# Fill-in the class template
|
||||||
numfields = len(field_names)
|
class_definition = _class_template.format(
|
||||||
argtxt = repr(field_names).replace("'", "")[1:-1] # tuple repr without parens or quotes
|
typename = typename,
|
||||||
reprtxt = ', '.join('%s=%%r' % name for name in field_names)
|
field_names = tuple(field_names),
|
||||||
template = '''class %(typename)s(tuple):
|
num_fields = len(field_names),
|
||||||
'%(typename)s(%(argtxt)s)' \n
|
arg_list = repr(tuple(field_names)).replace("'", "")[1:-1],
|
||||||
__slots__ = () \n
|
repr_fmt = ', '.join(_repr_template.format(name=name) for name in field_names),
|
||||||
_fields = %(field_names)r \n
|
field_defs = '\n'.join(_field_template.format(index=index, name=name)
|
||||||
def __new__(_cls, %(argtxt)s):
|
for index, name in enumerate(field_names))
|
||||||
'Create new instance of %(typename)s(%(argtxt)s)'
|
)
|
||||||
return _tuple.__new__(_cls, (%(argtxt)s)) \n
|
|
||||||
@classmethod
|
|
||||||
def _make(cls, iterable, new=tuple.__new__, len=len):
|
|
||||||
'Make a new %(typename)s object from a sequence or iterable'
|
|
||||||
result = new(cls, iterable)
|
|
||||||
if len(result) != %(numfields)d:
|
|
||||||
raise TypeError('Expected %(numfields)d arguments, got %%d' %% len(result))
|
|
||||||
return result \n
|
|
||||||
def __repr__(self):
|
|
||||||
'Return a nicely formatted representation string'
|
|
||||||
return '%(typename)s(%(reprtxt)s)' %% self \n
|
|
||||||
def _asdict(self):
|
|
||||||
'Return a new OrderedDict which maps field names to their values'
|
|
||||||
return OrderedDict(zip(self._fields, self)) \n
|
|
||||||
__dict__ = property(_asdict) \n
|
|
||||||
def _replace(_self, **kwds):
|
|
||||||
'Return a new %(typename)s object replacing specified fields with new values'
|
|
||||||
result = _self._make(map(kwds.pop, %(field_names)r, _self))
|
|
||||||
if kwds:
|
|
||||||
raise ValueError('Got unexpected field names: %%r' %% kwds.keys())
|
|
||||||
return result \n
|
|
||||||
def __getnewargs__(self):
|
|
||||||
'Return self as a plain tuple. Used by copy and pickle.'
|
|
||||||
return tuple(self) \n\n''' % locals()
|
|
||||||
for i, name in enumerate(field_names):
|
|
||||||
template += " %s = _property(_itemgetter(%d), doc='Alias for field number %d')\n" % (name, i, i)
|
|
||||||
if verbose:
|
if verbose:
|
||||||
print template
|
print class_definition
|
||||||
|
|
||||||
# Execute the template string in a temporary namespace and
|
# Execute the template string in a temporary namespace and
|
||||||
# support tracing utilities by setting a value for frame.f_globals['__name__']
|
# support tracing utilities by setting a value for frame.f_globals['__name__']
|
||||||
namespace = dict(_itemgetter=_itemgetter, __name__='namedtuple_%s' % typename,
|
namespace = dict(_itemgetter=_itemgetter, __name__='namedtuple_%s' % typename,
|
||||||
OrderedDict=OrderedDict, _property=property, _tuple=tuple)
|
OrderedDict=OrderedDict, _property=property, _tuple=tuple)
|
||||||
try:
|
try:
|
||||||
exec template in namespace
|
exec class_definition in namespace
|
||||||
except SyntaxError, e:
|
except SyntaxError as e:
|
||||||
raise SyntaxError(e.message + ':\n' + template)
|
raise SyntaxError(e.message + ':\n' + template)
|
||||||
result = namespace[typename]
|
result = namespace[typename]
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue