[英]python generator in recursive function
我看過許多帖子,但仍然不太明白。 我有以下代碼,該代碼以嵌入在代碼中的sql文本打印出此查詢的外層。
K.a
K.b
I
J
K
1
2
當遞歸調用函數時,我不清楚如何獲取內部層。 我在非常相似且有效的https://github.com/andialbrecht/sqlparse/blob/master/examples/extract_table_names.py之后對其進行了圖案化。 提前致謝。
# This example illustrates how to extract table names from nested
# SELECT statements.
# See:
# http://groups.google.com/group/sqlparse/browse_thread/thread/b0bd9a022e9d4895
sql = """
select K.a,K.b from (select H.b from (select G.c from (select F.d from
(select E.e from A, B, C, D, E), F), G), H), I, J, K order by 1,2;
"""
import sqlparse
from sqlparse.sql import IdentifierList, Identifier
from sqlparse.tokens import Keyword, DML
def is_subselect(parsed):
if not parsed.is_group():
return False
for item in parsed.tokens:
if item.ttype is DML and item.value.upper() == 'SELECT':
return True
return False
def extract_from_part(parsed):
from_seen = False
for item in parsed.tokens:
if item.ttype is Keyword:
lastKeyword = str(item)
#if from_seen:
if is_subselect(item):
for x in extract_from_part(item):
if type(item) in(sqlparse.sql.IdentifierList,sqlparse.sql.Identifier):
yield x
else:
if type(item) in( sqlparse.sql.IdentifierList,sqlparse.sql.Identifier):
yield item
def extract_table_identifiers(token_stream):
for item in token_stream:
if isinstance(item, IdentifierList):
for identifier in item.get_identifiers():
yield str(identifier)
elif isinstance(item, Identifier):
#yield item.get_name()
yield str(identifier)
# It's a bug to check for Keyword here, but in the example
# above some tables names are identified as keywords...
elif item.ttype is Keyword:
yield item.value
def extract_tables():
stream = extract_from_part(sqlparse.parse(sql)[0])
return list(extract_table_identifiers(stream))
#return stream
if __name__ == '__main__':
#print('Tables: %s' % ', '.join(extract_tables()))
ids = extract_tables()
for x in ids:
print x
您在遞歸調用之后的循環中執行的類型檢查將測試錯誤的對象的類型:
if is_subselect(item):
for x in extract_from_part(item):
# next line should check type(x) rather than type(item)
if type(item) in(sqlparse.sql.IdentifierList,sqlparse.sql.Identifier):
yield x
不過,這種類型檢查是不必要的,因為非遞歸情況只會產生Identifier
或IdentifierList
項目。 因此,只需擺脫該行(並縮進以下yield
)。
雖然這不是造成此問題的原因,但我也建議您使用isinstance
而不是直接比較類型:
if isinstance(item, (sqlparse.sql.Identifier, sqlparse.sql.IdentifierList)):
yield item
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.