在數(shù)據(jù)庫(kù)操作和SQL查詢的開(kāi)發(fā)過(guò)程中，有時(shí)候我們?yōu)榱藙?dòng)態(tài)生成查詢、進(jìn)行權(quán)限控制、進(jìn)行查詢優(yōu)化或者其他一些與數(shù)據(jù)庫(kù)交互相關(guān)、數(shù)據(jù)庫(kù)監(jiān)控等的需求，需要從SQL語(yǔ)句中提取表名。本文分別使用正則表達(dá)式和使用SQL解析庫(kù)的方式來(lái)獲取。當(dāng)然實(shí)際使用中需要進(jìn)行優(yōu)化，本次只是做初步的獲取操作。

1.  使用正則表達(dá)式    正則表達(dá)式是一種強(qiáng)大的文本匹配工具，通過(guò)定義模式，可以從文本中提取需要的信息。在SQL語(yǔ)句中，我們可以使用正則表達(dá)式匹配關(guān)鍵字（如FROM、JOIN、UPDATE等）后面的表名，但是通常會(huì)因?yàn)镾QL的復(fù)雜度的問(wèn)題導(dǎo)致提取不夠準(zhǔn)確。

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)

例如，采用如下方式獲取結(jié)果：

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)

得到的結(jié)果如下：

其中關(guān)聯(lián)查詢的SQL結(jié)果有誤

select * from tb1 a , tb2 b  where  a.id=b.id  and   c1='a';

2.  使用SQL解析庫(kù)

SQL解析庫(kù)能夠更全面地理解SQL語(yǔ)句的結(jié)構(gòu)，提供了更為準(zhǔn)確的分析。sqlparse是一個(gè)常用的SQL解析庫(kù)，它可以解析SQL語(yǔ)句并將其轉(zhuǎn)換成語(yǔ)法樹(shù)。

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測(cè)試一下：

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)

測(cè)試結(jié)果如下：

關(guān)聯(lián)查詢可以獲取到準(zhǔn)確的表名了。

注： 以上只是簡(jiǎn)單演示SQL解析庫(kù)的方式，對(duì)于DDL等也需要再優(yōu)化一下，否則可能獲取到錯(cuò)誤信息。

3.  小結(jié)

從SQL語(yǔ)句中提取表名可以在數(shù)據(jù)庫(kù)操作和應(yīng)用程序開(kāi)發(fā)中發(fā)揮重要作用，從而使系統(tǒng)更加靈活、安全、高效。選擇合適的提取表名的方法取決于具體的需求和應(yīng)用場(chǎng)景。例如可以在如下場(chǎng)景中使用：

• 動(dòng)態(tài)查詢生成： 通過(guò)提取SQL語(yǔ)句中的表名，可以動(dòng)態(tài)生成適應(yīng)不同條件的查詢語(yǔ)句，提高代碼的靈活性
• 權(quán)限控制：根據(jù)SQL語(yǔ)句中涉及的表名，可以實(shí)現(xiàn)更細(xì)粒度的權(quán)限控制，確保用戶只能訪問(wèn)其有權(quán)限的表
• 查詢優(yōu)化： 了解SQL語(yǔ)句中的表結(jié)構(gòu)有助于進(jìn)行查詢優(yōu)化，根據(jù)表的大小、索引情況等因素進(jìn)行優(yōu)化
• 日志記錄：記錄每個(gè)查詢涉及的表名，可以用于性能分析和日志記錄，幫助理解應(yīng)用程序的行為
• 數(shù)據(jù)遷移和同步：在數(shù)據(jù)遷移或同步過(guò)程中，了解SQL語(yǔ)句涉及的表結(jié)構(gòu)有助于更好地管理數(shù)據(jù)變更，確保數(shù)據(jù)一致性
• 數(shù)據(jù)庫(kù)監(jiān)控： 可以配合監(jiān)控?cái)?shù)據(jù)庫(kù)中對(duì)應(yīng)表的使用情況監(jiān)控等