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


18.6   操作mysql數據庫

18.6.1 python3操作mysql介紹

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

    不同的數據庫你需要下載不同的DB API模塊,例如你需要訪問Oracle數據庫和Mysql數據,你需要下載Oracle和MySQL數據庫模塊。

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

    如果想 Python3 操作 MySQL,就需要安裝MySQL的驅動PyMySQL。

18.6.2 Python DB-API使用流程

1、引入 API 模塊。

2、獲取與數據庫的連接。

3、執行SQL語句和存儲過程。

4、關閉數據庫連接。

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數據庫

連接數據庫前,請先確認以下事項:

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

2、您已經創建了數據庫 test

3、連接數據庫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特性。

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

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

3、隔離性(isolation)。一個事務的執行不能被其他事務干擾。即一個事務內部的操作及使用的數據對並發的其他事務是隔離的,並發執行的各個事務之間不能互相干擾。4、持久性(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()

運行結果:

 

 --------------------------------------

個人今日頭條賬號: 聽海8   (上面上傳了很多相關學習的視頻以及我書里的文章,大家想看視頻,可以關注我的今日頭條)

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM