SQLite - Python

在本章中,您将学习如何在 Python 程序中使用 SQLite。


安装

SQLite3 可以使用由 Gerhard Haring 编写的 sqlite3 模块与 Python 集成。 它提供了一个符合 PEP 249 描述的 DB-API 2.0 规范的 SQL 接口。您不需要单独安装此模块,因为它默认与 Python 2.5.x 版本一起提供。

要使用 sqlite3 模块,您必须首先创建一个代表数据库的连接对象,然后可以选择创建一个游标对象,这将帮助您执行所有 SQL 语句。


Python sqlite3 模块 API

以下是重要的 sqlite3 模块例程,它们可以满足您在 Python 程序中使用 SQLite 数据库的要求。 如果您正在寻找更复杂的应用程序,那么您可以查看 Python sqlite3 模块的官方文档。

序号 API & 说明
1

sqlite3.connect(database [,timeout ,other optional arguments])

此 API 打开与 SQLite 数据库文件的连接。 您可以使用 ":memory:" 打开与驻留在 RAM 中而不是磁盘中的数据库的数据库连接。 如果数据库打开成功,它会返回一个连接对象。

当一个数据库被多个连接访问,并且其中一个进程修改了数据库时,SQLite 数据库将被锁定,直到该事务被提交。 timeout 参数指定连接应该等待锁消失多长时间,直到引发异常。 超时参数的默认值为 5.0(五秒)。

如果给定的数据库名称不存在,则此调用将创建数据库。 如果要在除当前目录之外的任何其他位置创建数据库,也可以使用所需路径指定文件名。

2

connection.cursor([cursorClass])

此例程创建一个 游标,它将在您使用 Python 进行数据库编程的整个过程中使用。 此方法接受单个可选参数 cursorClass。 如果提供,这必须是扩展 sqlite3.Cursor 的自定义光标类。

3

cursor.execute(sql [, optional parameters])

该例程执行一条 SQL 语句。 SQL 语句可以是参数化的(即占位符而不是 SQL 文字)。 sqlite3 模块支持两种占位符:问号和命名占位符(命名样式)。

例如 − cursor.execute("insert into people values (?, ?)", (who, age))

4

connection.execute(sql [, optional parameters])

该例程是上述游标对象提供的execute方法的一个快捷方式,它通过调用游标方法创建一个中间游标对象,然后使用给定的参数调用游标的execute方法。

5

cursor.executemany(sql, seq_of_parameters)

此例程针对在序列 sql 中找到的所有参数序列或映射执行 SQL 命令。

6

connection.executemany(sql[, parameters])

这个例程是一个快捷方式,它通过调用 cursor 方法创建一个中间游标对象,然后使用给定的参数调用 cursor.s executemany 方法。

7

cursor.executescript(sql_script)

该例程一次执行多个以脚本形式提供的 SQL 语句。 它首先发出一个 COMMIT 语句,然后执行它作为参数获取的 SQL 脚本。 所有的 SQL 语句都应该用分号 (;) 分隔。

8

connection.executescript(sql_script)

这个例程是一个快捷方式,它通过调用游标方法创建一个中间游标对象,然后使用给定的参数调用游标的 executescript 方法。

9

connection.total_changes()

此例程返回自打开数据库连接以来已修改、插入或删除的数据库行总数。

10

connection.commit()

此方法提交当前事务。 如果您不调用此方法,则自上次调用 commit() 以来所做的任何事情都不会从其他数据库连接中看到。

11

connection.rollback()

此方法回滚自上次调用 commit() 以来对数据库的任何更改。

12

connection.close()

此方法关闭数据库连接。 请注意,这不会自动调用 commit()。 如果您只是关闭数据库连接而不先调用 commit(),您的更改将会丢失!

13

cursor.fetchone()

此方法获取查询结果集的下一行,返回单个序列,或者当没有更多数据可用时返回 None。

14

cursor.fetchmany([size = cursor.arraysize])

该例程获取查询结果的下一组行,并返回一个列表。 当没有更多行可用时,将返回一个空列表。 该方法尝试获取 size 参数所指示的行数.

15

cursor.fetchall()

此例程获取查询结果的所有(剩余)行,并返回一个列表。 当没有行可用时返回一个空列表。


连接数据库

以下 Python 代码展示了如何连接到现有数据库。 如果数据库不存在,则创建它,最后返回一个数据库对象。

#!/usr/bin/python

import sqlite3

conn = sqlite3.connect('test.db')

print "Opened database successfully";

在这里,您还可以提供数据库名称作为特殊名称 :memory: 以在 RAM 中创建数据库。 现在,让我们运行上面的程序,在当前目录中创建我们的数据库 test.db。您可以根据需要更改路径。 将上述代码保存在 sqlite.py 文件中并执行如下所示。 如果数据库创建成功,那么它将显示以下消息。

$chmod +x sqlite.py
$./sqlite.py
Open database successfully

