如何程序化编辑SQL查询？实现表名添加catalog前缀的方法

核心思路：精准匹配SQL子句中的表名 #

要批量替换database.xxx为catalog.database.xxx，关键是用正则定位FROM/JOIN关键字后的目标表名，同时兼容SQL的大小写不敏感特性，以及嵌套子查询的场景。

正则表达式设计 #

使用以下正则模式可以精准匹配目标：

\b(FROM|JOIN)\s+database\.(\w+)

• \b：单词边界，避免误匹配包含FROM/JOIN的其他字符串
• (FROM|JOIN)：匹配SQL关键字，保留原关键字的大小写
• \s+：匹配关键字与表名之间的任意数量空格
• database\.(\w+)：捕获database.开头的表名部分，\w+适配字母、数字、下划线组成的表名

替换规则为\1 catalog.database.\2，直接在原表名前插入catalog.前缀。

代码实现（Python示例） #

针对你的视图字典，遍历并批量替换：

import re

# 编译正则，开启忽略大小写模式
table_pattern = re.compile(r'\b(FROM|JOIN)\s+database\.(\w+)', re.IGNORECASE)

# 示例视图字典
view_queries = {
    "artist_sales_summary": """SELECT
  artists.first_name,
  artists.last_name,
  artist_sales.sales
FROM database.artists
JOIN (
    SELECT artist_id, SUM(sales_price) AS sales
    FROM database.sales
    GROUP BY artist_id
  ) AS artist_sales
  ON artists.id = artist_sales.artist_id;"""
}

# 批量更新查询语句
updated_queries = {
    view_name: table_pattern.sub(r'\1 catalog.database.\2', query)
    for view_name, query in view_queries.items()
}

# 验证结果
for name, sql in updated_queries.items():
    print(f">>> {name} <<<")
    print(sql)

额外注意点 #

• 若表名包含特殊字符（如连字符），需将正则中的\w+改为[\w-]+以适配
• 该正则会自动匹配嵌套子查询中的表名，无需额外处理层级问题
• 若SQL中存在FROM后接子查询的情况，正则只会匹配真正的表名，不会误改子查询结构

内容的提问来源于stack exchange，提问作者buttermilk