summaryrefslogtreecommitdiff
path: root/sqlparse
diff options
context:
space:
mode:
authorAndi Albrecht <albrecht.andi@gmail.com>2009-05-06 07:12:34 +0200
committerAndi Albrecht <albrecht.andi@gmail.com>2009-05-06 07:12:34 +0200
commit4fb9cb894e93c42a919ccd714d9a4a7d5ed2e940 (patch)
tree08dc4a866d3d6856b7b259d88e586af7dcf52fac /sqlparse
parent087de87e41669fc88594b8344515c83cf3f26eb3 (diff)
downloadsqlparse-4fb9cb894e93c42a919ccd714d9a4a7d5ed2e940.tar.gz
Do the type check once in Token.match.
Diffstat (limited to 'sqlparse')
-rw-r--r--sqlparse/sql.py7
1 files changed, 3 insertions, 4 deletions
diff --git a/sqlparse/sql.py b/sqlparse/sql.py
index ee18d4f..2ab05b9 100644
--- a/sqlparse/sql.py
+++ b/sqlparse/sql.py
@@ -64,10 +64,9 @@ class Token(object):
If *regex* is ``True`` (default is ``False``) the given values are
treated as regular expressions.
"""
- if self.ttype is not ttype:
- return False
- if values is None:
- return self.ttype is ttype
+ type_matched = self.ttype in ttype
+ if not type_matched or values is None:
+ return type_matched
if isinstance(values, basestring):
values = set([values])
if regex: