如何列出涉及SQLAlchemy語句的所有表格

Question

如何列出涉及SQLAlchemy語句的所有表格？ 讓我們假設有下面的查詢

query = session.query(Address).filter(Address.person == person) 

怎麼能列出了在這種情況下使用該查詢 表：人，地址

Answer 1

RMN答案是好的，但它不能處理所有的情況下，子查詢，delete語句，存在報表等

這裏是提取所有表

from sqlalchemy.sql.visitors import ClauseVisitor 
from sqlalchemy import Table 


def extract_tables(sql_stmt): 
    tables = [] 
    visitor = ClauseVisitor() 
    cluase_iter = visitor.iterate(elem) 
    for e in cluase_iter: 
     if isinstance(e, Table): 
      tables.append(e) 
     if isinstance(e, (ValuesBase, UpdateBase)): 
      tables.append(e.table) 
    return set(tables) 

Answer 2

你可以從locate_all_froms得到所有的表，這樣的事情

from sqlalchemy import Table 
from sqlalchemy.sql.selectable import Alias 

tables = set() 
for f in query.statement.locate_all_froms(): 
    if isinstance(f, Table): 
     tables.add(f) 
    elif isinstance(f, Alias): 
     tables.add(f.element)