### Python SQLite 工具类实现示例 以下是一个基于 Python 的 SQLite 工具类，支持创建表格和插入数据的功能。此工具类的设计参考了常见的数据库操作模式，并结合了 SQLite3 模块的特性[^1]。 ```python import sqlite3 from sqlite3 import Error class SQLiteTool: def __init__(self, db_path): """ 初始化 SQLite 工具类。 :param db_path: 数据库文件路径 """ self.db_path = db_path self.connection = None def connect(self): """ 连接到 SQLite 数据库。 """ try: self.connection = sqlite3.connect(self.db_path) print("成功连接到数据库") except Error as e: print(f"连接数据库时发生错误: {e}") def close(self): """ 关闭数据库连接。 """ if self.connection: self.connection.close() def execute(self, sql, parameters=None): """ 执行 SQL 语句。 :param sql: SQL 语句 :param parameters: 参数列表（可选） :return: 执行结果 """ try: cursor = self.connection.cursor() if parameters: cursor.execute(sql, parameters) else: cursor.execute(sql) self.connection.commit() return cursor except Error as e: print(f"执行 SQL 语句时发生错误: {e}") return None def create_table(self, table_name, columns): """ 创建表格。 :param table_name: 表格名称 :param columns: 列定义 (例如: "id INTEGER PRIMARY KEY, name TEXT") """ sql = f"CREATE TABLE IF NOT EXISTS {table_name} ({columns});" self.execute(sql) def insert_data(self, table_name, data): """ 插入数据到表格。 :param table_name: 表格名称 :param data: 数据字典 (键为列名，值为数据) """ keys = ", ".join(data.keys()) values = ", ".join(["?"] * len(data)) sql = f"INSERT INTO {table_name} ({keys}) VALUES ({values});" self.execute(sql, list(data.values())) # 示例用法 if __name__ == "__main__": # 初始化工具类 db_tool = SQLiteTool("./mysqlite.db") db_tool.connect() # 创建表格 table_columns = "id INTEGER PRIMARY KEY, name TEXT, age INTEGER" db_tool.create_table("staffs", table_columns) # 插入数据 staff_data = {"name": "Alice", "age": 30} db_tool.insert_data("staffs", staff_data) # 查询数据 query_sql = "SELECT * FROM staffs;" result = db_tool.execute(query_sql).fetchall() print("查询结果:", result) # 关闭连接 db_tool.close() ``` ### 功能说明 1. **连接数据库**：`connect` 方法用于建立与 SQLite 数据库的连接[^1]。 2. **关闭连接**：`close` 方法确保在操作完成后正确关闭数据库连接。 3. **执行 SQL**：`execute` 方法提供了一个通用接口来执行任意 SQL 语句，并支持参数化查询以防止 SQL 注入攻击[^2]。 4. **创建表格**：`create_table` 方法允许用户通过指定表名和列定义来创建新表格。 5. **插入数据**：`insert_data` 方法接受一个字典作为输入，将键值对映射到表格的列和数据中。 ### 注意事项 - 在实际应用中，建议使用更复杂的错误处理机制来增强程序的健壮性。 - `execute` 方法返回的是游标对象，可以通过它获取查询结果或受影响的行数。 - 数据库文件路径应根据实际需求设置，避免硬编码可能导致的问题。