使用sqlparse解析table_name,支持子查询, left join等
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 | import sqlparse from sqlparse.sql import IdentifierList, Identifier from sqlparse.tokens import Keyword, DML # 支持的join方式 ALL_JOIN_TYPE = ( 'LEFT JOIN' , 'RIGHT JOIN' , 'INNER JOIN' , 'FULL JOIN' , 'LEFT OUTER JOIN' , 'FULL OUTER JOIN' ) def is_subselect(parsed): """ 是否子查询 :param parsed: :return: """ 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之后模块 :param parsed: :return: """ from_seen = False for item in parsed.tokens: if from_seen: if is_subselect(item): for x in extract_from_part(item): yield x elif item.ttype is Keyword: from_seen = False continue else : yield item elif item.ttype is Keyword and item.value.upper() = = 'FROM' : from_seen = True def extract_join_part(parsed): """ 提交join之后模块 :param parsed: :return: """ flag = False for item in parsed.tokens: if flag: if item.ttype is Keyword: flag = False continue else : yield item if item.ttype is Keyword and item.value.upper() in ALL_JOIN_TYPE: flag = True def extract_table_identifiers(token_stream): for item in token_stream: if isinstance (item, IdentifierList): for identifier in item.get_identifiers(): yield identifier.get_name() elif isinstance (item, Identifier): yield item.get_name() elif item.ttype is Keyword: yield item.value def extract_tables(sql): """ 提取sql中的表名(select语句) :param sql: :return: """ from_stream = extract_from_part(sqlparse.parse(sql)[ 0 ]) join_stream = extract_join_part(sqlparse.parse(sql)[ 0 ]) return list (extract_table_identifiers(from_stream)) + list (extract_table_identifiers(join_stream)) print (extract_tables( "select x1, x2 from a left join b on a.id = b.id right join c on c.id = a.id" )) print (extract_tables( "select x1, x2 from (select x1, x2 from (select x1, x2 from a)) left join b on a.id=b.id" )) |
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· Linux系列:如何用heaptrack跟踪.NET程序的非托管内存泄露
· 开发者必知的日志记录最佳实践
· SQL Server 2025 AI相关能力初探
· Linux系列:如何用 C#调用 C方法造成内存泄露
· AI与.NET技术实操系列(二):开始使用ML.NET
· 无需6万激活码!GitHub神秘组织3小时极速复刻Manus,手把手教你使用OpenManus搭建本
· C#/.NET/.NET Core优秀项目和框架2025年2月简报
· Manus爆火,是硬核还是营销?
· 终于写完轮子一部分:tcp代理 了,记录一下
· Qt个人项目总结 —— MySQL数据库查询与断言