创建表

以下 Python 程序将用于在之前创建的数据库中创建表。

#!/usr/bin/python

import sqlite3

conn = sqlite3.connect('test.db')
print "Opened database successfully";

conn.execute('''CREATE TABLE COMPANY
         (ID INT PRIMARY KEY     NOT NULL,
         NAME           TEXT    NOT NULL,
         AGE            INT     NOT NULL,
         ADDRESS        CHAR(50),
         SALARY         REAL);''')
print "Table created successfully";

conn.close()

执行上述程序时,它将在您的 test.db 中创建 COMPANY 表,并显示以下消息 −

Opened database successfully
Table created successfully

INSERT 操作

下面的 Python 程序展示了如何在上面示例中创建的 COMPANY 表中创建记录。

#!/usr/bin/python

import sqlite3

conn = sqlite3.connect('test.db')
print "Opened database successfully";

conn.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
      VALUES (1, 'Paul', 32, 'California', 20000.00 )");

conn.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
      VALUES (2, 'Allen', 25, 'Texas', 15000.00 )");

conn.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
      VALUES (3, 'Teddy', 23, 'Norway', 20000.00 )");

conn.execute("INSERT INTO COMPANY (ID,NAME,AGE,ADDRESS,SALARY) \
      VALUES (4, 'Mark', 25, 'Rich-Mond ', 65000.00 )");

conn.commit()
print "Records created successfully";
conn.close()

执行上述程序时,它将在 COMPANY 表中创建给定的记录,并显示以下两行 −

Opened database successfully
Records created successfully

SELECT 操作

下面的 Python 程序展示了如何从上面示例中创建的 COMPANY 表中获取和显示记录。

#!/usr/bin/python

import sqlite3

conn = sqlite3.connect('test.db')
print "Opened database successfully";

cursor = conn.execute("SELECT id, name, address, salary from COMPANY")
for row in cursor:
   print "ID = ", row[0]
   print "NAME = ", row[1]
   print "ADDRESS = ", row[2]
   print "SALARY = ", row[3], "\n"

print "Operation done successfully";
conn.close()

执行上述程序时,会产生如下结果。

Opened database successfully
ID = 1
NAME = Paul
ADDRESS = California
SALARY = 20000.0

ID = 2
NAME = Allen
ADDRESS = Texas
SALARY = 15000.0

ID = 3
NAME = Teddy
ADDRESS = Norway
SALARY = 20000.0

ID = 4
NAME = Mark
ADDRESS = Rich-Mond
SALARY = 65000.0

Operation done successfully

UPDATE 操作

以下 Python 代码展示了如何使用 UPDATE 语句更新任何记录,然后从 COMPANY 表中获取并显示更新的记录。

#!/usr/bin/python

import sqlite3

conn = sqlite3.connect('test.db')
print "Opened database successfully";

conn.execute("UPDATE COMPANY set SALARY = 25000.00 where ID = 1")
conn.commit()
print "Total number of rows updated :", conn.total_changes

cursor = conn.execute("SELECT id, name, address, salary from COMPANY")
for row in cursor:
   print "ID = ", row[0]
   print "NAME = ", row[1]
   print "ADDRESS = ", row[2]
   print "SALARY = ", row[3], "\n"

print "Operation done successfully";
conn.close()

执行上述程序时,会产生如下结果。

Opened database successfully
Total number of rows updated : 1
ID = 1
NAME = Paul
ADDRESS = California
SALARY = 25000.0

ID = 2
NAME = Allen
ADDRESS = Texas
SALARY = 15000.0

ID = 3
NAME = Teddy
ADDRESS = Norway
SALARY = 20000.0

ID = 4
NAME = Mark
ADDRESS = Rich-Mond
SALARY = 65000.0

Operation done successfully

DELETE 操作

以下 Python 代码展示了如何使用 DELETE 语句删除任何记录,然后从 COMPANY 表中获取并显示剩余的记录。

#!/usr/bin/python

import sqlite3

conn = sqlite3.connect('test.db')
print "Opened database successfully";

conn.execute("DELETE from COMPANY where ID = 2;")
conn.commit()
print "Total number of rows deleted :", conn.total_changes

cursor = conn.execute("SELECT id, name, address, salary from COMPANY")
for row in cursor:
   print "ID = ", row[0]
   print "NAME = ", row[1]
   print "ADDRESS = ", row[2]
   print "SALARY = ", row[3], "\n"

print "Operation done successfully";
conn.close()

执行上述程序时,会产生如下结果。

Opened database successfully
Total number of rows deleted : 1
ID = 1
NAME = Paul
ADDRESS = California
SALARY = 20000.0

ID = 3
NAME = Teddy
ADDRESS = Norway
SALARY = 20000.0

ID = 4
NAME = Mark
ADDRESS = Rich-Mond
SALARY = 65000.0

Operation done successfully