python3 接口測試數據驅動之操做mysql數據庫

18.6   操做mysql數據庫python

18.6.1 python3操做mysql介紹

操做數據庫是程序員經常使用的技能之一,mysql數據庫是中小型項目的首選,Python 標準數據庫接口爲 Python DB-API,Python DB-API爲開發人員提供了數據庫應用編程接口,Python 數據庫接口支持很是多的數據庫,mysql

    不一樣的數據庫你須要下載不一樣的DB API模塊,例如你須要訪問Oracle數據庫和Mysql數據,你須要下載Oracle和MySQL數據庫模塊。程序員

    DB-API 是一個規範. 它定義了一系列必須的對象和數據庫存取方式, 以便爲各類各樣的底層數據庫系統和多種多樣的數據庫接口程序提供一致的訪問接口 。sql

    若是想 Python3 操做 MySQL,就須要安裝MySQL的驅動PyMySQL。數據庫

18.6.2 Python DB-API使用流程

一、引入 API 模塊。編程

二、獲取與數據庫的鏈接。服務器

三、執行SQL語句和存儲過程。併發

四、關閉數據庫鏈接。dom

18.6.3 PyMySQL 是什麼?

PyMySQL 是在 Python3.x 版本中用於鏈接 MySQL 服務器的一個庫,Python2中則使用mysqldb。PyMySQL 遵循 Python 數據庫 API v2.0 規範,幷包含了 pure-Python MySQL 客戶端庫。函數

18.6.4 PyMySQL庫的安裝

使用pip命令安裝:pip install PyMySQL

 

安裝完成以後,在C:\Python35\Lib\site-packages 下就有了pymysql 庫。

 

用 命令pip show PyMySQL查看PyMySQL安裝是否成功。

 

18.6.5 Python鏈接MySql數據庫

鏈接數據庫前,請先確認如下事項:

一、 在你的機子上已經安裝了 Python PyMySQL 模塊。

二、您已經建立了數據庫 test

三、鏈接數據庫test使用的用戶名爲 root,密碼爲 root,你能夠能夠本身設定或者直接使用root用戶名及其密碼。

例子:Python3鏈接MySQL數據,並查詢MySQL數據版本。

 

import pymysql

 

# 打開數據庫鏈接(ip/端口/數據庫用戶名/登陸密碼/數據庫名/編碼)

db = pymysql.connect(host="localhost",port=3306,user="root", password="root",db= "test",charset='utf8')

# 使用 cursor() 方法建立一個遊標對象 cursor

cursor = db.cursor()

 

# 使用 execute()  方法執行 SQL 查詢(查詢mysql的版本)

cursor.execute("SELECT VERSION()")

# 使用 fetchone() 方法獲取單條數據.

data = cursor.fetchone()

print("Database version : %s " % data)

 

# 關閉數據庫鏈接

db.close()

運行結果:

 

18.6.6建立數據庫表

咱們可使用execute()方法來爲數據庫建立表,以下所示建立表EMPLOYEE:

程序實現:

import pymysql

 

# 打開數據庫鏈接

db = pymysql.connect(host="localhost",port=3306,user="root", password="root",db= "test",charset='utf8')

 

# 使用cursor()方法獲取操做遊標

cursor = db.cursor()

 

# 建立數據表SQL語句

sql = """CREATE TABLE EMPLOYEE (

         FIRST_NAME  CHAR(20) NOT NULL,

         LAST_NAME  CHAR(20),

         AGE INT,

         SEX CHAR(1),

         INCOME FLOAT )"""

try:

   # 執行sql語句

    cursor.execute(sql)

   # 提交到數據庫執行

    db.commit()

except:

   # 若是發生錯誤則回滾

    db.rollback()

 

# 關閉數據庫鏈接

db.close()

 

運行結果:

 

18.6.7數據庫插入操做

如下實例使用執行 SQL INSERT 語句向表 EMPLOYEE 插入記錄:

方式一:

import pymysql

 

# 打開數據庫鏈接(ip/數據庫用戶名/登陸密碼/數據庫名)

db = pymysql.connect(host="localhost",port=3306,user="root", password="root",db= "test",charset='utf8')

# 使用 cursor() 方法建立一個遊標對象 cursor

cursor = db.cursor()

 

# SQL 插入語句

sql = """INSERT INTO EMPLOYEE(FIRST_NAME,

         LAST_NAME, AGE, SEX, INCOME)

         VALUES ('xiaoneng', 'xn', 20, 'M', 3000)"""

