diff options
| author | Andi Albrecht <albrecht.andi@gmail.com> | 2009-05-06 06:58:01 +0200 |
|---|---|---|
| committer | Andi Albrecht <albrecht.andi@gmail.com> | 2009-05-06 06:58:01 +0200 |
| commit | 087de87e41669fc88594b8344515c83cf3f26eb3 (patch) | |
| tree | 5027adbd0080a93f484d8cc2f1b3d4e9d198682f /sqlparse | |
| parent | 7b2bf235613759f1894fcdf9e2fc39af7ac21689 (diff) | |
| download | sqlparse-087de87e41669fc88594b8344515c83cf3f26eb3.tar.gz | |
* Added profile option to run_tests.py.
* Use sets in Token.match.
Diffstat (limited to 'sqlparse')
| -rw-r--r-- | sqlparse/sql.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/sqlparse/sql.py b/sqlparse/sql.py index 8eef036..ee18d4f 100644 --- a/sqlparse/sql.py +++ b/sqlparse/sql.py @@ -16,7 +16,7 @@ class Token(object): the type of the token. """ - __slots__ = ('value', 'ttype') + __slots__ = ('value', 'ttype',) def __init__(self, ttype, value): self.value = value @@ -69,19 +69,20 @@ class Token(object): if values is None: return self.ttype is ttype if isinstance(values, basestring): - values = [values] + values = set([values]) if regex: if self.ttype is T.Keyword: - values = [re.compile(v, re.IGNORECASE) for v in values] + values = set([re.compile(v, re.IGNORECASE) for v in values]) else: - values = [re.compile(v) for v in values] + values = set([re.compile(v) for v in values]) for pattern in values: if pattern.search(self.value): return True return False else: if self.ttype is T.Keyword: - return self.value.upper() in [v.upper() for v in values] + values = set([v.upper() for v in values]) + return self.value.upper() in values else: return self.value in values @@ -212,7 +213,6 @@ class TokenList(Token): for token in self.tokens[idx:]: for i, func in enumerate(funcs): if func(token): - print 'MATCHED', i, token return token return None |
