Skip to content

Commit 4edfd2f

Browse files
committed
Code cleanup.
1 parent b62b79f commit 4edfd2f

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

examples/extract_table_names.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
from sqlparse.sql import IdentifierList, Identifier
1414
from sqlparse.tokens import Keyword, DML
1515

16+
1617
def is_subselect(parsed):
1718
if not parsed.is_group():
1819
return False
@@ -21,6 +22,7 @@ def is_subselect(parsed):
2122
return True
2223
return False
2324

25+
2426
def extract_from_part(parsed):
2527
from_seen = False
2628
for item in parsed.tokens:
@@ -35,6 +37,7 @@ def extract_from_part(parsed):
3537
elif item.ttype is Keyword and item.value.upper() == 'FROM':
3638
from_seen = True
3739

40+
3841
def extract_table_identifiers(token_stream):
3942
for item in token_stream:
4043
if isinstance(item, IdentifierList):
@@ -47,9 +50,11 @@ def extract_table_identifiers(token_stream):
4750
elif item.ttype is Keyword:
4851
yield item.value
4952

53+
5054
def extract_tables():
5155
stream = extract_from_part(sqlparse.parse(sql)[0])
5256
return list(extract_table_identifiers(stream))
5357

58+
5459
if __name__ == '__main__':
5560
print 'Tables: %s' % ', '.join(extract_tables())

0 commit comments

Comments
 (0)