try:

   # 執行sql語句

   cursor.execute(sql)

   # 提交到數據庫執行

   db.commit()

except:

   # Rollback in case there is any error

   db.rollback()

 

# 關閉數據庫鏈接

db.close()

運行結果:

 

 

方式二:

import pymysql

 

# 打開數據庫鏈接(ip/數據庫用戶名/登陸密碼/數據庫名)

db = pymysql.connect(host="localhost",port=3306,user="root", password="root",db= "test",charset='utf8' )

# 使用 cursor() 方法建立一個遊標對象 cursor

cursor = db.cursor()

 

# SQL 插入語句

sql = "INSERT INTO EMPLOYEE(FIRST_NAME, \

       LAST_NAME, AGE, SEX, INCOME) \

       VALUES ('%s', '%s', '%d', '%c', '%d' )" % \

       ('李四', '小四', 22, 'M', 15000)

 

try:

   # 執行sql語句

   cursor.execute(sql)

   # 提交到數據庫執行

   db.commit()

except:

   # Rollback in case there is any error

   db.rollback()

 

# 關閉數據庫鏈接

db.close()

運行結果:

 

 

18.6.8數據庫查詢操做

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

fetchone(): 該方法獲取下一個查詢結果集。結果集是一個對象

fetchall():接收所有的返回結果行.

rowcount: 這是一個只讀屬性,並返回執行execute()方法後影響的行數。

實例:

查詢EMPLOYEE表中salary(工資)字段大於1000的全部數據:

import pymysql

 

# 打開數據庫鏈接

db = pymysql.connect(host="localhost",port=3306,user="root", password="root",db= "test",charset='utf8' )

 

# 使用cursor()方法獲取操做遊標

cursor = db.cursor()

 

# SQL 查詢語句

sql = "SELECT * FROM EMPLOYEE \

       WHERE INCOME > '%d'" % (1000)

try:

   # 執行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]

      # 打印結果

      print("fname=%s,lname=%s,age=%d,sex=%s,income=%d" % \

             (fname, lname, age, sex, income ))

except:

   print("Error: unable to fecth data")

 

# 關閉數據庫鏈接

db.close()

運行結果:

 

18.6.9數據庫更新操做

更新操做用於更新數據表的的數據,如下實例將 EMPLOYEE 表中的FIRST_NAME = '李四',INCOME字段的值15000改成25000

程序實現:

import pymysql

 

# 打開數據庫鏈接

db = pymysql.connect(host="localhost",port=3306,user="root", password="root",db= "test",charset='utf8' )

 

# 使用cursor()方法獲取操做遊標

cursor = db.cursor()

 

# SQL 更新語句

sql = "UPDATE EMPLOYEE SET INCOME = 25000 WHERE FIRST_NAME = '李四'"

try:

   # 執行SQL語句

   cursor.execute(sql)

   # 提交到數據庫執行

   db.commit()

except:

   # 發生錯誤時回滾

   db.rollback()

 

# 關閉數據庫鏈接

db.close()

運行結果:

 

18.6.10刪除操做

刪除操做用於刪除數據表中的數據,如下實例演示了刪除數據表 EMPLOYEE 中 AGE 大於 20 的全部數據:

程序實現:

import pymysql

 

# 打開數據庫鏈接

db = pymysql.connect(host="localhost",port=3306,user="root", password="root",db= "test",charset='utf8')

 

# 使用cursor()方法獲取操做遊標

cursor = db.cursor()

 

# SQL 刪除語句

sql = "DELETE FROM EMPLOYEE WHERE FIRST_NAME = '李四'"

 

try:

   # 執行SQL語句

   cursor.execute(sql)

   # 提交到數據庫執行

   db.commit()

except:

   # 發生錯誤時回滾

   db.rollback()

 

# 關閉數據庫鏈接

db.close()

 

運行結果:

 

18.6.11處理事務的機制

事務機制能夠確保數據一致性。

事務應該具備4個屬性:原子性、一致性、隔離性、持久性。這四個屬性一般稱爲ACID特性。

一、原子性(atomicity)。一個事務是一個不可分割的工做單位,事務中包括的諸操做要麼都作,要麼都不作。

二、一致性(consistency)。事務必須是使數據庫從一個一致性狀態變到另外一個一致性狀態。一致性與原子性是密切相關的。

