从 Cloud SQL 导出到带有列标题的 CSV

Question

我正在尝试使用 Gcloud SDK 中的

gcloud sql export csv

命令将表从 Google Cloud SQL 导出到 CSV 文件中，但我没有选项在文件顶部也导出列名称。有什么解决办法吗？

谢谢

Answer 1

我相信这个命令应该产生必要的功能：

gcloud sql 导出 csv 实例名称 gs://bucket_name/ --query="SELECT '列名1'，'列名2'，'列名3' UNION SELECT 列名1， 列名2、列名3 FROM 表名" --database=database_name

这样做的一个缺点是您必须指定所有列。如果数量很多，最好编写一些脚本来编写 SQL 查询部分。

已代表您创建功能请求。请为其加注星标，以便您可以收到有关此功能请求的更新，并随时添加其他评论以提供所需实现的详细信息。您可以通过以下此链接跟踪功能请求。

Answer 2

建立在@philipp-sh 答案之上。请为本机解决方案的链接票证投票。

如果您将

Python

与

SQLAlchemy

一起使用，您可以将

Selectable

转换为原始 SQL 字符串。以下辅助函数将所有列转换为

varchar

。第一行将是选定的列名称。

def convert_query_to_raw_gcp_csv_export_sql(query: Select) -> str:
    # Get the column names from the query
    column_names = [c.name for c in query.selected_columns]

    # Create a SQL string that selects the column names as the first row
    column_names_sql = "SELECT " + ", ".join(f"'{c}'" for c in column_names)

    # Cast all columns to varchar/string
    casted_query = query.with_only_columns(*[cast(c, String) for c in query.selected_columns])

    # Compile the casted query to get the SQL string, while keeping the original query
    psycopg2_dialect = dialects.postgresql.psycopg2.dialect()
    casted_sql = str(casted_query.compile(dialect=psycopg2_dialect, compile_kwargs={"literal_binds": True}))

    # Use UNION ALL to combine the column names with the original query
    raw_gcp_csv_export_sql = f"{column_names_sql} UNION ALL {casted_sql}"

    return raw_gcp_csv_export_sql

我希望这对某人有帮助。