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

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

在Python中操作SQLite数据库是一项既基础又强大的技能，它允许你在不需要安装额外数据库服务器的情况下，轻松地在你的项目中集成数据库功能。SQLite是一个轻量级的数据库，它完全嵌入到程序中，因此无需运行一个独立的服务进程或守护进程。这使得SQLite成为小型项目、脚本和测试的理想选择。下面，我们将详细探讨如何在Python中使用SQLite数据库。

引入SQLite

在Python中，你可以通过内置的sqlite3模块来操作SQLite数据库。这个模块提供了一个与SQLite数据库交互的接口，包括创建数据库、执行SQL语句、处理查询结果等功能。

连接数据库

首先，你需要使用sqlite3.connect()函数来创建一个到SQLite数据库的连接。如果指定的数据库文件不存在，SQLite会自动创建一个新的数据库文件。

import sqlite3

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

# 创建一个Cursor: 通过游标执行SQL命令
cur = conn.cursor()

# 执行一些操作...

# 关闭Connection:
conn.close()

创建表

创建表是数据库操作的基本步骤之一。你可以通过执行一个CREATE TABLE SQL语句来完成这个任务。

# 创建一个表
cur.execute('''CREATE TABLE IF NOT EXISTS stocks
             (date text, trans text, symbol text, qty real, price real)''')

# 保存（提交）更改
conn.commit()

这里，CREATE TABLE IF NOT EXISTS是一个SQL语句，用于创建表，但如果表已存在，则不会报错。stocks是表名，括号内定义了表的列名和每列的数据类型。

插入数据

向表中插入数据，可以使用INSERT INTO SQL语句。

# 插入一行记录
cur.execute("INSERT INTO stocks VALUES ('2023-04-01','BUY','RHAT',100,35.14)")

# 插入多行数据，可以使用executemany
data = [('2023-04-01', 'SELL', 'IBM', 1000, 45.00),
        ('2023-04-02', 'BUY', 'MICROSOFT', 100, 72.00)]
cur.executemany("INSERT INTO stocks VALUES (?,?,?,?,?)", data)

# 保存更改
conn.commit()

注意，在executemany方法中，我们使用了参数化查询（?作为占位符），这是一种防止SQL注入的安全做法。

查询数据

查询数据是数据库操作中最常见的任务之一。你可以使用SELECT SQL语句来查询数据，并通过游标（Cursor）获取查询结果。

# 查询所有记录
cur.execute("SELECT * FROM stocks")

# 获取所有记录列表
rows = cur.fetchall()

for row in rows:
    print(row)

# 查询特定条件的数据
cur.execute("SELECT * FROM stocks WHERE symbol='RHAT'")

# 获取一行记录
row = cur.fetchone()
print(row)

fetchall()方法返回查询结果的所有行，而fetchone()方法仅返回结果的第一行。

更新和删除数据

使用UPDATE和DELETE SQL语句可以修改或删除表中的现有数据。

# 更新数据
cur.execute("UPDATE stocks SET price = 35.14 WHERE symbol = 'RHAT'")

# 删除数据
cur.execute("DELETE FROM stocks WHERE symbol = 'IBM'")

# 保存更改
conn.commit()

错误处理

在进行数据库操作时，处理可能出现的错误是非常重要的。你可以使用try-except语句来捕获并处理异常。

try:
    # 尝试执行数据库操作
    cur.execute("INSERT INTO stocks VALUES ('2023-04-01','BUY','RHAT',100,35.14)")
    conn.commit()
except sqlite3.Error as e:
    # 如果发生错误，打印错误信息
    print(f"An error occurred: {e}")
finally:
    # 确保数据库连接被关闭
    conn.close()

使用with语句管理连接

为了更好地管理数据库连接，可以使用with语句。这可以确保即使在发生异常时，连接也能被正确关闭。

import sqlite3

# 使用with语句自动管理连接
with sqlite3.connect('example.db') as conn:
    cur = conn.cursor()
    cur.execute("INSERT INTO stocks VALUES ('2023-04-01','BUY','RHAT',100,35.14)")
    conn.commit()

# 连接在with语句块结束时自动关闭

高级功能

SQLite和sqlite3模块支持许多高级功能，如事务处理、触发器、视图、索引等。这些功能可以极大地增强你的数据库应用的性能和灵活性。

  • 事务处理：SQLite支持ACID事务处理，即原子性（Atomicity）、一致性（Consistency）、隔离性（Isolation）、持久性（Durability）。在Python中，你可以通过调用conn.commit()来提交事务，通过conn.rollback()来回滚事务。

  • 索引：通过在表上创建索引，可以显著提高查询性能。你可以使用CREATE INDEX语句来创建索引。

  • 视图：视图是虚拟的表，其内容由查询定义。你可以像查询普通表一样查询视图，但视图本身不包含数据。

总结

在Python中使用SQLite数据库是一项简单而强大的任务，它允许你轻松地在你的项目中集成数据库功能。通过sqlite3模块，你可以执行各种数据库操作，包括创建和删除表、插入和查询数据、更新和删除记录等。此外，SQLite还支持许多高级功能，如事务处理、索引和视图，这些功能可以进一步增强你的数据库应用的性能和灵活性。在你的码小课网站中分享这些知识，可以帮助更多的开发者了解和掌握Python中的SQLite数据库操作技巧。

推荐文章