本篇文章介绍了python使用ply和re模块提取sql语句中表名的方法，具有一定的参考价值，希望对各位学习Python的朋友有帮助！
Python怎么从sql提取表名
ply：词法语法解析模块；
re：正则表达式模块。
全部代码如下：
import ply.lex as lex, re
def extract_table_name_from_sql(sql_str):
# remove the comments
q = re.sub(r"][^*]**+)*/", "", sql_str)
# remove whole line -- and # comments
lines = [line for line in q.splitlines() if not re.match("^s*(--|#)", line)]
# remove trailing -- and # comments
q = " ".join([re.split("--|#", line)[0] for line in lines])
# split on blanks, parens and semicolons
tokens = re.split(r"[s)(;]+", q)
# scan the tokens. if we see a FROM or JOIN, we set the get_next
# flag, and grab the next one (unless it's SELECt).
result = []
get_next = False
for token in tokens:
if get_next:
if token.lower() not in ["", "select"]:
result.append(token)
get_next = False
get_next = token.lower() in ["from", "join"]
return result
sql2="SELECt a.time_updated_server/1000,content,nick,name FROM "
"table1 a JOIN "
"table2 b ON a.sender_id = b.user_id JOIN table3 c ON a.channel_id = c.channel_id JOIN table4 d ON c.store_id = d.store_id WHERe sender_id NOT IN(SELECt user_id FROM table5 WHERe store_id IN ('agent_store:1', 'ask:1')) AND to_timestamp(a.time_updated_server/1000)::date >= '2014-05-01' GROUP BY 1,2,3,4 HAVINg sum(1) > 500 ORDER BY 1 ASC"
print(extract_table_name_from_sql(sql2))更多技术请关注Python视频教程。
