本文我們?yōu)榇蠹医榻B Python3 使用 PyMySQL 連接數(shù)據(jù)庫(kù)，并實(shí)現(xiàn)簡(jiǎn)單的增刪改查。

1. 什么是 PyMySQL？

• PyMySQL 是在 Python3.x 版本中用于連接 MySQL 服務(wù)器的一個(gè)庫(kù)，Python2中則使用mysqldb。
• PyMySQL 遵循 Python 數(shù)據(jù)庫(kù) API v2.0 規(guī)范，并包含了 pure-Python MySQL 客戶端庫(kù)。

2. PyMySQL 安裝

• 在使用 PyMySQL 之前，我們需要確保 PyMySQL 已安裝。
• PyMySQL 下載地址：https://github.com/PyMySQL/PyMySQL。
• 如果還未安裝，我們可以使用以下命令安裝最新版的 PyMySQL：

$ pip3 install PyMySQL</pre>

如果你的系統(tǒng)不支持 pip 命令，可以使用以下方式安裝：
1、使用 git 命令下載安裝包安裝(你也可以手動(dòng)下載)：

$ git clone https://github.com/PyMySQL/PyMySQL 
$ cd PyMySQL/ 
$ python3 setup.py install

2、如果需要制定版本號(hào)，可以使用 curl 命令來安裝：

$ # X.X 為 PyMySQL 的版本號(hào) 
$ curl -L https://github.com/PyMySQL/PyMySQL/tarball/pymysql-X.X | tar xz 
$ cd PyMySQL* 
$ python3 setup.py install
$ # 現(xiàn)在你可以刪除 PyMySQL* 目錄

注意：請(qǐng)確保您有root權(quán)限來安裝上述模塊。

安裝的過程中可能會(huì)出現(xiàn)"ImportError: No module named setuptools"的錯(cuò)誤提示，意思是你沒有安裝setuptools，你可以訪問https://pypi.python.org/pypi/setuptools 找到各個(gè)系統(tǒng)的安裝方法。

Linux 系統(tǒng)安裝實(shí)例：

$ wget https://bootstrap.pypa.io/ez_setup.py 
$ python3 ez_setup.py

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

連接數(shù)據(jù)庫(kù)前，請(qǐng)先確認(rèn)以下事項(xiàng)：

• 您已經(jīng)創(chuàng)建了數(shù)據(jù)庫(kù) INCOME.
• 在INCOME數(shù)據(jù)庫(kù)中您已經(jīng)創(chuàng)建了表 EMPLOYER
• EMPLOYER表字段為 first_name, last_name, age, sex和 income。
• 連接數(shù)據(jù)庫(kù)INCOME使用的用戶名為 "root" ，密碼為 "123456",，端口號(hào)為3308，你可以可以自己設(shè)定或者直接使用root用戶名及其密碼，Mysql數(shù)據(jù)庫(kù)用戶授權(quán)請(qǐng)使用Grant命令。
• 在你的機(jī)子上已經(jīng)安裝了 Python MySQLdb 模塊。
• 如果您對(duì)sql語句不熟悉，可以訪問我們的 SQL基礎(chǔ)教程

實(shí)例：

以下實(shí)例鏈接 Mysql 的 income 數(shù)據(jù)庫(kù)：

實(shí)例(Python 3.0+)

#!/usr/bin/python3
import  pymysql
# 打開數(shù)據(jù)庫(kù)連接
db = pymysql.connect(host = 'localhost', user= 'root', passwd = '123456',port=3308, database= 'income')
# 使用 cursor() 方法創(chuàng)建一個(gè)游標(biāo)對(duì)象cursor  
cursor = db.cursor()
# 使用 execute() 方法執(zhí)行 SQL 查詢
cursor.execute("SELECT VERSION()")
# 使用 fetchone() 方法獲取單條數(shù)據(jù).
data = cursor.fetchone()
print  ("Database version : %s " % data)
# 關(guān)閉數(shù)據(jù)庫(kù)連接
db.close()

執(zhí)行以上腳本輸出結(jié)果如下：

Database version : 8.0.12

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

如果數(shù)據(jù)庫(kù)連接存在我們可以使用execute()方法來為數(shù)據(jù)庫(kù)創(chuàng)建表，如下所示創(chuàng)建表employer：

實(shí)例(Python 3.0+)

#!/usr/bin/python3

import pymysql

# 打開數(shù)據(jù)庫(kù)連接
db = pymysql.connect(host = 'localhost', user= 'root', passwd = '123456',port=3308, database= 'income')

# 使用 cursor() 方法創(chuàng)建一個(gè)游標(biāo)對(duì)象 cursor
cursor = db.cursor()

