SQLite介紹

SQLite 是一款輕型、小巧、高效的關(guān)系型數(shù)據(jù)庫(kù)管理系統(tǒng)，它可以在任何操作系統(tǒng)中運(yùn)行。而且它的驅(qū)動(dòng)程序也非常豐富，各種語(yǔ)言都有對(duì)應(yīng)的驅(qū)動(dòng)。

這里我們使用 Python 自帶的 sqlite3 模塊來(lái)演示如何使用 Python 操作 SQLite 數(shù)據(jù)庫(kù)。

引入模塊

使用 Python 操作 SQLite 需要先引入 sqlite3 模塊。

import sqlite3

連接數(shù)據(jù)庫(kù)

連接 SQLite 數(shù)據(jù)庫(kù)可以使用 connect() 方法，它有一個(gè)必需參數(shù)是數(shù)據(jù)庫(kù)文件名，如果文件不存在則會(huì)自動(dòng)創(chuàng)建，如果文件存在則會(huì)連接該文件。

# 連接或創(chuàng)建數(shù)據(jù)庫(kù)
conn = sqlite3.connect('test.db')

connect() 方法還可以有其他參數(shù)，例如 timeout，它表示在等待某個(gè)鎖的時(shí)間內(nèi)如果沒(méi)有成功，則會(huì)拋出 sqlite3.BusyError 異常。

創(chuàng)建表

創(chuàng)建表可以執(zhí)行 SQL 語(yǔ)句，也可以使用 Python 操作數(shù)據(jù)庫(kù)的 API。

# 創(chuàng)建表
conn.execute('''
CREATE TABLE if not exists company
(ID INT PRIMARY KEY NOT NULL,
NAME TEXT NOT NULL,
AGE INT NOT NULL,
ADDRESS CHAR(50),
SALARY REAL);
''')

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

插入數(shù)據(jù)需要使用 INSERT 語(yǔ)句，語(yǔ)法為：

conn.execute(f"INSERT INTO company (ID,NAME,AGE,ADDRESS,SALARY)
              VALUES ({id}, '{name}', {age}, '{address}', {salary})")

或者使用參數(shù)化 SQL 語(yǔ)句，語(yǔ)法為：

conn.execute("INSERT INTO company (ID,NAME,AGE,ADDRESS,SALARY) VALUES (?, ?, ?, ?, ?)", 
             (id, name, age, address, salary))

其中 ? 表示占位符，后面的元組中為需要插入的數(shù)據(jù)。

查詢數(shù)據(jù)

查詢數(shù)據(jù)需要使用 SELECT 語(yǔ)句，語(yǔ)法為：

cursor = conn.execute("SELECT ID, NAME, AGE, ADDRESS, SALARY FROM company")
for row in cursor:
    print(f"ID = {row[0]}, NAME = {row[1]}, AGE = {row[2]}, ADDRESS = {row[3]}, SALARY = {row[4]}")

更新數(shù)據(jù)

更新數(shù)據(jù)需要使用 UPDATE 語(yǔ)句，語(yǔ)法為：

conn.execute(f"UPDATE company SET SALARY = {salary} WHERE ID = {id}")

刪除數(shù)據(jù)

刪除數(shù)據(jù)需要使用 DELETE 語(yǔ)句，語(yǔ)法為：

conn.execute(f"DELETE from company where ID = {id}")

提交事務(wù)

Python 操作 SQLite 數(shù)據(jù)庫(kù)自動(dòng)開(kāi)啟一個(gè)事務(wù)，如果要提交事務(wù)，需要使用 commit() 方法。

conn.commit()

關(guān)閉數(shù)據(jù)庫(kù)連接

不需要訪問(wèn)數(shù)據(jù)庫(kù)時(shí)，需要關(guān)閉連接。

conn.close()

總結(jié)

通過(guò)上述代碼可以看出，使用 Python 操作 SQLite 數(shù)據(jù)庫(kù)非常簡(jiǎn)單，只需要使用一些簡(jiǎn)單的 SQL 語(yǔ)句或者 Python API 就可以完成增刪改查等操作。SQLite 數(shù)據(jù)庫(kù)的優(yōu)點(diǎn)是體積小、運(yùn)行速度快，尤其適合一些嵌入式設(shè)備或需要大量讀寫的場(chǎng)景。