三、隔離性(isolation)。一個事務的執行不能被其餘事務干擾。即一個事務內部的操做及使用的數據對併發的其餘事務是隔離的,併發執行的各個事務之間不能互相干擾。四、持久性(durability)。持續性也稱永久性(permanence),指一個事務一旦提交,它對數據庫中數據的改變就應該是永久性的。接下來的其餘操做或故障不該該對其有任何影響。

Python DB API 2.0 的事務提供了兩個方法 commit 或 rollback。

例子:

import pymysql

 

# 打開數據庫鏈接

db = pymysql.connect(host="localhost",port=3306,user="root", password="root",db= "test",charset='utf8' )

 

# 使用cursor()方法獲取操做遊標

cursor = db.cursor()

 

# SQL 刪除語句

sql = "DELETE FROM EMPLOYEE WHERE FIRST_NAME = '李四'"

 

try:

   # 執行SQL語句

   cursor.execute(sql)

   # 向數據庫提交

   db.commit()

except:

   # 發生錯誤時回滾

   db.rollback()

 

# 關閉數據庫鏈接

db.close()

對於支持事務的數據庫, 在Python數據庫編程中,當遊標創建之時,就自動開始了一個隱形的數據庫事務。

commit()方法遊標的全部更新操做,rollback()方法回滾當前遊標的全部操做。每個方法都開始了一個新的事務。

18.6.12錯誤的處理

DB API中定義了一些數據庫操做的錯誤及異常,下表列出了這些錯誤和異常:

異常

描述

Warning

當有嚴重警告時觸發,例如插入數據是被截斷等等。必須是 StandardError 的子類。

Error

警告之外全部其餘錯誤類。必須是 StandardError 的子類。

InterfaceError

當有數據庫接口模塊自己的錯誤(而不是數據庫的錯誤)發生時觸發。 必須是Error的子類。

DatabaseError

和數據庫有關的錯誤發生時觸發。 必須是Error的子類。

DataError

當有數據處理時的錯誤發生時觸發,例如:除零錯誤,數據超範圍等等。 必須是DatabaseError的子類。

OperationalError

指非用戶控制的,而是操做數據庫時發生的錯誤。例如:鏈接意外斷開、 數據庫名未找到、事務處理失敗、內存分配錯誤等等操做數據庫是發生的錯誤。 必須是DatabaseError的子類。

IntegrityError

完整性相關的錯誤,例如外鍵檢查失敗等。必須是DatabaseError子類。

InternalError

數據庫的內部錯誤,例如遊標(cursor)失效了、事務同步失敗等等。 必須是DatabaseError子類。

ProgrammingError

程序錯誤,例如數據表(table)沒找到或已存在、SQL語句語法錯誤、 參數數量錯誤等等。必須是DatabaseError的子類。

NotSupportedError

不支持錯誤,指使用了數據庫不支持的函數或API等。例如在鏈接對象上 使用.rollback()函數,然而數據庫並不支持事務或者事務已關閉。 必須是DatabaseError的子類。

 

18.6.13 python3操做mysql案例

例子一:在一個表(2個字段:id,name)裏隨機插入100條數據。

第一步:先新建一張Test1的表,存放後續的數據。

import pymysql

 

# 打開數據庫鏈接

db = pymysql.connect(host="localhost",port=3306,user="root", password="root",db= "test",charset='utf8' )

 

# 使用cursor()方法獲取操做遊標

cursor = db.cursor()

 

# 建立數據表SQL語句

sql = """CREATE TABLE Test1 (

         ID  CHAR(20) NOT NULL,

         T_NAME  CHAR(20))"""

try:

   # 執行sql語句

    cursor.execute(sql)

   # 提交到數據庫執行

    db.commit()

except:

   # 若是發生錯誤則回滾

    db.rollback()

 

# 關閉數據庫鏈接

db.close()

運行結果:

 

第二步:在表(2個字段:id,name)裏隨機插入100條數據。

程序實現:

import pymysql

import random

conn=pymysql.connect(host="localhost",port=3306,user="root", password="root",db= "test",charset='utf8')

cur=conn.cursor()

for k in range(1,100):    #寫個循環

    id=random.randint(1000,10000)

    t_name="IEX"+str(id)

    sql='insert into Test1(ID,T_NAME) values(%d,"%s")'%(id,t_name)

    cur.execute(sql)

    conn.commit()

cur.execute('select * from Test1')

ret=cur.fetchall()

#print ret

for x in ret :

    print(x)

cur.close()

conn.close()

運行結果:

 

相關文章
相關標籤/搜索