diff options
author | Sjoerd Job Postmus <sjoerdjob@sjec.nl> | 2016-06-02 10:28:54 +0200 |
---|---|---|
committer | Victor Uriarte <victor.m.uriarte@intel.com> | 2016-06-12 17:38:22 -0700 |
commit | 8f7968ed5c649e5227e605ee272f59dd5ca75adb (patch) | |
tree | 2941c9afb4d32507d563071c6d8dd1325d7cfaa1 /sqlparse/sql.py | |
parent | 67dc823e1174eee9ea2159674c8eb016b2f95b54 (diff) | |
download | sqlparse-8f7968ed5c649e5227e605ee272f59dd5ca75adb.tar.gz |
Index-based token_idx_prev
Prevent some more calls to token_index in group_identifier_list. They
are now all gone.
Diffstat (limited to 'sqlparse/sql.py')
-rw-r--r-- | sqlparse/sql.py | 28 |
1 files changed, 22 insertions, 6 deletions
diff --git a/sqlparse/sql.py b/sqlparse/sql.py index 928b784..9782c33 100644 --- a/sqlparse/sql.py +++ b/sqlparse/sql.py @@ -233,12 +233,18 @@ class TokenList(Token): if not isinstance(funcs, (list, tuple)): funcs = (funcs,) - iterable = enumerate(self.tokens[start:end], start=start) - - for idx, token in iterable: - for func in funcs: - if func(token): - return idx, token + if reverse: + assert end is None + for idx in range(start - 2, -1, -1): + token = self.tokens[idx] + for func in funcs: + if func(token): + return idx, token + else: + for idx, token in enumerate(self.tokens[start:end], start=start): + for func in funcs: + if func(token): + return idx, token return None, None def _token_matching(self, funcs, start=0, end=None, reverse=False): @@ -312,6 +318,16 @@ class TokenList(Token): def token_matching(self, idx, funcs): return self._token_matching(funcs, idx) + def token_idx_prev(self, idx, skip_ws=True): + """Returns the previous token relative to *idx*. + + If *skip_ws* is ``True`` (the default) whitespace tokens are ignored. + ``None`` is returned if there's no previous token. + """ + idx += 1 # alot of code usage current pre-compensates for this + funcs = lambda tk: not (tk.is_whitespace() and skip_ws) + return self._token_idx_matching(funcs, idx, reverse=True) + def token_prev(self, idx, skip_ws=True): """Returns the previous token relative to *idx*. |