Remove dependency on the collections module.

This commit is contained in:
Raymond Hettinger 2009-04-29 14:21:25 +00:00
parent df9d4d6c7f
commit aa17a7fc98
1 changed files with 41 additions and 3 deletions

View File

@ -24,7 +24,6 @@ __credits__ = ('GvR, ESR, Tim Peters, Thomas Wouters, Fred Drake, '
'Skip Montanaro, Raymond Hettinger, Trent Nelson, ' 'Skip Montanaro, Raymond Hettinger, Trent Nelson, '
'Michael Foord') 'Michael Foord')
import collections
import re, string, sys import re, string, sys
from token import * from token import *
from codecs import lookup, BOM_UTF8 from codecs import lookup, BOM_UTF8
@ -32,7 +31,7 @@ cookie_re = re.compile("coding[:=]\s*([-\w.]+)")
import token import token
__all__ = [x for x in dir(token) if x[0] != '_'] + ["COMMENT", "tokenize", __all__ = [x for x in dir(token) if x[0] != '_'] + ["COMMENT", "tokenize",
"detect_encoding", "NL", "untokenize", "ENCODING", "Tokenize"] "detect_encoding", "NL", "untokenize", "ENCODING", "TokenInfo"]
del token del token
COMMENT = N_TOKENS COMMENT = N_TOKENS
@ -43,7 +42,46 @@ ENCODING = N_TOKENS + 2
tok_name[ENCODING] = 'ENCODING' tok_name[ENCODING] = 'ENCODING'
N_TOKENS += 3 N_TOKENS += 3
TokenInfo = collections.namedtuple('TokenInfo', 'type string start end line') class TokenInfo(tuple):
'TokenInfo(type, string, start, end, line)'
__slots__ = ()
_fields = ('type', 'string', 'start', 'end', 'line')
def __new__(cls, type, string, start, end, line):
return tuple.__new__(cls, (type, string, start, end, line))
@classmethod
def _make(cls, iterable, new=tuple.__new__, len=len):
'Make a new TokenInfo object from a sequence or iterable'
result = new(cls, iterable)
if len(result) != 5:
raise TypeError('Expected 5 arguments, got %d' % len(result))
return result
def __repr__(self):
return 'TokenInfo(type=%r, string=%r, start=%r, end=%r, line=%r)' % self
def _asdict(self):
'Return a new dict which maps field names to their values'
return dict(zip(self._fields, self))
def _replace(self, **kwds):
'Return a new TokenInfo object replacing specified fields with new values'
result = self._make(map(kwds.pop, ('type', 'string', 'start', 'end', 'line'), self))
if kwds:
raise ValueError('Got unexpected field names: %r' % kwds.keys())
return result
def __getnewargs__(self):
return tuple(self)
type = property(lambda t: t[0])
string = property(lambda t: t[1])
start = property(lambda t: t[2])
end = property(lambda t: t[3])
line = property(lambda t: t[4])
def group(*choices): return '(' + '|'.join(choices) + ')' def group(*choices): return '(' + '|'.join(choices) + ')'
def any(*choices): return group(*choices) + '*' def any(*choices): return group(*choices) + '*'