schema2dwh

️重要提示！

schema2dwh由 AI - Google 的 Gemini 提供支持 - 请确保在将其注册到生产环境之前检查所有代码输出。另请注意，通过此脚本进行处理将利用 Gemini API，其中数据由 Google 处理。

schema2dwh [AI 支持的 DB 架构到 DWH 架构创建脚本]

schema2dwh是一个开源框架，旨在使用其信息模式简化并自动生成基于数据库的数据模型，利用人工智能以及有关您输入的一些问题，它将快速为您生成一个 SQL DDL 文件，准备创建新建数据仓库背后的骨架。

特征

• 以交互方式收集用户的必要输入
• 使用 Google Generative AI 生成 STAR/SNOWFLAKE 模式
• 支持各种数据库软件（MySQL、SQL Server、PostgreSQL等）
• 输出与指定数据库软件兼容的SQL DDL语句
• 向所有列添加注释以供数据库用户参考
• 将生成的模式保存到指定的输出文件

预览：

请参阅下面的示例来了解其工作原理。

输入（使用my_schema.csv运行脚本：

输出DDL（ output_schema.sql ：

先决条件

• Python 3.x
• 谷歌生成人工智能Python包（ google-generativeai ）
• Google 生成式 AI API 密钥 - https://aistudio.google.com/app/apikey

安装

1. 克隆存储库：

   git clone https://github.com/cqllum/schema2dwh.git
   cd schema2dwh

2. 安装所需的软件包：

   pip install google-generativeai

3. 设置您的 Google Generative AI API 密钥：将脚本中的占位符your-api-key替换为您的实际 Google Generative AI API 密钥。在这里生成密钥：https://aistudio.google.com/app/apikey

4. 替换示例输入架构文件 (my_schema.csv)：请注意，为了使其尽可能准确，您需要从数据库中获取信息架构。在大多数情况下，它就像SELECT * FROM information_schema.columns一样简单 - 否则，请参阅数据库软件文档。

用法

1. 运行脚本：

   python schema2dwh.py

2. 按照提示操作：该脚本将提示您输入以下信息：

   • 数据库软件（例如 MySQL、SQL Server、PostgreSQL）
   • 数据库/模式名称
   • 与您的数据库相关的行业（例如电子商务、建筑、保险）
   • 表/列的命名约定（例如，snake_case、PascalCase、camelCase）
   • CSV 格式的信息模式的文件路径
   • 报告的重要数据点

3. 输出：该脚本将生成 DDL SQL 语句并将其保存到output_schema.sql 。如果无法生成模式，它将提示您提供正确的信息模式。

脚本结构

• configure_genai(api_key) ：使用提供的 API 密钥配置生成式 AI 模型。
• create_model() ：使用指定的设置创建并配置生成式 AI 模型。
• start_chat_session(model) ：使用预定义的指令启动与模型的聊天会话。
• gather_user_input() ：以交互方式收集用户的必要输入。
• format_response(db_software, db_name, industry, case_type, input_schema, additional_value) ：将用户输入格式化为聊天会话的响应。
• save_schema_to_file(output_file, schema_output) ：将生成的模式输出保存到指定文件。
• main() ：驱动脚本的主函数。