跳转至

SQLite Python 教程

原文: http://zetcode.com/db/sqlitepythontutorial/

该教程已被 Python SQLite 教程取代。

这是用于 SQLite 数据库的 Python 编程教程。 它涵盖了使用 Python 语言进行 SQLite 编程的基础。 您可能还需要查看 ZetCode 上的 Python 教程SQLite 教程MySQL Python 教程PostgreSQL Python 教程

要使用本教程,我们必须在系统上安装 Python 语言,SQLite 数据库,pysqlite语言绑定和sqlite3命令行工具。

Tweet

如果我们拥有 Python 2.5+,则只需安装sqlite3命令行工具。 SQLite 库和pysqlite语言绑定都内置在 Python 语言中。

$ python2
Python 2.7.12 (default, Nov 12 2018, 14:36:49)
[GCC 5.4.0 20160609] on linux2
Type "help", "copyright", "credits" or "license" for more information.

>>> import sqlite3
>>> sqlite3.version
'2.6.0'
>>> sqlite3.sqlite_version
'3.16.2'

在外壳程序中,我们启动 Python 交互式解释器。 我们可以看到 Python 版本。 在我们的例子中是 Python 2.7.12。 sqlite.versionpysqlite(2.6.0)的版本,它是 Python 语言与 SQLite 数据库的绑定。 sqlite3.sqlite_version为我们提供了 SQLite 数据库库的版本。 在我们的例子中,版本是 3.16.2。

SQLite

SQLite 是嵌入式关系数据库引擎。 该文档称其为自包含,无服务器,零配置和事务型 SQL 数据库引擎。 它非常受欢迎,当今全球有数亿本使用。 几种编程语言都内置了对 SQLite 的支持,包括 Python 和 PHP。

创建 SQLite 数据库

现在,我们将使用sqlite3命令行工具创建一个新数据库。

$ sqlite3 test.db
SQLite version 3.16.2 2017-01-06 16:32:41
Enter ".help" for usage hints.
sqlite>

我们为sqlite3 tool提供了一个参数; test.db是数据库名称。 这是我们磁盘上的文件。 如果存在,则将其打开。 如果不是,则创建它。

sqlite> .tables
sqlite> .exit
$ ls
test.db

.tables命令提供了test.db数据库中的表列表。 当前没有表。 .exit命令终止 sqlite3 命令行工具的交互式会话。 ls Unix 命令显示当前工作目录的内容。 我们可以看到test.db文件。 所有数据将存储在该单个文件中。

SQLite 版本示例

在第一个代码示例中,我们将获得 SQLite 数据库的版本。

version.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite
import sys

con = None

try:
    con = lite.connect('test.db')

    cur = con.cursor()
    cur.execute('SELECT SQLITE_VERSION()')

    data = cur.fetchone()[0]

    print "SQLite version: {}".format(data)

except lite.Error, e:

    print "Error {}:".format(e.args[0])
    sys.exit(1)

finally:

    if con:
        con.close()

在上面的 Python 脚本中,我们连接到先前创建的test.db数据库。 我们执行一条 SQL 语句,该语句返回 SQLite 数据库的版本。

import sqlite3 as lite

sqlite3模块用于处理 SQLite 数据库。

con = None

我们将con变量初始化为无。 如果无法创建与数据库的连接(例如磁盘已满),则不会定义连接变量。 这将导致finally子句中的错误。

con = lite.connect('test.db')

在这里,我们连接到test.db数据库。 connect()方法返回一个连接对象。

cur = con.cursor()
cur.execute('SELECT SQLITE_VERSION()')

从连接中,我们得到光标对象。 游标用于遍历结果集中的记录。 我们调用游标的execute()方法并执行 SQL 语句。

data = cur.fetchone()[0]

我们获取数据。 由于只检索一条记录,因此我们称为fetchone()方法。

print "SQLite version: {}".format(data)

我们将检索到的数据打印到控制台。

except lite.Error, e:

    print "Error {}:".format(e.args[0])
    sys.exit(1)

如果发生异常,我们将输出一条错误消息,并以错误代码 1 退出脚本。

finally:

    if con:
        con.close()

在最后一步,我们释放资源。

在第二个示例中,我们再次获得 SQLite 数据库的版本。 这次我们将使用with关键字。

