当前位置：首页 > article >正文

python--sqlite

article 2025/2/11 10:38:57

1. 连接到数据库

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

import sqlite3

# 连接到数据库，如果数据库文件不存在则会创建一个新的
conn = sqlite3.connect('example.db')

# 创建一个游标对象，用于执行SQL语句
cursor = conn.cursor()

2. 创建表

使用游标对象的 execute() 方法执行SQL语句来创建表。

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

# 提交事务
conn.commit()

在上述代码中，CREATE TABLE IF NOT EXISTS 语句用于创建一个名为 users 的表，如果该表不存在的话。表包含三个列：id（主键，自动递增）、name（文本类型，不能为空）和 age（整数类型）。

3. 插入数据

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

# 插入单条记录
cursor.execute("INSERT INTO users (name, age) VALUES (?, ?)", ('Alice', 25))

# 插入多条记录
users = [('Bob', 30), ('Charlie', 35)]
cursor.executemany("INSERT INTO users (name, age) VALUES (?, ?)", users)

# 提交事务
conn.commit()

这里使用了参数化查询（? 占位符），可以避免SQL注入攻击。executemany() 方法用于一次性插入多条记录。

4. 查询数据

使用 SELECT 语句从表中查询数据。

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

# 查询特定条件的记录
cursor.execute("SELECT * FROM users WHERE age > 30")
rows = cursor.fetchall()
for row in rows:
    print(row)

fetchall() 方法用于获取查询结果的所有行。

5. 更新数据

使用 UPDATE 语句更新表中的数据。

# 更新 age 为 26 的用户的 name 为 'Alice Smith'
cursor.execute("UPDATE users SET name = ? WHERE age = ?", ('Alice Smith', 25))

# 提交事务
conn.commit()

6. 删除数据

使用 DELETE 语句从表中删除数据。

# 删除 age 大于 35 的用户
cursor.execute("DELETE FROM users WHERE age > 35")

# 提交事务
conn.commit()

7. 关闭连接

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

# 关闭游标
cursor.close()

# 关闭数据库连接
conn.close()

完整示例代码

import sqlite3

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

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

# 插入数据
cursor.execute("INSERT INTO users (name, age) VALUES (?, ?)", ('Alice', 25))
users = [('Bob', 30), ('Charlie', 35)]
cursor.executemany("INSERT INTO users (name, age) VALUES (?, ?)", users)
conn.commit()

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

# 更新数据
cursor.execute("UPDATE users SET name = ? WHERE age = ?", ('Alice Smith', 25))
conn.commit()

# 删除数据
cursor.execute("DELETE FROM users WHERE age > 35")
conn.commit()

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