如何用Python操作MySQL

發(fā)布時(shí)間：2023-04-14 09:34:05 來(lái)源：億速云閱讀：78 作者：iii 欄目：編程語(yǔ)言

這篇文章主要介紹“如何用Python操作MySQL”，在日常操作中，相信很多人在如何用Python操作MySQL問(wèn)題上存在疑惑，小編查閱了各式資料，整理出簡(jiǎn)單好用的操作方法，希望對(duì)大家解答”如何用Python操作MySQL”的疑惑有所幫助！接下來(lái)，請(qǐng)跟著小編一起來(lái)學(xué)習(xí)吧！

一. python操作數(shù)據(jù)庫(kù)介紹

Python 標(biāo)準(zhǔn)數(shù)據(jù)庫(kù)接口為 Python DB-API，Python DB-API為開(kāi)發(fā)人員提供了數(shù)據(jù)庫(kù)應(yīng)用編程接口。Python 數(shù)據(jù)庫(kù)接口支持非常多的數(shù)據(jù)庫(kù)，你可以選擇適合你項(xiàng)目的數(shù)據(jù)庫(kù)：

GadFly
mSQL
MySQL
PostgreSQL
Microsoft SQL Server 2000
Informix
Interbase
Oracle
Sybase …

你可以訪問(wèn)Python數(shù)據(jù)庫(kù)接口及API查看詳細(xì)的支持?jǐn)?shù)據(jù)庫(kù)列表。

不同的數(shù)據(jù)庫(kù)你需要下載不同的DB API模塊，例如你需要訪問(wèn)Oracle數(shù)據(jù)庫(kù)和Mysql數(shù)據(jù)，你需要下載Oracle和MySQL數(shù)據(jù)庫(kù)模塊。

DB-API 是一個(gè)規(guī)范. 它定義了一系列必須的對(duì)象和數(shù)據(jù)庫(kù)存取方式, 以便為各種各樣的底層數(shù)據(jù)庫(kù)系統(tǒng)和多種多樣的數(shù)據(jù)庫(kù)接口程序提供一致的訪問(wèn)接口。

Python的DB-API，為大多數(shù)的數(shù)據(jù)庫(kù)實(shí)現(xiàn)了接口，使用它連接各數(shù)據(jù)庫(kù)后，就可以用相同的方式操作各數(shù)據(jù)庫(kù)。

Python DB-API使用流程：

引入 API 模塊。
獲取與數(shù)據(jù)庫(kù)的連接。
執(zhí)行SQL語(yǔ)句和存儲(chǔ)過(guò)程。
關(guān)閉數(shù)據(jù)庫(kù)連接。

二. python操作MySQL模塊

Python操作MySQL主要使用兩種方式：

DB模塊(原生SQL)

PyMySQL(支持python2.x/3.x)
MySQLdb(目前僅支持python2.x)

ORM框架

SQLAchemy

2.1 PyMySQL模塊

本文主要介紹PyMySQL模塊,MySQLdb使用方式類似

2.1.1 安裝PyMySQL

PyMySQL是一個(gè)Python編寫(xiě)的MySQL驅(qū)動(dòng)程序，讓我們可以用Python語(yǔ)言操作MySQL數(shù)據(jù)庫(kù)。

pip install PyMySQL

2.2 基本使用

#! /usr/bin/env python
# -*- coding: utf-8 -*-
# __author__ = "shuke"
# Date: 2018/5/13
import pymysql
# 創(chuàng)建連接
conn = pymysql.connect(host="127.0.0.1", port=3306, user='zff', passwd='zff123', db='zff', charset='utf8mb4')
# 創(chuàng)建游標(biāo)(查詢數(shù)據(jù)返回為元組格式)
# cursor = conn.cursor()
# 創(chuàng)建游標(biāo)(查詢數(shù)據(jù)返回為字典格式)
cursor = conn.cursor(pymysql.cursors.DictCursor)
# 1. 執(zhí)行SQL,返回受影響的行數(shù)
effect_row1 = cursor.execute("select * from USER")
# 2. 執(zhí)行SQL,返回受影響的行數(shù),一次插入多行數(shù)據(jù)
effect_row2 = cursor.executemany("insert into USER (NAME) values(%s)", [("jack"), ("boom"), ("lucy")])# 3
# 查詢所有數(shù)據(jù),返回?cái)?shù)據(jù)為元組格式
result = cursor.fetchall()
# 增/刪/改均需要進(jìn)行commit提交,進(jìn)行保存
conn.commit()
# 關(guān)閉游標(biāo)
cursor.close()
# 關(guān)閉連接
conn.close()
print(result)
"""
[{'id': 6, 'name': 'boom'}, {'id': 5, 'name': 'jack'}, {'id': 7, 'name': 'lucy'}, {'id': 4, 'name': 'tome'}, {'id': 3, 'name': 'zff'}, {'id': 1, 'name': 'zhaofengfeng'}, {'id': 2, 'name': 'zhaofengfeng02'}]
"""

