Pythonで様々なSQL処理を行えるSQLGlotの機能を試してみる

pip install sqlglot

import sqlglot
from sqlglot import exp

sql = """
SELECT
    u.user_name,
    p.product_name,
    s.sale_date
FROM
    sales AS s
JOIN
    users AS u ON s.user_id = u.id
LEFT JOIN
    products p ON s.product_id = p.id
WHERE
    s.amount > 1000
"""

ast = sqlglot.parse_one(sql)

# すべてのテーブル名（エイリアスも含む）を抽出
tables = {table.name for table in ast.find_all(exp.Table)}
print(f"使われているテーブル: {tables}")

# すべてのカラム名を抽出
columns = {column.name for column in ast.find_all(exp.Column)}
print(f"使われているカラム: {columns}")

使われているテーブル: {'sales', 'products', 'users'}
使われているカラム: {'product_name', 'id', 'user_id', 'product_id', 'sale_date', 'user_name', 'amount'}

import sqlglot

# フォーマット前のSQL
ugly_sql = "SELECT id,name,created_at FROM users WHERE status='active' ORDER BY created_at DESC"

# フォーマット
formatted_sql = sqlglot.parse_one(ugly_sql).sql(pretty=True)
print(formatted_sql)

SELECT
  id,
  name,
  created_at
FROM users
WHERE
  status = 'active'
ORDER BY
  created_at DESC

import sqlglot

# DuckDBの構文からHiveの構文へ変換
duckdb_sql = "SELECT EPOCH_MS(1618088028295)"
hive_sql = sqlglot.transpile(duckdb_sql, read="duckdb", write="hive")
print(hive_sql[0])

# PostgreSQLからBigQueryへの変換
postgres_sql = "SELECT NOW()::DATE"
bigquery_sql = sqlglot.transpile(postgres_sql, read="postgres", write="bigquery")
print(bigquery_sql[0])

SELECT FROM_UNIXTIME(1618088028295 / 1000)

SELECT CAST(CURRENT_TIMESTAMP() AS DATE)