当前位置： 技术文章>> Python 如何操作 SQLite 数据库？

文章标题：Python 如何操作 SQLite 数据库？
  • 文章分类: 后端
  • 8128 阅读

在Python中操作SQLite数据库是一种高效且便捷的方式来存储和检索数据，尤其适用于小型应用、原型开发或需要轻量级数据库解决方案的场景。SQLite是一个自包含的、高可靠的、基于磁盘的数据库管理系统，它不需要一个独立的服务器进程或操作，可以直接在应用程序中嵌入。接下来，我们将深入探讨如何在Python中使用SQLite数据库，包括如何连接到数据库、创建表、插入数据、查询数据、更新数据和删除数据。

引入SQLite库

在Python中，操作SQLite数据库主要依赖于内置的sqlite3模块。这个模块提供了一个简单的API来与SQLite数据库进行交互。首先，我们需要导入这个模块：

import sqlite3

连接到SQLite数据库

要操作SQLite数据库，首先需要建立与数据库的连接。如果数据库文件不存在，SQLite会自动创建一个新的数据库文件。以下是如何连接到SQLite数据库的一个示例：

# 连接到SQLite数据库
# 如果文件不存在，会自动在当前目录创建:
conn = sqlite3.connect('example.db')

创建游标对象

在Python的sqlite3模块中，游标（Cursor）对象用于执行SQL命令。通过游标，我们可以向数据库发送SQL命令并接收结果。

# 创建一个Cursor对象并使用它
cursor = conn.cursor()

创建表

创建表是数据库操作的基础。使用SQL语句通过游标对象执行可以创建新的表。以下是一个创建简单用户表的例子：

# 创建一个表
cursor.execute('''CREATE TABLE IF NOT EXISTS users
               (id INTEGER PRIMARY KEY, name TEXT, age INTEGER)''')

这条SQL语句创建了一个名为users的表，包含三个字段：id（主键）、name（文本类型）和age（整数类型）。IF NOT EXISTS是可选的，用于防止在表已存在时执行创建表的语句导致错误。

插入数据

向表中插入数据同样是通过执行SQL语句完成的。以下是如何向users表中插入新记录的示例：

# 插入一行记录
cursor.execute("INSERT INTO users (name, age) VALUES ('Alice', 30)")

# 对于多个插入，可以使用executemany()
users = [('Bob', 25), ('Charlie', 35)]
cursor.executemany("INSERT INTO users (name, age) VALUES (?, ?)", users)

# 提交事务
conn.commit()

注意，在插入数据后，需要调用commit()方法来提交事务，以确保数据被实际写入数据库。

查询数据

查询数据库是获取存储数据的重要方式。使用execute()方法执行SELECT语句，并通过游标对象的fetchall()、fetchone()或fetchmany()方法来获取查询结果。

# 查询所有记录
cursor.execute("SELECT * FROM users")
rows = cursor.fetchall()
for row in rows:
    print(row)

# 查询特定记录
cursor.execute("SELECT * FROM users WHERE name = ?", ('Alice',))
row = cursor.fetchone()
if row:
    print(row)

在上面的示例中，?是一个占位符，用于防止SQL注入攻击，并允许我们通过参数化查询安全地传递数据。

更新数据

更新数据库中的记录也是常见的操作。使用UPDATE语句通过游标对象执行即可。

# 更新记录
cursor.execute("UPDATE users SET age = ? WHERE name = ?", (31, 'Alice'))
conn.commit()

同样，更新后需要调用commit()方法来提交事务。

删除数据

删除数据库中的记录使用DELETE语句。

# 删除记录
cursor.execute("DELETE FROM users WHERE name = ?", ('Bob',))
conn.commit()

在执行删除操作后，同样需要调用commit()方法来提交事务。

关闭连接

完成数据库操作后，应关闭游标和连接以释放资源。

# 关闭Cursor和Connection
cursor.close()
conn.close()

额外技巧与最佳实践

  • 使用参数化查询：如上所述，使用?作为占位符并通过execute()或executemany()方法的参数传递数据，可以防止SQL注入攻击。
  • 异常处理：在实际应用中，应对数据库操作进行异常处理，以捕获并处理可能出现的错误，如连接失败、SQL语法错误等。
  • 使用with语句管理资源：利用Python的with语句可以自动管理资源，确保即使在发生异常时也能正确关闭游标和连接。
with sqlite3.connect('example.db') as conn:
    cursor = conn.cursor()
    # 在此执行数据库操作
    cursor.close()  # 实际上，在with语句块结束时，连接会自动关闭，因此这行代码是可选的
  • 定期备份数据库：对于重要的数据，定期备份是避免数据丢失的重要措施。

结论

在Python中操作SQLite数据库是一个直接且高效的过程，通过内置的sqlite3模块可以轻松实现数据的增删改查操作。掌握这些基础操作后，你就可以在需要轻量级数据库支持的项目中灵活运用SQLite了。无论是在开发个人项目、小型应用还是进行原型设计时，SQLite都是一个非常实用的选择。希望本文的介绍能对你有所帮助，在你的编程旅程中提供一份助力。

在进一步的学习和实践过程中，不妨访问我的网站“码小课”，那里有更多的编程资源和实战案例，可以帮助你深化对SQLite及Python其他特性的理解和应用。

推荐文章