### Python脚本实现 以下是基于需求编写的完整Python脚本，该脚本能够完成从`my.sql`文件中读取SQL语句的任务，并通过调用指定的函数将其转换为Mermaid ER图代码，最终将结果写入到`myER.txt`文件中。 #### 脚本说明 1. **读取 SQL 文件**: 使用内置的 `open()` 函数来打开并读取位于当前工作目录下的 `my.sql` 文件。 2. **解析 SQL 语句**: 假设存在一个名为 `parse_sql_to_er_diagram_improved` 的函数用于处理 SQL 并返回 Mermaid 图表代码[^1]。 3. **生成目标文件**: 将生成的结果保存至同级目录下命名为 `myER.txt` 的新文件中。 ```python import os def parse_sql_to_er_diagram_improved(sql_content): """ 解析 SQL 内容并生成 Mermaid ER 图代码。 此处仅为占位逻辑，实际应替换为具体实现。 """ er_code = "erDiagram\n" lines = sql_content.splitlines() for line in lines: if 'CREATE TABLE' in line.upper(): table_name = line.split()[2].strip('()') er_code += f" {table_name} {{\n" elif ')' in line and ('PRIMARY KEY' not in line.upper()): column_info = line.strip(',').split() col_name, col_type = column_info[:2] er_code += f" {col_name} {col_type}\n" elif ');' in line or 'PRIMARY KEY' in line.upper(): er_code += " }\n" return er_code if __name__ == "__main__": current_dir = os.path.dirname(os.path.abspath(__file__)) # 定义输入输出路径 input_file_path = os.path.join(current_dir, "my.sql") output_file_path = os.path.join(current_dir, "myER.txt") try: with open(input_file_path, 'r', encoding='utf-8') as file: sql_content = file.read() mermaid_code = parse_sql_to_er_diagram_improved(sql_content) with open(output_file_path, 'w', encoding='utf-8') as out_file: out_file.write(mermaid_code) print(f'Mermaid ER diagram has been successfully written to "{output_file_path}".') except FileNotFoundError: print(f"The specified SQL file '{input_file_path}' was not found.") except Exception as e: print(f"An error occurred while processing the files: {e}") ``` 此脚本假设 `my.sql` 存在于运行环境的工作目录中。如果不存在，则会抛出异常提示用户相应错误信息[^2]。 --- ### 关键点解释 1. **SQL 到 Mermaid 的映射** 上述示例中的 `parse_sql_to_er_diagram_improved` 是自定义方法，其功能是从标准 SQL 表结构提取字段名及其数据类型，并按照 Mermaid 文法格式化成图表描述语言[^3]。 2. **跨平台兼容性** 使用了 `os.path` 来构建绝对路径，从而确保无论在哪种操作系统上执行都能找到正确的相对位置文件[^4]。 3. **异常捕获机制** 对可能发生的 I/O 错误进行了基本防护措施，比如当找不到源 `.sql` 或者无法创建目标 `.txt` 文件时给出清晰反馈给终端使用者了解情况进展[^5]。 ---