# 使用 execute() 方法執(zhí)行 SQL，如果表存在則刪除
cursor.execute("DROP TABLE IF EXISTS EMPLOYER")

# 使用預(yù)處理語句創(chuàng)建表
sql = """CREATE TABLE EMPLOYER (
         first_name  CHAR(20) NOT NULL,
         last_name  CHAR(20),
         age INT,
         sex CHAR(1),
         income FLOAT )"""

cursor.execute(sql)

執(zhí)行結(jié)果如圖：

image.png

5. 數(shù)據(jù)庫(kù)插入操作

以下實(shí)例使用執(zhí)行 SQL INSERT 語句向表 EMPLOYER 插入記錄：

實(shí)例(Python 3.0+)

#!/usr/bin/python3

import pymysql

# 打開數(shù)據(jù)庫(kù)連接
db = pymysql.connect(host = 'localhost', user= 'root', passwd = '123456',port=3308, database= 'income')

# 使用 cursor() 方法創(chuàng)建一個(gè)游標(biāo)對(duì)象 cursor
cursor = db.cursor()

# SQL 插入語句
sql = """INSERT INTO EMPLOYER(FIRST_NAME,
         LAST_NAME, AGE, SEX, INCOME)
         VALUES ('LI', 'Lei', 30, 'M', 20000)"""
try:
   # 執(zhí)行sql語句
   cursor.execute(sql)
   # 提交到數(shù)據(jù)庫(kù)執(zhí)行
   db.commit()
except:
   # 如果發(fā)生錯(cuò)誤則回滾
   db.rollback()

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

以上例子也可以寫成如下形式：

實(shí)例(Python 3.0+)

#!/usr/bin/python3

import pymysql

# 打開數(shù)據(jù)庫(kù)連接
db = pymysql.connect(host = 'localhost', user= 'root', passwd = '123456',port=3308, database= 'income')

# 使用 cursor() 方法創(chuàng)建一個(gè)游標(biāo)對(duì)象 cursor
cursor = db.cursor()

# SQL 插入語句
sql = "INSERT INTO EMPLOYER(FIRST_NAME, \
       LAST_NAME, AGE, SEX, INCOME) \
       VALUES ('%s', '%s',  %s,  '%s',  %s)" % \
       ('LI', 'Lei', 30, 'M', 20000)

try:
   # 執(zhí)行sql語句
   cursor.execute(sql)
   # 提交到數(shù)據(jù)庫(kù)執(zhí)行
   db.commit()
except:
   # 如果發(fā)生錯(cuò)誤則回滾
   db.rollback()

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

執(zhí)行結(jié)果如下圖

image.png

以下代碼使用變量向SQL語句中傳遞參數(shù):

.................................. 
user_id =  "test123" 
password =  "password" 
con.execute('insert into Login values( %s,  %s)'  % \ 
             (user_id, password))  
..................................

6. 數(shù)據(jù)庫(kù)查詢操作

Python查詢Mysql使用 fetchone() 方法獲取單條數(shù)據(jù), 使用fetchall() 方法獲取多條數(shù)據(jù)。

• fetchone(): 該方法獲取下一個(gè)查詢結(jié)果集。結(jié)果集是一個(gè)對(duì)象
• **fetchall(): **接收全部的返回結(jié)果行.
• rowcount: 這是一個(gè)只讀屬性，并返回執(zhí)行execute()方法后影響的行數(shù)。

實(shí)例：

查詢EMPLOYER表中收入（工資）字段大于1000的所有數(shù)據(jù)：

實(shí)例(Python 3.0+)

#!/usr/bin/python3

import pymysql

# 打開數(shù)據(jù)庫(kù)連接
db = pymysql.connect(host = 'localhost', user= 'root', passwd = '123456',port=3308, database= 'income')

# 使用 cursor() 方法創(chuàng)建一個(gè)游標(biāo)對(duì)象 cursor
cursor = db.cursor()

# SQL 查詢語句
sql = "SELECT * FROM EMPLOYER \
       WHERE INCOME > %s" % (1000)
try:
   # 執(zhí)行SQL語句
   cursor.execute(sql)
   # 獲取所有記錄列表
   results = cursor.fetchall()
   for row in results:
      fname = row[0]
      lname = row[1]
      age = row[2]
      sex = row[3]
      income = row[4]
       # 打印結(jié)果
      print ("fname=%s,lname=%s,age=%s,sex=%s,income=%s" % \
             (fname, lname, age, sex, income ))
except:
   print ("Error: unable to fetch data")

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

以上腳本執(zhí)行結(jié)果如下：

fname=LI,lname=Lei,age=30,sex=M,income=20000.0

