Python3 MySQL 數(shù)據(jù)庫連接

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

什么是 PyMySQL？

PyMySQL 是在 Python3.x 版本中用于連接 MySQL 服務(wù)器的一個(gè)庫，Python2 中則使用mysqldb。

PyMySQL 遵循 Python 數(shù)據(jù)庫 API v2.0 規(guī)范，并包含了 pure-Python MySQL 客戶端庫。

PyMySQL 安裝

在使用 PyMySQL 之前，我們需要確保 PyMySQL 已安裝。

PyMySQL 下載地址：https://github.com/PyMySQL/PyMySQL。

如果還未安裝，我們可以使用以下命令安裝最新版的 PyMySQL：

$ pip install PyMySQL

如果你的系統(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* 目錄

注意：請確保您有 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

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

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

• 您已經(jīng)創(chuàng)建了數(shù)據(jù)庫 TESTDB.
• 在 TESTDB 數(shù)據(jù)庫中您已經(jīng)創(chuàng)建了表 EMPLOYEE
• EMPLOYEE 表字段為 FIRST_NAME, LAST_NAME, AGE, SEX 和 INCOME。
• 連接數(shù)據(jù)庫 TESTDB 使用的用戶名為 "testuser" ，密碼為 "test123",你可以可以自己設(shè)定或者直接使用 root 用戶名及其密碼，Mysql 數(shù)據(jù)庫用戶授權(quán)請使用 Grant 命令。
• 在你的機(jī)子上已經(jīng)安裝了 Python MySQLdb 模塊。
• 如果您對 sql 語句不熟悉，可以訪問我們的 SQL教程

實(shí)例：

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

#!/usr/bin/python3

import pymysql

# 打開數(shù)據(jù)庫連接
db = pymysql.connect("localhost","testuser","test123","TESTDB" )

# 使用 cursor() 方法創(chuàng)建一個(gè)游標(biāo)對象 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ù)庫連接
db.close()

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

Database version : 5.5.20-log

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

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

#!/usr/bin/python3

import pymysql

# 打開數(shù)據(jù)庫連接
db = pymysql.connect("localhost","testuser","test123","TESTDB" )

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

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

# 使用預(yù)處理語句創(chuàng)建表
sql = """CREATE TABLE EMPLOYEE (
         FIRST_NAME  CHAR(20) NOT NULL,
         LAST_NAME  CHAR(20),
         AGE INT,  
         SEX CHAR(1),
         INCOME FLOAT )"""

cursor.execute(sql)

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

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

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

#!/usr/bin/python3

import pymysql

# 打開數(shù)據(jù)庫連接
db = pymysql.connect("localhost","testuser","test123","TESTDB" )

# 使用cursor()方法獲取操作游標(biāo) 
cursor = db.cursor()

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

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

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

#!/usr/bin/python3

import pymysql

# 打開數(shù)據(jù)庫連接
db = pymysql.connect("localhost","testuser","test123","TESTDB" )

# 使用cursor()方法獲取操作游標(biāo) 
cursor = db.cursor()

# SQL 插入語句
sql = "INSERT INTO EMPLOYEE(FIRST_NAME, \
       LAST_NAME, AGE, SEX, INCOME) \
       VALUES ('%s', '%s', '%d', '%c', '%d' )" % \
       ('Mac', 'Mohan', 20, 'M', 2000)
try:
   # 執(zhí)行sql語句
   cursor.execute(sql)
   # 執(zhí)行sql語句
   db.commit()
except:
   # 發(fā)生錯(cuò)誤時(shí)回滾
   db.rollback()

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

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

..................................
user_id = "test123"
password = "password"

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

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

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

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

實(shí)例：

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

#!/usr/bin/python3

import pymysql

# 打開數(shù)據(jù)庫連接
db = pymysql.connect("localhost","testuser","test123","TESTDB" )

# 使用cursor()方法獲取操作游標(biāo) 
cursor = db.cursor()

# SQL 查詢語句
sql = "SELECT * FROM EMPLOYEE \
       WHERE INCOME > '%d'" % (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=%d,sex=%s,income=%d" % \
             (fname, lname, age, sex, income ))
except:
   print ("Error: unable to fecth data")

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

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

fname=Mac, lname=Mohan, age=20, sex=M, income=2000

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

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

#!/usr/bin/python3

import pymysql

# 打開數(shù)據(jù)庫連接
db = pymysql.connect("localhost","testuser","test123","TESTDB" )

# 使用cursor()方法獲取操作游標(biāo) 
cursor = db.cursor()

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

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

刪除操作

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

#!/usr/bin/python3

import pymysql

# 打開數(shù)據(jù)庫連接
db = pymysql.connect("localhost","testuser","test123","TESTDB" )

# 使用cursor()方法獲取操作游標(biāo) 
cursor = db.cursor()

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

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

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

事務(wù)機(jī)制可以確保數(shù)據(jù)一致性。

事務(wù)應(yīng)該具有 4 個(gè)屬性：原子性、一致性、隔離性、持久性。這四個(gè)屬性通常稱為 ACID 特性。

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

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

實(shí)例

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

對于支持事務(wù)的數(shù)據(jù)庫， 在 Python 數(shù)據(jù)庫編程中，當(dāng)游標(biāo)建立之時(shí)，就自動(dòng)開始了一個(gè)隱形的數(shù)據(jù)庫事務(wù)。

commit() 方法游標(biāo)的所有更新操作，rollback（）方法回滾當(dāng)前游標(biāo)的所有操作。每一個(gè)方法都開始了一個(gè)新的事務(wù)。

錯(cuò)誤處理

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