如何獲取一條SQL語句中涉及的表名
作者：懂點IT的耿小廚
在數據庫操作和SQL查詢的開發過程中，有時候我們為了動態生成查詢、進行權限控制、進行查詢優化或者其他一些與數據庫交互相關、數據庫監控等的需求，需要從SQL語句中提取表名。
在數據庫操作和SQL查詢的開發過程中，有時候我們為了動態生成查詢、進行權限控制、進行查詢優化或者其他一些與數據庫交互相關、數據庫監控等的需求，需要從SQL語句中提取表名。本文分別使用正則表達式和使用SQL解析庫的方式來獲取。當然實際使用中需要進行優化，本次只是做初步的獲取操作。
1. 使用正則表達式 正則表達式是一種強大的文本匹配工具，通過定義模式，可以從文本中提取需要的信息。在SQL語句中，我們可以使用正則表達式匹配關鍵字（如FROM、JOIN、UPDATE等）后面的表名，但是通常會因為SQL的復雜度的問題導致提取不夠準確。
import re
def get_table1(sql_statement):
table_names = re.findall(r'FROM\s+(\w+)|JOIN\s+(\w+)|UPDATE\s+(\w+)|INTO\s+(\w+)|TABLE\s+(\w+)', sql_statement, re.IGNORECASE)
for tbs in table_names:
for tb in tbs:
print(tb)例如，采用如下方式獲取結果：
if __name__ == '__main__':
sql = "select * from tb1 where c1='a';"
get_table1(sql)
sql = "select * from tb1 a , tb2 b where a.id=b.id and c1='a';"
get_table1(sql)
sql ="UPDATE tb3 SET b = 1 WHERE c1='47d8af9d8cd1459a927327b9d548a37b' "
get_table1(sql)得到的結果如下：
其中關聯查詢的SQL結果有誤
select * from tb1 a , tb2 b where a.id=b.id and c1='a';2. 使用SQL解析庫
SQL解析庫能夠更全面地理解SQL語句的結構，提供了更為準確的分析。sqlparse是一個常用的SQL解析庫，它可以解析SQL語句并將其轉換成語法樹。
import sqlparse
def get_table2(sql_statement):
parsed = sqlparse.parse(sql_statement)
for stmt in parsed:
for token in stmt.tokens:
if isinstance(token, sqlparse.sql.IdentifierList):
for identifier in token.get_identifiers():
print(identifier.get_real_name())
elif isinstance(token, sqlparse.sql.Identifier):
print(token.get_real_name())在使用上述SQL測試一下：
if __name__ == '__main__':
sql = "select * from tb1 where c1='a';"
get_table2(sql)
sql = "select * from tb1 a , tb2 b where a.id=b.id and c1='a';"
get_table2(sql)
sql ="UPDATE tb3 SET b = 1 WHERE c1='47d8af9d8cd1459a927327b9d548a37b' "
get_table2(sql)測試結果如下：
關聯查詢可以獲取到準確的表名了。
注： 以上只是簡單演示SQL解析庫的方式，對于DDL等也需要再優化一下，否則可能獲取到錯誤信息。
3. 小結
從SQL語句中提取表名可以在數據庫操作和應用程序開發中發揮重要作用，從而使系統更加靈活、安全、高效。選擇合適的提取表名的方法取決于具體的需求和應用場景。例如可以在如下場景中使用：
- 動態查詢生成： 通過提取SQL語句中的表名，可以動態生成適應不同條件的查詢語句，提高代碼的靈活性
- 權限控制：根據SQL語句中涉及的表名，可以實現更細粒度的權限控制，確保用戶只能訪問其有權限的表
- 查詢優化： 了解SQL語句中的表結構有助于進行查詢優化，根據表的大小、索引情況等因素進行優化
- 日志記錄：記錄每個查詢涉及的表名，可以用于性能分析和日志記錄，幫助理解應用程序的行為
- 數據遷移和同步：在數據遷移或同步過程中，了解SQL語句涉及的表結構有助于更好地管理數據變更，確保數據一致性
- 數據庫監控： 可以配合監控數據庫中對應表的使用情況監控等
責任編輯：華軒
來源：
數據庫干貨鋪