7. 數(shù)據(jù)庫(kù)更新操作

更新操作用于更新數(shù)據(jù)表的的數(shù)據(jù)，以下實(shí)例將 INCOME表中 SEX 為 'M' 的 AGE 字段遞增 1：

實(shí)例(Python 3.0+)

#!/usr/bin/python3

import pymysql

# 打開數(shù)據(jù)庫(kù)連接
db = pymysql.connect(host = 'localhost', user= 'root', passwd = '123456',port=3308, database= 'income')

# 使用 cursor() 方法創(chuàng)建一個(gè)游標(biāo)對(duì)象 cursor
cursor = db.cursor()

# SQL 更新語句
sql = "UPDATE EMPLOYER SET AGE = AGE + 1 WHERE SEX = '%c'" % ('M')
try:
   # 執(zhí)行SQL語句
   cursor.execute(sql)
   # 提交到數(shù)據(jù)庫(kù)執(zhí)行
   db.commit()
except:
   # 發(fā)生錯(cuò)誤時(shí)回滾
   db.rollback()

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

8. 刪除操作

刪除操作用于刪除數(shù)據(jù)表中的數(shù)據(jù)，以下實(shí)例演示了刪除數(shù)據(jù)表 EMPLOYER 中 AGE 大于 20 的所有數(shù)據(jù)：

實(shí)例(Python 3.0+)

#!/usr/bin/python3

import pymysql

# 打開數(shù)據(jù)庫(kù)連接
db = pymysql.connect(host = 'localhost', user= 'root', passwd = '123456',port=3308, database= 'income')

# 使用 cursor() 方法創(chuàng)建一個(gè)游標(biāo)對(duì)象 cursor
cursor = db.cursor()

# SQL 刪除語句
sql = "DELETE FROM EMPLOYER WHERE AGE > %s" % (20)
try:
   # 執(zhí)行SQL語句
   cursor.execute(sql)
   # 提交修改
   db.commit()
except:
   # 發(fā)生錯(cuò)誤時(shí)回滾
   db.rollback()

# 關(guān)閉連接
db.close()

9. 執(zhí)行事務(wù)

事務(wù)機(jī)制可以確保數(shù)據(jù)一致性。
事務(wù)應(yīng)該具有4個(gè)屬性：原子性、一致性、隔離性、持久性。這四個(gè)屬性通常稱為ACID特性。

• 原子性（atomicity）。一個(gè)事務(wù)是一個(gè)不可分割的工作單位，事務(wù)中包括的諸操作要么都做，要么都不做。
• 一致性（consistency）。事務(wù)必須是使數(shù)據(jù)庫(kù)從一個(gè)一致性狀態(tài)變到另一個(gè)一致性狀態(tài)。一致性與原子性是密切相關(guān)的。
• 隔離性（isolation）。一個(gè)事務(wù)的執(zhí)行不能被其他事務(wù)干擾。即一個(gè)事務(wù)內(nèi)部的操作及使用的數(shù)據(jù)對(duì)并發(fā)的其他事務(wù)是隔離的，并發(fā)執(zhí)行的各個(gè)事務(wù)之間不能互相干擾。
• 持久性（durability）。持續(xù)性也稱永久性（permanence），指一個(gè)事務(wù)一旦提交，它對(duì)數(shù)據(jù)庫(kù)中數(shù)據(jù)的改變就應(yīng)該是永久性的。接下來的其他操作或故障不應(yīng)該對(duì)其有任何影響。

Python DB API 2.0 的事務(wù)提供了兩個(gè)方法 commit 或 rollback。

實(shí)例(Python 3.0+)

# SQL刪除記錄語句
sql = "DELETE FROM EMPLOYEE WHERE AGE > %s" % (20)
try:
   # 執(zhí)行SQL語句
   cursor.execute(sql)
   # 向數(shù)據(jù)庫(kù)提交
   db.commit()
except:
   # 發(fā)生錯(cuò)誤時(shí)回滾
   db.rollback()

對(duì)于支持事務(wù)的數(shù)據(jù)庫(kù)， 在Python數(shù)據(jù)庫(kù)編程中，當(dāng)游標(biāo)建立之時(shí)，就自動(dòng)開始了一個(gè)隱形的數(shù)據(jù)庫(kù)事務(wù)。
commit()方法游標(biāo)的所有更新操作，rollback（）方法回滾當(dāng)前游標(biāo)的所有操作。每一個(gè)方法都開始了一個(gè)新的事務(wù)。

10. 錯(cuò)誤處理

DB API中定義了一些數(shù)據(jù)庫(kù)操作的錯(cuò)誤及異常，下表列出了這些錯(cuò)誤和異常: