SQLite是一種嵌入式數據庫，它的數據庫就是一個文件。由于SQLite本身是C寫的，而且體積很小，所以，經常被集成到各種應用程序中，甚至在iOS和Android的App中都可以集成。 Python就內置了SQLite3，所以，在Python中使用SQLite，不需要安裝任何東西，直接使用。 在使用SQLite前，我們先要搞清楚幾個概念： 表是數據庫中存放關系數據的集合，一個數據庫里面通常都包含多個表，比如學生的表，班級的表，學校的表，等等。表和表之間通過外鍵關聯。 要操作關系數據庫，首先需要連接到數據庫，一個數據庫連接稱為Connection； 連接到數據庫后，需要打開游標，稱之為Cursor，通過Cursor執行SQL語句，然后，獲得執行結果。 Python定義了一套操作數據庫的API接口，任何數據庫要連接到Python，只需要提供符合Python標準的數據庫驅動即可。 由于SQLite的驅動內置在Python標準庫中，所以我們可以直接來操作SQLite數據庫。 我們在Python交互式命令行實踐一下： ~~~ # 導入SQLite驅動: >>> import sqlite3 # 連接到SQLite數據庫 # 數據庫文件是test.db # 如果文件不存在，會自動在當前目錄創建: >>> conn = sqlite3.connect('test.db') # 創建一個Cursor: >>> cursor = conn.cursor() # 執行一條SQL語句，創建user表: >>> cursor.execute('create table user (id varchar(20) primary key, name varchar(20))') <sqlite3.Cursor object at 0x10f8aa260> # 繼續執行一條SQL語句，插入一條記錄: >>> cursor.execute('insert into user (id, name) values (\'1\', \'Michael\')') <sqlite3.Cursor object at 0x10f8aa260> # 通過rowcount獲得插入的行數: >>> cursor.rowcount 1 # 關閉Cursor: >>> cursor.close() # 提交事務: >>> conn.commit() # 關閉Connection: >>> conn.close() ~~~ 我們再試試查詢記錄： ~~~ >>> conn = sqlite3.connect('test.db') >>> cursor = conn.cursor() # 執行查詢語句: >>> cursor.execute('select * from user where id=?', '1') <sqlite3.Cursor object at 0x10f8aa340> # 獲得查詢結果集: >>> values = cursor.fetchall() >>> values [('1', 'Michael')] >>> cursor.close() >>> conn.close() ~~~ 使用Python的DB-API時，只要搞清楚`Connection`和`Cursor`對象，打開后一定記得關閉，就可以放心地使用。 使用`Cursor`對象執行`insert`，`update`，`delete`語句時，執行結果由`rowcount`返回影響的行數，就可以拿到執行結果。 使用`Cursor`對象執行`select`語句時，通過`featchall()`可以拿到結果集。結果集是一個list，每個元素都是一個tuple，對應一行記錄。 如果SQL語句帶有參數，那么需要把參數按照位置傳遞給`execute()`方法，有幾個`?`占位符就必須對應幾個參數，例如： ~~~ cursor.execute('select * from user where id=?', '1') ~~~ SQLite支持常見的標準SQL語句以及幾種常見的數據類型。具體文檔請參閱SQLite官方網站。 ### 小結 在Python中操作數據庫時，要先導入數據庫對應的驅動，然后，通過`Connection`對象和`Cursor`對象操作數據。 要確保打開的`Connection`對象和`Cursor`對象都正確地被關閉，否則，資源就會泄露。 如何才能確保出錯的情況下也關閉掉`Connection`對象和`Cursor`對象呢？請回憶`try:...except:...finally:...`的用法。 ### 練習 請編寫函數，在Sqlite中根據分數段查找指定的名字： ~~~ # -*- coding: utf-8 -*- import os, sqlite3 db_file = os.path.join(os.path.dirname(__file__), 'test.db') if os.path.isfile(db_file): os.remove(db_file) # 初始數據: conn = sqlite3.connect(db_file) cursor = conn.cursor() cursor.execute('create table user(id varchar(20) primary key, name varchar(20), score int)') cursor.execute(r"insert into user values ('A-001', 'Adam', 95)") cursor.execute(r"insert into user values ('A-002', 'Bart', 62)") cursor.execute(r"insert into user values ('A-003', 'Lisa', 78)") cursor.close() conn.commit() conn.close() def get_score_in(low, high): ' 返回指定分數區間的名字，按分數從低到高排序 ' # 測試: assert get_score_in(80, 95) == ['Adam'], get_score_in(80, 95) assert get_score_in(60, 80) == ['Bart', 'Lisa'], get_score_in(60, 80) assert get_score_in(60, 100) == ['Bart', 'Lisa', 'Adam'], get_score_in(60, 100) print('Pass') ~~~ ?Run ### 參考源碼 [do_sqlite.py](https://github.com/michaelliao/learn-python3/blob/master/samples/db/do_sqlite.py)