2.3 獲取最新創(chuàng)建的數(shù)據(jù)自增ID

#! /usr/bin/env python
# -*- coding: utf-8 -*-
# __author__ = "shuke"
# Date: 2018/5/13
import pymysql
# 創(chuàng)建連接
conn = pymysql.connect(host="127.0.0.1", port=3306, user='zff', passwd='zff123', db='zff', charset='utf8mb4')
# 創(chuàng)建游標(biāo)(查詢數(shù)據(jù)返回為元組格式)
cursor = conn.cursor()
# 獲取新創(chuàng)建數(shù)據(jù)自增ID
effect_row = cursor.executemany("insert into USER (NAME)values(%s)", [("eric")])
# 增刪改均需要進(jìn)行commit提交
conn.commit()
# 關(guān)閉游標(biāo)
cursor.close()
# 關(guān)閉連接
conn.close()
new_id = cursor.lastrowid
print(new_id)
"""
8
"""

2.4 查詢操作

#! /usr/bin/env python
# -*- coding: utf-8 -*-
# __author__ = "shuke"
# Date: 2018/5/13
import pymysql
# 創(chuàng)建連接
conn = pymysql.connect(host="127.0.0.1", port=3306, user='zff', passwd='zff123', db='zff', charset='utf8mb4')
# 創(chuàng)建游標(biāo)
cursor = conn.cursor()
cursor.execute("select * from USER")
# 獲取第一行數(shù)據(jù)
row_1 = cursor.fetchone()
# 獲取前n行數(shù)據(jù)
row_2 = cursor.fetchmany(3)
#
# # 獲取所有數(shù)據(jù)
row_3 = cursor.fetchall()
# 關(guān)閉游標(biāo)
cursor.close()
# 關(guān)閉連接
conn.close()
print(row_1)
print(row_2)
print(row_3)

?? 在fetch數(shù)據(jù)時(shí)按照順序進(jìn)行，可以使用cursor.scroll(num,mode)來(lái)移動(dòng)游標(biāo)位置，如：

cursor.scroll(1,mode='relative') # 相對(duì)當(dāng)前位置移動(dòng)
cursor.scroll(2,mode='absolute') # 相對(duì)絕對(duì)位置移動(dòng)

2.5 防止SQL注入

#! /usr/bin/env python
# -*- coding: utf-8 -*-
# __author__ = "shuke"
# Date: 2018/5/13
import pymysql
# 創(chuàng)建連接
conn = pymysql.connect(host="127.0.0.1", port=3306, user='zff', passwd='zff123', db='zff', charset='utf8mb4')
# 創(chuàng)建游標(biāo)
cursor = conn.cursor()
# 存在sql注入情況(不要用格式化字符串的方式拼接SQL)
sql = "insert into USER (NAME) values('%s')" % ('zhangsan',)
effect_row = cursor.execute(sql)
# 正確方式一
# execute函數(shù)接受一個(gè)元組/列表作為SQL參數(shù),元素個(gè)數(shù)只能有1個(gè)
sql = "insert into USER (NAME) values(%s)"
effect_row1 = cursor.execute(sql, ['wang6'])
effect_row2 = cursor.execute(sql, ('wang7',))
# 正確方式二
sql = "insert into USER (NAME) values(%(name)s)"
effect_row1 = cursor.execute(sql, {'name': 'wudalang'})
# 寫(xiě)入插入多行數(shù)據(jù)
effect_row2 = cursor.executemany("insert into USER (NAME) values(%s)", [('ermazi'), ('dianxiaoer')])
# 提交
conn.commit()
# 關(guān)閉游標(biāo)
cursor.close()
# 關(guān)閉連接
conn.close()

這樣，SQL操作就更安全了。如果需要更詳細(xì)的文檔參考PyMySQL文檔吧。不過(guò)好像這些SQL數(shù)據(jù)庫(kù)的實(shí)現(xiàn)還不太一樣，PyMySQL的參數(shù)占位符使用%s這樣的C格式化符，而Python自帶的sqlite3模塊的占位符好像是問(wèn)號(hào)(?)。因此在使用其他數(shù)據(jù)庫(kù)的時(shí)候還是仔細(xì)閱讀文檔吧。

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

上文中的方式存在一個(gè)問(wèn)題,單線程情況下可以滿足,程序需要頻繁的創(chuàng)建釋放連接來(lái)完成對(duì)數(shù)據(jù)庫(kù)的操作,那么,我們的程序/腳本在多線程情況下會(huì)引發(fā)什么問(wèn)題呢?此時(shí),我們就需要使用數(shù)據(jù)庫(kù)連接池來(lái)解決這個(gè)問(wèn)題!

3.1 DBUtils模塊

DBUtils是Python的一個(gè)用于實(shí)現(xiàn)數(shù)據(jù)庫(kù)連接池的模塊。

此連接池有兩種連接模式：

為每個(gè)線程創(chuàng)建一個(gè)連接，線程即使調(diào)用了close方法，也不會(huì)關(guān)閉，只是把連接重新放到連接池，供自己線程再次使用。當(dāng)線程終止時(shí)，連接才會(huì)自動(dòng)關(guān)閉
創(chuàng)建一批連接到連接池，供所有線程共享使用(推薦使用)

3.2 模式一

#! /usr/bin/env python
# -*- coding: utf-8 -*-
# __author__ = "shuke"
# Date: 2018/5/13
from DBUtils.PersistentDB import PersistentDB
import pymysql
POOL = PersistentDB(
 creator=pymysql,# 使用鏈接數(shù)據(jù)庫(kù)的模塊
 maxusage=None,# 一個(gè)鏈接最多被重復(fù)使用的次數(shù)，None表示無(wú)限制
 setsession=[],# 開(kāi)始會(huì)話前執(zhí)行的命令列表。如：["set datestyle to ...", "set time zone ..."]
 ping=0,
 # ping MySQL服務(wù)端，檢查是否服務(wù)可用。# 如：0 = None = never, 1 = default = whenever it is requested, 2 = when a cursor is created, 4 = when a query is executed, 7 = always
 closeable=False,
 # 如果為False時(shí)， conn.close() 實(shí)際上被忽略，供下次使用，再線程關(guān)閉時(shí)，才會(huì)自動(dòng)關(guān)閉鏈接。如果為True時(shí)， conn.close()則關(guān)閉鏈接，那么再次調(diào)用pool.connection時(shí)就會(huì)報(bào)錯(cuò)，因?yàn)橐呀?jīng)真的關(guān)閉了連接（pool.steady_connection()可以獲取一個(gè)新的鏈接）
 threadlocal=None,# 本線程獨(dú)享值得對(duì)象，用于保存鏈接對(duì)象，如果鏈接對(duì)象被重置
 host='127.0.0.1',
 port=3306,
 user='zff',
 password='zff123',
 database='zff',
 charset='utf8',
)
def func():
 conn = POOL.connection(shareable=False)
 cursor = conn.cursor()
 cursor.execute('select * from USER')
 result = cursor.fetchall()
 cursor.close()
 conn.close()
 return result
result = func()
print(result)

3.2 模式二

#! /usr/bin/env python
# -*- coding: utf-8 -*-
# __author__ = "shuke"
# Date: 2018/5/13
import time
import pymysql
import threading
from DBUtils.PooledDB import PooledDB, SharedDBConnection
POOL = PooledDB(
 creator=pymysql,# 使用鏈接數(shù)據(jù)庫(kù)的模塊
 maxconnections=6,# 連接池允許的最大連接數(shù)，0和None表示不限制連接數(shù)
 mincached=2,# 初始化時(shí)，鏈接池中至少創(chuàng)建的空閑的鏈接，0表示不創(chuàng)建
 maxcached=5,# 鏈接池中最多閑置的鏈接，0和None不限制
 maxshared=3,
 # 鏈接池中最多共享的鏈接數(shù)量，0和None表示全部共享。PS: 無(wú)用，因?yàn)閜ymysql和MySQLdb等模塊的 threadsafety都為1，所有值無(wú)論設(shè)置為多少，_maxcached永遠(yuǎn)為0，所以永遠(yuǎn)是所有鏈接都共享。
 blocking=True,# 連接池中如果沒(méi)有可用連接后，是否阻塞等待。True，等待；False，不等待然后報(bào)錯(cuò)
 maxusage=None,# 一個(gè)鏈接最多被重復(fù)使用的次數(shù)，None表示無(wú)限制
 setsession=[],# 開(kāi)始會(huì)話前執(zhí)行的命令列表。如：["set datestyle to ...", "set time zone ..."]
 ping=0,
 # ping MySQL服務(wù)端，檢查是否服務(wù)可用。# 如：0 = None = never, 1 = default = whenever it is requested, 2 = when a cursor is created, 4 = when a query is executed, 7 = always
 host='127.0.0.1',
 port=3306,
 user='zff',
 password='zff123',
 database='zff',
 charset='utf8'
)
def func():
 # 檢測(cè)當(dāng)前正在運(yùn)行連接數(shù)的是否小于最大鏈接數(shù)，如果不小于則：等待或報(bào)raise TooManyConnections異常
 # 否則
 # 則優(yōu)先去初始化時(shí)創(chuàng)建的鏈接中獲取鏈接 SteadyDBConnection。
 # 然后將SteadyDBConnection對(duì)象封裝到PooledDedicatedDBConnection中并返回。
 # 如果最開(kāi)始創(chuàng)建的鏈接沒(méi)有鏈接，則去創(chuàng)建一個(gè)SteadyDBConnection對(duì)象，再封裝到PooledDedicatedDBConnection中并返回。
 # 一旦關(guān)閉鏈接后，連接就返回到連接池讓后續(xù)線程繼續(xù)使用。
 conn = POOL.connection()
 # print('連接被拿走了', conn._con)
 # print('池子里目前有', POOL._idle_cache, 'rn')
 cursor = conn.cursor()
 cursor.execute('select * from USER')
 result = cursor.fetchall()
 conn.close()
 return result
result = func()
print(result)

?? 由于pymysql、MySQLdb等threadsafety值為1，所以該模式連接池中的線程會(huì)被所有線程共享,因此是線程安全的。如果沒(méi)有連接池，使用pymysql來(lái)連接數(shù)據(jù)庫(kù)時(shí)，單線程應(yīng)用完全沒(méi)有問(wèn)題，但如果涉及到多線程應(yīng)用那么就需要加鎖，一旦加鎖那么連接勢(shì)必就會(huì)排隊(duì)等待，當(dāng)請(qǐng)求比較多時(shí)，性能就會(huì)降低了。

3.3 加鎖

#! /usr/bin/env python
# -*- coding: utf-8 -*-
# __author__ = "shuke"
# Date: 2018/5/13
import pymysql
import threading
from threading import RLock
LOCK = RLock()
CONN = pymysql.connect(host='127.0.0.1',
port=3306,
user='zff',
password='zff123',
database='zff',
charset='utf8')
def task(arg):
 with LOCK:
 cursor = CONN.cursor()
 cursor.execute('select * from USER ')
 result = cursor.fetchall()
 cursor.close()
 print(result)
for i in range(10):
 t = threading.Thread(target=task, args=(i,))
 t.start()

3.4 無(wú)鎖(報(bào)錯(cuò))

#! /usr/bin/env python
# -*- coding: utf-8 -*-
# __author__ = "shuke"
# Date: 2018/5/13
import pymysql
import threading
CONN = pymysql.connect(host='127.0.0.1',
port=3306,
user='zff',
password='zff123',
database='zff',
charset='utf8')
def task(arg):
 cursor = CONN.cursor()
 cursor.execute('select * from USER ')
 # cursor.execute('select sleep(10)')
 result = cursor.fetchall()
 cursor.close()
 print(result)
for i in range(10):
 t = threading.Thread(target=task, args=(i,))
 t.start()

此時(shí)可以在數(shù)據(jù)庫(kù)中查看連接情況: show status like 'Threads%';

四. 數(shù)據(jù)庫(kù)連接池結(jié)合pymsql使用

# cat sql_helper.py
import pymysql
import threading
from DBUtils.PooledDB import PooledDB, SharedDBConnection
POOL = PooledDB(
 creator=pymysql,# 使用鏈接數(shù)據(jù)庫(kù)的模塊
 maxconnections=20,# 連接池允許的最大連接數(shù)，0和None表示不限制連接數(shù)
 mincached=2,# 初始化時(shí)，鏈接池中至少創(chuàng)建的空閑的鏈接，0表示不創(chuàng)建
 maxcached=5,# 鏈接池中最多閑置的鏈接，0和None不限制
 #maxshared=3,# 鏈接池中最多共享的鏈接數(shù)量，0和None表示全部共享。PS: 無(wú)用，因?yàn)閜ymysql和MySQLdb等模塊的 threadsafety都為1，所有值無(wú)論設(shè)置為多少，_maxcached永遠(yuǎn)為0，所以永遠(yuǎn)是所有鏈接都共享。
 blocking=True,# 連接池中如果沒(méi)有可用連接后，是否阻塞等待。True，等待；False，不等待然后報(bào)錯(cuò)
 maxusage=None,# 一個(gè)鏈接最多被重復(fù)使用的次數(shù)，None表示無(wú)限制
 setsession=[],# 開(kāi)始會(huì)話前執(zhí)行的命令列表。如：["set datestyle to ...", "set time zone ..."]
 ping=0,
 # ping MySQL服務(wù)端，檢查是否服務(wù)可用。# 如：0 = None = never, 1 = default = whenever it is requested, 2 = when a cursor is created, 4 = when a query is executed, 7 = always
 host='192.168.11.38',
 port=3306,
 user='root',
 passwd='apNXgF6RDitFtDQx',
 db='m2day03db',
 charset='utf8'
)
def connect():
 # 創(chuàng)建連接
 # conn = pymysql.connect(host='192.168.11.38', port=3306, user='root', passwd='apNXgF6RDitFtDQx', db='m2day03db')
 conn = POOL.connection()
 # 創(chuàng)建游標(biāo)
 cursor = conn.cursor(pymysql.cursors.DictCursor)
 return conn,cursor
def close(conn,cursor):
 # 關(guān)閉游標(biāo)
 cursor.close()
 # 關(guān)閉連接
 conn.close()
def fetch_one(sql,args):
 conn,cursor = connect()
 # 執(zhí)行SQL，并返回收影響行數(shù)
 effect_row = cursor.execute(sql,args)
 result = cursor.fetchone()
 close(conn,cursor)
 return result
def fetch_all(sql,args):
 conn, cursor = connect()
 # 執(zhí)行SQL，并返回收影響行數(shù)
 cursor.execute(sql,args)
 result = cursor.fetchall()
 close(conn, cursor)
 return result
def insert(sql,args):
 """
 創(chuàng)建數(shù)據(jù)
 :param sql: 含有占位符的SQL
 :return:
 """
 conn, cursor = connect()
 # 執(zhí)行SQL，并返回收影響行數(shù)
 effect_row = cursor.execute(sql,args)
 conn.commit()
 close(conn, cursor)
def delete(sql,args):
 """
 創(chuàng)建數(shù)據(jù)
 :param sql: 含有占位符的SQL
 :return:
 """
 conn, cursor = connect()
 # 執(zhí)行SQL，并返回收影響行數(shù)
 effect_row = cursor.execute(sql,args)
 conn.commit()
 close(conn, cursor)
 return effect_row
def update(sql,args):
 conn, cursor = connect()
 # 執(zhí)行SQL，并返回收影響行數(shù)
 effect_row = cursor.execute(sql, args)
 conn.commit()
 close(conn, cursor)
 return effect_row

PS: 可以利用靜態(tài)方法封裝到一個(gè)類中,方便使用

到此，關(guān)于“如何用Python操作MySQL”的學(xué)習(xí)就結(jié)束了，希望能夠解決大家的疑惑。理論與實(shí)踐的搭配能更好的幫助大家學(xué)習(xí)，快去試試吧！若想繼續(xù)學(xué)習(xí)更多相關(guān)知識(shí)，請(qǐng)繼續(xù)關(guān)注億速云網(wǎng)站，小編會(huì)繼續(xù)努力為大家?guī)?lái)更多實(shí)用的文章！

向AI問(wèn)一下細(xì)節(jié)