[python] 使用sqlparse 解析和美化SQL

在 Python3 中，可以使用 sqlparse 模块来美化 SQL 语句。sqlparse 是一个开源的 Python 模块，它可以解析 SQL 语句并将其格式化为易于阅读和理解的结构。具体来说，sqlparse 可以：

1. 将 SQL 语句解析为语法树，并以易于操作的对象形式呈现；
2. 格式化 SQL 语句，包括缩进、大小写、空格等方面的优化；
3. 高亮显示 SQL 语句中的关键字、函数、表名等，以提高可读性。

以下是一个示例代码，演示如何使用 sqlparse 模块对 SQL 语句进行美化：

import sqlparse
# 定义一个 SQL 语句
sql = "SELECT id, name FROM users WHERE age > 18 ORDER BY create_time DESC;"
# 使用 sqlparse 解析和美化 SQL 语句
formatted_sql = sqlparse.format(sql, reindent=True, keyword_case='upper')
# 输出美化后的 SQL 语句
print(formatted_sql)

sqlparse 模块的 format() 方法对 SQL 语句进行美化，

在 format() 方法中，我们指定了 reindent=True 参数，表示需要重新缩进 SQL 语句；

同时指定了 keyword_case=‘upper’ 参数，表示将 SQL 关键字转换为大写形式。

需要注意的是，sqlparse 只是一种美化 SQL 的工具，它并不会对 SQL 的语义进行修改或优化。因此，在实际使用中，还需要仔细检查美化后的 SQL 语句，确保其与原始 SQL 语句的含义相同。

• 美化后输出如下

  SELECT id,
         name,
         age
  FROM users
  WHERE age > 18
  ORDER BY create_time DESC;
  

  获取sql语句中select的字段名

  使用 sqlparse 模块获取 SQL 语句中 SELECT 语句的字段名：

  import sqlparse
  # 定义一个 SQL 语句
  sql = "SELECT id, name, age FROM users WHERE age > 18 ORDER BY create_time DESC;"
  # 使用 sqlparse 解析 SQL 语句
  parsed = sqlparse.parse(sql)
  # 遍历解析后的结果，提取 SELECT 语句中的字段名
  for stmt in parsed:
      for token in stmt.tokens:
          if isinstance(token, sqlparse.sql.Token) and token.value.upper() == 'SELECT':
              for identifier in token.get_identifiers():
                  print(identifier.get_name())
  

  在上面的示例代码中，我们首先定义了一个 SQL 语句，然后使用 sqlparse 模块的 parse() 方法解析该 SQL 语句。接着，我们遍历解析后的结果，找到 SELECT 关键字所在的位置，然后提取其中的字段名。最后，我们使用 print() 函数将提取出的字段名打印出来。

  当你运行上述代码时，如果输入的 SQL 语句中有 SELECT 语句，那么会输出该 SELECT 语句中的字段名。例如，对于给定的 SQL 语句，输出可能如下所示：

  id
  name
  age