### 使用 Flask 执行 SQL 文件以创建 SQLite3 数据库 要通过 Flask 加载并执行外部 `.sql` 文件的内容来初始化或更新 SQLite3 数据库，可以按照以下方式实现： --- #### 1. 定义配置项 首先需要在 Flask 的配置中指定数据库文件的位置以及 SQL 脚本的路径。 ```python import os from flask import Flask, current_app app = Flask(__name__) # 设置项目根目录 PROJECT_ROOT = os.path.dirname(os.path.abspath(__file__)) # 配置数据库路径和 SQL 初始化脚本路径 DATABASE_PATH = os.path.join(PROJECT_ROOT, 'instance', 'my_database.db') SQL_SCRIPT_PATH = os.path.join(PROJECT_ROOT, 'schema.sql') app.config['DATABASE'] = DATABASE_PATH app.config['SQL_SCRIPT'] = SQL_SCRIPT_PATH ``` 此部分代码设置了数据库文件和 SQL 脚本的具体存储位置[^2]。 --- #### 2. 编写 `init_db()` 方法加载 SQL 文件 接下来定义一个函数读取 SQL 文件内容并通过 SQLite 连接执行其中的语句。 ```python import sqlite3 def init_db(): """ 初始化数据库：从 schema.sql 文件中读取 SQL 并执行 """ database_path = current_app.config['DATABASE'] script_path = current_app.config['SQL_SCRIPT'] try: # 确保数据库父级目录存在 os.makedirs(os.path.dirname(database_path), exist_ok=True) # 建立数据库连接 with sqlite3.connect(database_path) as conn: cursor = conn.cursor() # 打开 .sql 文件并逐行执行其内容 with open(script_path, 'r') as sql_file: sql_script_content = sql_file.read() # 执行完整的 SQL 脚本 cursor.executescript(sql_script_content) conn.commit() except FileNotFoundError: raise RuntimeError(f"无法找到 SQL 文件: {script_path}") from None except sqlite3.Error as e: raise RuntimeError(f"执行 SQL 文件失败: {e.args[0]}") from e ``` 这里的关键在于利用 `executescript` 方法一次性运行多个 SQL 语句，并且提供了异常处理机制以增强健壮性[^3]。 --- #### 3. 注册 CLI 命令便于调用 为了让开发者能够在开发环境中更便捷地初始化数据库，可将上述功能转化为 Flask 提供的支持命令行工具的形式。 ```python import click @app.cli.command('init-db') def cli_init_db(): """ 添加自定义命令用于初始化数据库 """ init_db() click.echo("成功根据 schema.sql 初始化了数据库！") ``` 这样用户仅需输入如下命令即可完成全部操作： ```bash flask init-db ``` 这种方式极大地提升了工作效率同时也减少了人为失误的可能性[^3]。 --- #### 4. 结合全局请求钩子优化资源管理 最后建议加入一些辅助性的生命周期事件处理器，比如每次请求之前获取数据库连接实例，在请求结束之后释放它等等。这有助于提高性能表现同时降低潜在风险因素的影响程度。 ```python @app.before_request def before_request(): """ 请求开始前建立数据库连接 """ g.db = sqlite3.connect(current_app.config['DATABASE']) @app.teardown_request def teardown_request(exception=None): """ 请求结束后关闭数据库连接 """ db = getattr(g, 'db', None) if db is not None: db.close() ``` 这些措施共同构成了更加完善的解决方案体系结构。 --- ### 总结 综上所述，本文介绍了如何借助 Flask 来解析本地存在的 SQL 文件从而达到自动化构建 SQLite 数据库的目的。整个过程涵盖了必要的错误捕捉环节以及良好的实践指导原则，旨在帮助使用者快速掌握相关技能要点！ ---