diff options
Diffstat (limited to 'examples')
| -rw-r--r-- | examples/extract_table_names.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/examples/extract_table_names.py b/examples/extract_table_names.py index b153850..dbe83f9 100644 --- a/examples/extract_table_names.py +++ b/examples/extract_table_names.py @@ -24,7 +24,9 @@ def is_subselect(parsed): def extract_from_part(parsed): from_seen = False for item in parsed.tokens: - if from_seen: + if item.ttype is Keyword: + from_seen = False + elif from_seen: if is_subselect(item): for x in extract_from_part(item): yield x |