version2.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite

con = lite.connect('test.db')

with con:

    cur = con.cursor()
    cur.execute('SELECT SQLITE_VERSION()')

    data = cur.fetchone()[0]

    print "SQLite version: {}".format(data)

该脚本返回 SQLite 数据库的当前版本。 通过使用with关键字。 代码更紧凑。

with con:

使用with关键字,Python 解释器会自动释放资源。 它还提供错误处理。

$ ./version2.py
SQLite version: 3.16.2

这是输出。

SQLite Python 创建表

我们创建一个cars表并在其中插入几行。

create_table.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite

con = lite.connect('test.db')

with con:

    cur = con.cursor()

    cur.execute("CREATE TABLE cars(id INT, name TEXT, price INT)")
    cur.execute("INSERT INTO cars VALUES(1,'Audi',52642)")
    cur.execute("INSERT INTO cars VALUES(2,'Mercedes',57127)")
    cur.execute("INSERT INTO cars VALUES(3,'Skoda',9000)")
    cur.execute("INSERT INTO cars VALUES(4,'Volvo',29000)")
    cur.execute("INSERT INTO cars VALUES(5,'Bentley',350000)")
    cur.execute("INSERT INTO cars VALUES(6,'Citroen',21000)")
    cur.execute("INSERT INTO cars VALUES(7,'Hummer',41400)")
    cur.execute("INSERT INTO cars VALUES(8,'Volkswagen',21600)")

上面的脚本创建一个cars表,并将 8 行插入到该表中。

cur.execute("CREATE TABLE cars(id INT, name TEXT, price INT)")

该 SQL 语句创建一个新的cars表。 该表有三列。

cur.execute("INSERT INTO cars VALUES(1,'Audi',52642)")
cur.execute("INSERT INTO cars VALUES(2,'Mercedes',57127)")

这两行将两辆车插入表。 使用with关键字,更改将自动提交。 否则,我们将不得不手动提交它们。

sqlite> .mode column
sqlite> .headers on

我们使用sqlite3工具验证写入的数据。 首先,我们修改数据在控制台中的显示方式。 我们使用列模式并打开标题。

sqlite>  select * from cars;
id          name        price
----------  ----------  ----------
1           Audi        52642
2           Mercedes    57127
3           Skoda       9000
4           Volvo       29000
5           Bentley     350000
6           Citroen     21000
7           Hummer      41400
8           Volkswagen  21600

这是我们已写入cars表的数据。

我们将创建相同的表。 这次使用便捷的executemany()方法。

create_table2.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite

cars = (
    (1, 'Audi', 52642),
    (2, 'Mercedes', 57127),
    (3, 'Skoda', 9000),
    (4, 'Volvo', 29000),
    (5, 'Bentley', 350000),
    (6, 'Hummer', 41400),
    (7, 'Volkswagen', 21600)
)

con = lite.connect('test.db')

with con:

    cur = con.cursor()

    cur.execute("DROP TABLE IF EXISTS cars")
    cur.execute("CREATE TABLE cars(id INT, name TEXT, price INT)")
    cur.executemany("INSERT INTO cars VALUES(?, ?, ?)", cars)

程序将删除cars表(如果存在)并重新创建它。

cur.execute("DROP TABLE IF EXISTS cars")
cur.execute("CREATE TABLE cars(id INT, name TEXT, price INT)")

如果存在,则第一个 SQL 语句将删除cars表。 第二条 SQL 语句创建cars表。

cur.executemany("INSERT INTO cars VALUES(?, ?, ?)", cars)

我们使用便捷的executemany()方法将 8 行插入到表中。 此方法的第一个参数是参数化的 SQL 语句。 第二个参数是数据,以元组的元组的形式。

我们提供了另一种创建cars表的方法。 我们手动提交更改并提供我们自己的错误处理。

create_table3.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite
import sys

try:
    con = lite.connect('test.db')

    cur = con.cursor()

    cur.executescript("""
        DROP TABLE IF EXISTS cars;
        CREATE TABLE cars(id INT, name TEXT, price INT);
        INSERT INTO cars VALUES(1,'Audi',52642);
        INSERT INTO cars VALUES(2,'Mercedes',57127);
        INSERT INTO cars VALUES(3,'Skoda',9000);
        INSERT INTO cars VALUES(4,'Volvo',29000);
        INSERT INTO cars VALUES(5,'Bentley',350000);
        INSERT INTO cars VALUES(6,'Citroen',21000);
        INSERT INTO cars VALUES(7,'Hummer',41400);
        INSERT INTO cars VALUES(8,'Volkswagen',21600);
        """)

    con.commit()

