从零开始的Python开发日记（4）：如何连接SQLite数据库

如何使用Python连接SQLite数据库

SQLite 是一种轻量级的嵌入式数据库，广泛用于小型应用和开发原型。Python 提供了内置的 sqlite3 库，可以方便地连接和操作 SQLite 数据库。本文将介绍如何使用 Python 连接 SQLite 数据库，并包含代码示例。

1. 安装和导入 `sqlite3` 模块

Python 自带 sqlite3 模块，无需额外安装。可以直接在代码中导入。

import sqlite3

2. 连接到数据库

可以使用 sqlite3.connect() 方法连接到 SQLite 数据库。如果指定的数据库文件不存在，将创建一个新的数据库文件。

# 连接到SQLite数据库
conn = sqlite3.connect('example.db')

# 创建一个游标对象
cursor = conn.cursor()

3. 创建表

使用 SQL 语句创建表。例如，创建一个名为 users 的表，包含 id, name, 和 age 列。

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

4. 插入数据

使用 INSERT INTO 语句插入数据。

# 插入数据
cursor.execute('''
INSERT INTO users (name, age)
VALUES ('Alice', 30)
''')

cursor.execute('''
INSERT INTO users (name, age)
VALUES ('Bob', 25)
''')

# 提交事务
conn.commit()

5. 查询数据

使用 SELECT 语句查询数据，并使用游标对象的 fetchall() 方法获取所有结果。

# 查询数据
cursor.execute('SELECT * FROM users')
rows = cursor.fetchall()

# 打印查询结果
for row in rows:
    print(row)

6. 更新数据

使用 UPDATE 语句更新数据。

# 更新数据
cursor.execute('''
UPDATE users
SET age = 31
WHERE name = 'Alice'
''')

# 提交事务
conn.commit()

7. 删除数据

使用 DELETE FROM 语句删除数据。

# 删除数据
cursor.execute('''
DELETE FROM users
WHERE name = 'Bob'
''')

# 提交事务
conn.commit()

8. 关闭连接

完成数据库操作后，关闭游标和连接。

# 关闭游标
cursor.close()

# 关闭连接
conn.close()

完整示例代码

以下是一个完整的示例代码，演示了如何使用 Python 连接 SQLite 数据库并进行基本的数据库操作。

import sqlite3

# 连接到SQLite数据库
conn = sqlite3.connect('example.db')
cursor = conn.cursor()

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

# 插入数据
cursor.execute('''
INSERT INTO users (name, age)
VALUES ('Alice', 30)
''')
cursor.execute('''
INSERT INTO users (name, age)
VALUES ('Bob', 25)
''')

# 提交事务
conn.commit()

# 查询数据
cursor.execute('SELECT * FROM users')
rows = cursor.fetchall()
for row in rows:
    print(row)

# 更新数据
cursor.execute('''
UPDATE users
SET age = 31
WHERE name = 'Alice'
''')
conn.commit()

# 删除数据
cursor.execute('''
DELETE FROM users
WHERE name = 'Bob'
''')
conn.commit()

# 关闭游标和连接
cursor.close()
conn.close()

结论

本文介绍了如何使用 Python 连接 SQLite 数据库，并演示了创建表、插入数据、查询数据、更新数据和删除数据的基本操作。这些示例代码可以帮助你快速上手使用 SQLite 数据库进行开发。