快捷導(dǎo)航

詳解Python 數(shù)據(jù)庫(kù) (sqlite3)應(yīng)用

更新時(shí)間：2016年12月07日 08:40:41 作者：Vamei

本篇文章主要介紹了Python標(biāo)準(zhǔn)庫(kù)14 數(shù)據(jù)庫(kù) (sqlite3)，小編覺得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧。

Python自帶一個(gè)輕量級(jí)的關(guān)系型數(shù)據(jù)庫(kù)SQLite。這一數(shù)據(jù)庫(kù)使用SQL語(yǔ)言。SQLite作為后端數(shù)據(jù)庫(kù)，可以搭配Python建網(wǎng)站，或者制作有數(shù)據(jù)存儲(chǔ)需求的工具。SQLite還在其它領(lǐng)域有廣泛的應(yīng)用，比如HTML5和移動(dòng)端。Python標(biāo)準(zhǔn)庫(kù)中的sqlite3提供該數(shù)據(jù)庫(kù)的接口。

我將創(chuàng)建一個(gè)簡(jiǎn)單的關(guān)系型數(shù)據(jù)庫(kù)，為一個(gè)書店存儲(chǔ)書的分類和價(jià)格。數(shù)據(jù)庫(kù)中包含兩個(gè)表：category用于記錄分類，book用于記錄某個(gè)書的信息。一本書歸屬于某一個(gè)分類，因此book有一個(gè)外鍵(foreign key)，指向catogory表的主鍵id。

創(chuàng)建數(shù)據(jù)庫(kù)

我首先來創(chuàng)建數(shù)據(jù)庫(kù)，以及數(shù)據(jù)庫(kù)中的表。在使用connect()連接數(shù)據(jù)庫(kù)后，我就可以通過定位指針cursor，來執(zhí)行SQL命令：

# By Vamei
import sqlite3

# test.db is a file in the working directory.
conn = sqlite3.connect("test.db")

c = conn.cursor()

# create tables
c.execute('''CREATE TABLE category
   (id int primary key, sort int, name text)''')
c.execute('''CREATE TABLE book
   (id int primary key, 
    sort int, 
    name text, 
    price real, 
    category int,
    FOREIGN KEY (category) REFERENCES category(id))''')

# save the changes
conn.commit()

# close the connection with the database
conn.close()

SQLite的數(shù)據(jù)庫(kù)是一個(gè)磁盤上的文件，如上面的test.db，因此整個(gè)數(shù)據(jù)庫(kù)可以方便的移動(dòng)或復(fù)制。test.db一開始不存在，所以SQLite將自動(dòng)創(chuàng)建一個(gè)新文件。

利用execute()命令，我執(zhí)行了兩個(gè)SQL命令，創(chuàng)建數(shù)據(jù)庫(kù)中的兩個(gè)表。創(chuàng)建完成后，保存并斷開數(shù)據(jù)庫(kù)連接。

插入數(shù)據(jù)

上面創(chuàng)建了數(shù)據(jù)庫(kù)和表，確立了數(shù)據(jù)庫(kù)的抽象結(jié)構(gòu)。下面將在同一數(shù)據(jù)庫(kù)中插入數(shù)據(jù)：

# By Vamei

import sqlite3

conn = sqlite3.connect("test.db")
c  = conn.cursor()

books = [(1, 1, 'Cook Recipe', 3.12, 1),
      (2, 3, 'Python Intro', 17.5, 2),
      (3, 2, 'OS Intro', 13.6, 2),
      ]

# execute "INSERT" 
c.execute("INSERT INTO category VALUES (1, 1, 'kitchen')")

# using the placeholder
c.execute("INSERT INTO category VALUES (?, ?, ?)", [(2, 2, 'computer')])

# execute multiple commands
c.executemany('INSERT INTO book VALUES (?, ?, ?, ?, ?)', books)

conn.commit()
conn.close()

插入數(shù)據(jù)同樣可以使用execute()來執(zhí)行完整的SQL語(yǔ)句。SQL語(yǔ)句中的參數(shù)，使用"?"作為替代符號(hào)，并在后面的參數(shù)中給出具體值。這里不能用Python的格式化字符串，如"%s"，因?yàn)檫@一用法容易受到SQL注入攻擊。

我也可以用executemany()的方法來執(zhí)行多次插入，增加多個(gè)記錄。每個(gè)記錄是表中的一個(gè)元素，如上面的books表中的元素。

查詢

在執(zhí)行查詢語(yǔ)句后，Python將返回一個(gè)循環(huán)器，包含有查詢獲得的多個(gè)記錄。你循環(huán)讀取，也可以使用sqlite3提供的fetchone()和fetchall()方法讀取記錄：

# By Vamei

import sqlite3

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

# retrieve one record
c.execute('SELECT name FROM category ORDER BY sort')
print(c.fetchone())
print(c.fetchone())

# retrieve all records as a list
c.execute('SELECT * FROM book WHERE book.category=1')
print(c.fetchall())

# iterate through the records
for row in c.execute('SELECT name, price FROM book ORDER BY sort'):
  print(row)

更新與刪除

你可以更新某個(gè)記錄，或者刪除記錄：

# By Vamei

conn = sqlite3.connect("test.db")
c = conn.cursor()

c.execute('UPDATE book SET price=? WHERE id=?',(1000, 1))
c.execute('DELETE FROM book WHERE id=2')

conn.commit()
conn.close()

你也可以直接刪除整張表：