sqlglot，一个神奇的 python 库！

SQLGlot 是一个专为处理 SQL 设计的 Python 库，支持解析、转换（跨方言）、优化和格式化 SQL 查询。它特别适用于需要兼容多种 SQL方言或自动化操作 SQL 的任务。

它可用于格式化 SQL 或在 24 种不同的方言之间进行转换，如 DuckDB、PrestoTrino、SparkDatabricks、Snowflake 和 BigQuery。它旨在读取各种 SQL 输入，并在目标方言中输出语法和语义正确的 SQL。

资料

https://sqlglot.com/sqlglot.html

安装

pip install sqlglot[rs]

核心功能

解析 SQL 为抽象语法树 (AST)

import sqlglot

sql = "SELECT a FROM tbl WHERE b = 'foo'"

# 解析 SQL，返回 AST 列表（每个语句一个 AST）

ast = sqlglot.parse(sql)[0]

print(ast)

输出

SQL 方言转换（Transpile）

SQL 从一种方言转换为另一种（如 PostgreSQL → Spark）

transpiled = sqlglot.transpile(

"SELECT DATE_TRUNC('month', ts) FROM tbl",

read="postgres", # 输入方言

write="spark" # 输出方言

)[0]

print(transpiled)

输出

SELECT DATE_TRUNC('MONTH', ts) FROM tbl

优化 SQL 查询

简化查询逻辑或提升性能（如消除冗余子查询）

import sqlglot

from sqlglot.optimizer import optimize

sql = "SELECT * FROM (SELECT * FROM tbl) AS x WHERE x.a = 1"

optimized = optimize(

sqlglot.parse_one(sql),

schema={"tbl": {"a": "INT"}} # 提供表结构以辅助优化

).sql()

格式化 SQL

美化 SQL 代码，提升可读性

sql = SELECT foo, bar, baz FROM table_one WHERE foo>1

formatted = sqlglot.parse_one(sql).sql(pretty=True)

print(formatted)

输出

SELECT

foo,

bar,

baz

FROM table_one

WHERE

foo > 1

应用场景

数据库迁移

SQL 标准化

查询性能优化

语法检查与审计

SQLGlot 的多种方言支持使其成为处理复杂 SQL 任务的强大工具。

更多相关技术内容咨询欢迎前往并持续关注好学星城论坛了解详情。

想高效系统的学习Python编程语言，推荐大家关注一个微信公众号：Python编程学习圈。每天分享行业资讯、技术干货供大家阅读，关注即可免费领取整套Python入门到进阶的学习资料以及教程，感兴趣的小伙伴赶紧行动起来吧。

发表于 2025-05-17 09:25
阅读 ( 81 )
分类：Python开发

sqlglot，一个神奇的 python 库！

你可能感兴趣的文章

相关问题

0 条评论

作家榜 »