SQLite Python


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

安装


可以使用由Gerhard Haring编写的sqlite3模块将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])

此例程创建一个cursor在使用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方法的快捷方式,它通过调用cursor方法创建中间游标对象,然后使用给定的参数调用游标的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)

此例程是一种快捷方式,它通过调用cursor方法来创建中间光标对象,然后使用给定的参数来调用光标的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()

执行上述程序后,它将在你的计算机中创建COMPANY表。test.db并显示以下消息:

Opened database successfully
Table created successfully

插入操作


以下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

查询操作


以下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

更新操作


以下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

删除操作


以下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