Python sqlparse 解析库的基础使用过程解析

目录

• 安装
• 使用
• 1. parse
• 2. format
• 3. split

pip install sqlparse

sql = "select * from table1 where id = 1;"
# 1. parse
# parse方法将 SQL语句 解析成一颗解析树的列表，每个解析树都是一个 SQL语句 的逻辑结构
parsed = sqlparse.parse(sql)
for item in parsed:
    # 解析树以Token的形式存在，每个Token代表SQL语句的一个部分
    print(item.tokens)

# 定义SQL语句
sql = "SELECT name, age FROM users WHERE id = 1;"
# 解析SQL语句
parsed = sqlparse.parse(sql)[0]
# 获取SQL语句的组成部分
for token in parsed.tokens:
    print("Token:", token)

sql = "select * from table1 where id = 1;"
# 2. format
# 将 sql 语句进行格式化的操作
# reindent：表示是否根据关键字来控制缩进
# keyword_case：表示是否将关键字变为大写
formatted = sqlparse.format(sql, reindent=True, keyword_case='upper')
print(formatted)

raw = 'select * from foo; select * from bar;'
parsed = sqlparse.split(raw, strip_semicolon=True)  # strip_semicolon 表示是否要将分号去掉，默认是 False
print(parsed)