except lite.Error, e:

    if con:
        con.rollback()

    print "Error {}:".format(e.args[0])
    sys.exit(1)

finally:

    if con:
        con.close()

在上面的脚本中,我们使用executescript()方法(重新)创建cars表。

cur.executescript("""
    DROP TABLE IF EXISTS cars;
    CREATE TABLE cars(id INT, name TEXT, price INT);
    INSERT INTO cars VALUES(1,'Audi',52642);
    INSERT INTO cars VALUES(2,'Mercedes',57127);
...

executescript()方法允许我们一步执行整个 SQL 代码。

con.commit()

如果没有with关键字,则必须使用commit()方法来提交更改。

except lite.Error, e:

    if con:
        con.rollback()

    print "Error {}:".format(e.args[0])
    sys.exit(1)

发生错误时,所做的更改将回滚,并在终端上显示一条错误消息。

SQLite Python lastrowid

有时,我们需要确定最后插入的行的 ID。 在 Python SQLite 中,我们使用光标对象的lastrowid属性。

lastrowid.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite

con = lite.connect(':memory:')

with con:

    cur = con.cursor()
    cur.execute("CREATE TABLE friends(id INTEGER PRIMARY KEY, name TEXT);")
    cur.execute("INSERT INTO friends(name) VALUES ('Tom');")
    cur.execute("INSERT INTO friends(name) VALUES ('Rebecca');")
    cur.execute("INSERT INTO friends(name) VALUES ('Jim');")
    cur.execute("INSERT INTO friends(name) VALUES ('Robert');")

    last_row_id = cur.lastrowid
    print "The last Id of the inserted row is {}".format(last_row_id)

我们在内存中创建一个friends表。 ID 会自动递增。

cur.execute("CREATE TABLE friends(id INTEGER PRIMARY KEY, name TEXT);")

在 SQLite 中,INTEGER PRIMARY KEY列自动增加。 还有一个AUTOINCREMENT关键字。 在INTEGER PRIMARY KEY AUTOINCREMENT中使用时,会使用稍微不同的 ID 创建算法。

cur.execute("INSERT INTO friends(name) VALUES ('Tom');")
cur.execute("INSERT INTO friends(name) VALUES ('Rebecca');")
cur.execute("INSERT INTO friends(name) VALUES ('Jim');")
cur.execute("INSERT INTO friends(name) VALUES ('Robert');")

使用自动增量时,我们必须明确声明列名,而忽略自动增量的列名。 这四个语句在friends表中插入四行。

last_row_id = cur.lastrowid

使用lastrowid获得最后插入的行 ID。

$ ./lastrowid.py
The last Id of the inserted row is 4

我们看到了程序的输出。

SQLite Python 检索数据

现在我们已经将一些数据插入数据库中,我们想要取回它。

select_all.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite

con = lite.connect('test.db')

with con:

    cur = con.cursor()
    cur.execute("SELECT * FROM cars")

    rows = cur.fetchall()

    for row in rows:
        print row

在此示例中,我们从cars表中检索所有数据。

cur.execute("SELECT * FROM cars")

该 SQL 语句从cars表中选择所有数据。

rows = cur.fetchall()

fetchall()方法获取所有记录。 它返回一个结果集。 从技术上讲,它是一个元组的元组。 每个内部元组代表表中的一行。

for row in rows:
    print row

我们将数据逐行打印到控制台。

$ ./select_all.py
(1, u'Audi', 52642)
(2, u'Mercedes', 57127)
(3, u'Skoda', 9000)
(4, u'Volvo', 29000)
(5, u'Bentley', 350000)
(6, u'Citroen', 21000)
(7, u'Hummer', 41400)
(8, u'Volkswagen', 21600)

这是示例的输出。

一次返回所有数据可能不可行。 我们可以一一读取行。

fetch_one.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite

con = lite.connect('test.db')

with con:

    cur = con.cursor()
    cur.execute("SELECT * FROM cars")

    while True:

        row = cur.fetchone()

        if row == None:
            break

        print row[0], row[1], row[2]

在此脚本中,我们连接到数据库,并逐个读取cars表的行。

while True:

我们从while循环访问数据。 当我们读取最后一行时,循环终止。

row = cur.fetchone()

if row == None:
    break

fetchone()方法返回表的下一行。 如果没有剩余数据,则返回None。 在这种情况下,我们打破了循环。

print row[0], row[1], row[2]

数据以元组的形式返回。 在这里,我们从元组中选择记录。 第一个是 ID,第二个是汽车名称,第三个是汽车的价格。

$ ./fetch_one.py
1 Audi 52642
2 Mercedes 57127
3 Skoda 9000
4 Volvo 29000
5 Bentley 350000
6 Citroen 21000
7 Hummer 41400
8 Volkswagen 21600

这是示例的输出。

SQLite Python 字典游标

默认游标以元组的元组返回数据。 当我们使用字典游标时,数据以 Python 字典的形式发送。 这样,我们可以通过列名称来引用数据。

dictionary_cursor.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite

con = lite.connect('test.db')

with con:

    con.row_factory = lite.Row

    cur = con.cursor()
    cur.execute("SELECT * FROM cars")

    rows = cur.fetchall()

    for row in rows:
        print "{} {} {}".format(row["id"], row["name"], row["price"])

在此示例中,我们使用字典光标打印cars表的内容。

con.row_factory = lite.Row

我们选择一个字典光标。 现在,我们可以按列名访问记录。

for row in rows:
    print "{} {} {}".format(row["id"], row["name"], row["price"])

通过列名访问数据。

SQLite Python 参数化查询

现在,我们将关注参数化查询。 当使用参数化查询时,我们使用占位符,而不是直接将值写入语句。 参数化查询可提高安全性和性能。

Python sqlite3模块支持两种类型的占位符:问号和命名占位符。

parameterized_query.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite

uId = 1
uPrice = 62300

con = lite.connect('test.db')

with con:

    cur = con.cursor()

    cur.execute("UPDATE cars SET price=? WHERE id=?", (uPrice, uId))

    print "Number of rows updated: {}".format(cur.rowcount)

我们更新了一辆车的价格。 在此代码示例中,我们使用问号占位符。

cur.execute("UPDATE cars SET price=? WHERE id=?", (uPrice, uId))

问号?是值的占位符。 这些值将添加到占位符。

print "Number of rows updated: {}".format(cur.rowcount)

rowcount属性返回更新的行数。 在我们的情况下,一行已更新。

第二个示例使用带有命名占位符的参数化语句。

parameterized_query2.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite

uId = 4

con = lite.connect('test.db')

with con:

    cur = con.cursor()

    cur.execute("SELECT name, price FROM cars WHERE Id=:Id", {"Id": uId})

    row = cur.fetchone()
    print row[0], row[1]

我们使用命名的占位符选择汽车的名称和价格。

cur.execute("SELECT name, price FROM cars WHERE Id=:Id", {"Id": uId})

命名的占位符以冒号开头。

SQLite Python 插入图像

在本节中,我们将图像插入到 SQLite 数据库中。 请注意,有些人反对将图像放入数据库。 在这里,我们只展示如何做。 我们不讨论是否将图像保存在数据库中的技术问题。

sqlite> CREATE TABLE images(id INTEGER PRIMARY KEY, data BLOB);

对于此示例,我们创建一个名为Images的新表。 对于图像,我们使用BLOB数据类型,表示二进制大型对象。

insert_image.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite
import sys

def readImage():

    fin = None

    try:
        fin = open("sid.png", "rb")
        img = fin.read()
        return img

    except IOError, e:

        print e
        sys.exit(1)

    finally:

        if fin:
            fin.close()

try:
    con = lite.connect('test.db')

    cur = con.cursor()

    data = readImage()
    binary = lite.Binary(data)
    cur.execute("INSERT INTO images(data) VALUES (?)", (binary,) )

    con.commit()

except lite.Error, e:

    if con:
        con.rollback()

    print e
    sys.exit(1)

finally:

    if con:
        con.close()

在此脚本中,我们从当前工作目录中读取图像,并将其写入 SQLite test.db数据库的images表中。

try:
    fin = open("sid.png", "rb")
    img = fin.read()
    return img

我们从文件系统读取二进制数据。 我们有一个名为sid.png的 JPG 图像。

binary = lite.Binary(data)

使用 SQLite Binary对象对数据进行编码。

cur.execute("INSERT INTO images(data) VALUES (?)", (binary,) )

该 SQL 语句用于将映像插入数据库。

SQLite Python 读取图像

在本节中,我们将执行相反的操作:我们从数据库表中读取一个图像。

read_image.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite
import sys

def writeImage(data):

    fout = None

    try:
        fout = open('sid2.png','wb')
        fout.write(data)

    except IOError, e:

        print e
        sys.exit(1)

    finally:

        if fout:
            fout.close()

try:
    con = lite.connect('test.db')

    cur = con.cursor()
    cur.execute("SELECT data FROM images LIMIT 1")
    data = cur.fetchone()[0]

    writeImage(data)

except lite.Error, e:

    print e
    sys.exit(1)

finally:

    if con:
        con.close()

我们从Images表中读取图像数据,并将其写入另一个文件woman2.jpg中。

try:
    fout = open('sid2.png','wb')
    fout.write(data)

我们以写入模式打开一个二进制文件。 来自数据库的数据被写入文件。

cur.execute("SELECT data FROM images LIMIT 1")
data = cur.fetchone()[0]

这两行从images表中选择并获取数据。 我们从第一行获取二进制数据。

SQLite Python 元数据

元数据是有关数据库中数据的信息。 SQLite 中的元数据包含有关表和列的信息,我们在其中存储数据。 受 SQL 语句影响的行数是元数据。 结果集中返回的行数和列数也属于元数据。

可以使用PRAGMA命令获取 SQLite 中的元数据。 SQLite 对象可能具有属性,即元数据。 最后,我们还可以通过查询 SQLite 系统sqlite_master表来获取特定的元数据。

column_names.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite

con = lite.connect('test.db')

with con:

    cur = con.cursor()

    cur.execute('PRAGMA table_info(cars)')

    data = cur.fetchall()

    for d in data:
        print d[0], d[1], d[2]

在此示例中,我们发出PRAGMA table_info(tableName)命令,以获取有关cars表的一些元数据信息。

cur.execute('PRAGMA table_info(cars)')

PRAGMA table_info(tableName)命令为cars表中的每一列返回一行。 结果集中的列包括列顺序号,列名称,数据类型,该列是否可以为NULL以及该列的默认值。

for d in data:
    print d[0], d[1], d[2]

根据提供的信息,我们打印列顺序号,列名称和列数据类型。

$ ./column_names.py
0 id INT
1 name TEXT
2 price INT

示例的输出。

接下来,我们将打印cars表中的所有行及其列名。

column_names2.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite

con = lite.connect('test.db')

with con:

    cur = con.cursor()
    cur.execute('SELECT * FROM cars')

    col_names = [cn[0] for cn in cur.description]

    rows = cur.fetchall()

    print "{:3} {:10} {:7}".format(col_names[0], col_names[1], col_names[2])

    for row in rows:
        print "{:<3} {:<10} {:7}".format(row[0], row[1], row[2])

我们将cars表的内容打印到控制台。 现在,我们也包括列的名称。 记录与列名对齐。

col_names = [cn[0] for cn in cur.description]

我们从游标对象的description属性获得列名。

print "{:3} {:10} {:7}".format(col_names[0], col_names[1], col_names[2])

此行打印cars表的三个列名。

for row in rows:
    print "{:<3} {:<10} {:7}".format(row[0], row[1], row[2])

我们使用for循环打印行。 数据与列名对齐。

$  ./column_names2.py 
id  name       price  
1   Audi         62300
2   Mercedes     57127
3   Skoda         9000
4   Volvo        29000
5   Bentley     350000
6   Hummer       41400
7   Volkswagen   21600

这是输出。

在与元数据有关的最后一个示例中,我们将列出test.db数据库中的所有表。

list_tables.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite

con = lite.connect('test.db')

with con:

    cur = con.cursor()
    cur.execute("SELECT name FROM sqlite_master WHERE type='table'")

    rows = cur.fetchall()

    for row in rows:
        print row[0]

该代码示例将当前数据库中的所有可用表打印到终端。

cur.execute("SELECT name FROM sqlite_master WHERE type='table'")

表名存储在系统sqlite_master表中。

$ ./list_tables.py
cars
images

这些是我们系统上的表。

SQLite Python 数据导出&导入

我们可以转储 SQL 格式的数据以创建数据库表的简单备份。

export_table.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite

cars = (
    (1, 'Audi', 52643),
    (2, 'Mercedes', 57642),
    (3, 'Skoda', 9000),
    (4, 'Volvo', 29000),
    (5, 'Bentley', 350000),
    (6, 'Hummer', 41400),
    (7, 'Volkswagen', 21600)
)

def writeData(data):

    f = open('cars.sql', 'w')

    with f:
        f.write(data)

con = lite.connect(':memory:')

with con:

    cur = con.cursor()

    cur.execute("DROP TABLE IF EXISTS cars")
    cur.execute("CREATE TABLE cars(id INT, name TEXT, price INT)")
    cur.executemany("INSERT INTO cars VALUES(?, ?, ?)", cars)
    cur.execute("DELETE FROM cars WHERE price < 30000")

    data = '\n'.join(con.iterdump())

    writeData(data)

在上面的示例中,我们在内存中重新创建了cars表。 我们从表中删除一些行,并将表的当前状态转储到cars.sql文件中。 该文件可以用作表的当前备份。

def writeData(data):

    f = open('cars.sql', 'w')

    with f:
        f.write(data)

表中的数据正在写入文件。

con = lite.connect(':memory:')

我们在内存中创建一个临时表。

cur.execute("DROP TABLE IF EXISTS cars")
cur.execute("CREATE TABLE cars(id INT, name TEXT, price INT)")
cur.executemany("INSERT INTO cars VALUES(?, ?, ?)", cars)
cur.execute("DELETE FROM cars WHERE price < 30000")

这些行创建cars表,插入值并删除行,其中price小于 30000 单位。

data = '\n'.join(con.iterdump())

con.iterdump()返回一个迭代器,以 SQL 文本格式转储数据库。 内置的join()函数采用迭代器,并将迭代器中的所有字符串连接在一起,并用新行分隔。 此数据将写入writeData()函数中的cars.sql文件中。

$ cat cars.sql
BEGIN TRANSACTION;
CREATE TABLE cars(id INT, name TEXT, price INT);
INSERT INTO "cars" VALUES(1,'Audi',52643);
INSERT INTO "cars" VALUES(2,'Mercedes',57642);
INSERT INTO "cars" VALUES(5,'Bentley',350000);
INSERT INTO "cars" VALUES(6,'Hummer',41400);
COMMIT;

废弃的内存车表中的内容。

现在,我们将执行反向操作。 我们将转储的表导入回内存。

import_table.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite

def readData():

    f = open('cars.sql', 'r')

    with f:

        data = f.read()

        return data

con = lite.connect(':memory:')

with con:

    cur = con.cursor()

    sql = readData()
    cur.executescript(sql)

    cur.execute("SELECT * FROM cars")

    rows = cur.fetchall()

    for row in rows:
        print row

在此脚本中,我们读取cars.sql文件的内容并执行它。 这将重新创建转储的表。

def readData():

    f = open('cars.sql', 'r')

    with f:

        data = f.read()

        return data

readData()方法内部,我们读取cars.sql表的内容。

cur.executescript(sql)

我们调用executescript()方法来启动 SQL 脚本。

cur.execute("SELECT * FROM cars")

rows = cur.fetchall()

for row in rows:
    print row

我们验证数据。

$ ./import_table.py
(1, u'Audi', 52643)
(2, u'Mercedes', 57642)
(5, u'Bentley', 350000)
(6, u'Hummer', 41400)

输出显示我们已经成功地重新创建了保存的汽车表。

Python SQLite 事务

事务是针对一个或多个数据库中数据的数据库操作的基本单位。 事务中所有 SQL 语句的影响可以全部提交给数据库,也可以全部回滚。

在 SQLite 中,除SELECT以外的任何命令都将启动隐式事务。 同样,在事务中,诸如CREATE TABLE ...,VACUUMPRAGMA之类的命令将在执行之前提交先前的更改。

手动事务以BEGIN TRANSACTION语句开始,并以COMMITROLLBACK语句结束。

SQLite 支持三种非标准事务级别:DEFERREDIMMEDIATEEXCLUSIVE。 SQLite Python 模块还支持自动提交模式,该模式下对表的所有更改均立即生效。

no_commit.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite
import sys

try:
    con = lite.connect('test.db')

    cur = con.cursor()
    cur.execute("DROP TABLE IF EXISTS friends")
    cur.execute("CREATE TABLE friends(id INTEGER PRIMARY KEY, name TEXT)")
    cur.execute("INSERT INTO friends(name) VALUES ('Tom')")
    cur.execute("INSERT INTO friends(name) VALUES ('Rebecca')")
    cur.execute("INSERT INTO friends(name) VALUES ('Jim')")
    cur.execute("INSERT INTO friends(name) VALUES ('Robert')")

    #con.commit()

except lite.Error, e:

    if con:
        con.rollback()

    print e
    sys.exit(1)

finally:

    if con:
        con.close()

我们创建一个friends表,并尝试用数据填充它。 但是,正如我们将看到的,数据未提交。

#con.commit()

注释commit()方法。 如果我们取消注释该行,则数据将被写入表中。

sqlite> .tables
cars     friends  images 
sqlite> SELECT Count(id) FROM friends;
Count(id) 
----------
0         
sqlite>

表已创建,但数据未写入表中。

在第二个示例中,我们演示了一些命令将先前的更改隐式提交到数据库。

implicit_commit.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite
import sys

try:
    con = lite.connect('test.db')
    cur = con.cursor()
    cur.execute("DROP TABLE IF EXISTS friends")
    cur.execute("CREATE TABLE friends(id INTEGER PRIMARY KEY, name TEXT)")
    cur.execute("INSERT INTO friends(name) VALUES ('Tom')")
    cur.execute("INSERT INTO friends(name) VALUES ('Rebecca')")
    cur.execute("INSERT INTO friends(name) VALUES ('Jim')")
    cur.execute("INSERT INTO friends(name) VALUES ('Robert')")

    cur.execute("CREATE TABLE IF NOT EXISTS temporary(id INT)")

except lite.Error, e:

    if con:
        con.rollback()

    print e
    sys.exit(1)

finally:

    if con:
        con.close()

同样,我们没有显式调用commit()命令。 但是这一次,数据被写入了Friends表。

cur.execute("CREATE TABLE IF NOT EXISTS temporary(id INT)")

该 SQL 语句创建一个新表。 它还会提交以前的更改。

$ ./implicit_commit.py

sqlite> SELECT * FROM friends;
id          name
----------  ----------
1           Tom
2           Rebecca
3           Jim
4           Robert

数据已写入friends表。

在自动提交模式下,将立即执行一条 SQL 语句。

autocommit.py

#!/usr/bin/env python2
# -*- coding: utf-8 -*-

import sqlite3 as lite
import sys

try:
    con = lite.connect('test.db', isolation_level=None)

    cur = con.cursor()
    cur.execute("DROP TABLE IF EXISTS friends")
    cur.execute("CREATE TABLE friends(id INTEGER PRIMARY KEY, name TEXT)")
    cur.execute("INSERT INTO friends(name) VALUES ('Tom')")
    cur.execute("INSERT INTO friends(name) VALUES ('Rebecca')")
    cur.execute("INSERT INTO friends(name) VALUES ('Jim')")
    cur.execute("INSERT INTO friends(name) VALUES ('Robert')")

except lite.Error, e:

    print e
    sys.exit(1)

finally:

    if con:
        con.close()

在此示例中,我们以自动提交模式连接到数据库。

con = lite.connect('test.db', isolation_level=None)

当将isolation_level设置为None时,我们具有自动提交模式。

$ ./autocommit.py

sqlite> SELECT * FROM friends;
Id          Name
----------  ----------
1           Tom
2           Rebecca
3           Jim
4           Robert

数据已成功提交到friends表。

这是 SQLite Python 教程。 ZetCode 拥有用于 SQLite Python 的完整电子书: SQLite Python 电子书


我们一直在努力

apachecn/AiLearning

【布客】中文翻译组