快捷導(dǎo)航

python上下文管理器協(xié)議的實(shí)現(xiàn)

更新時(shí)間：2022年06月23日 16:02:58 作者：henry_rhy

這篇文章主要介紹了python上下文管理器協(xié)議的實(shí)現(xiàn)，在python中所有實(shí)現(xiàn)了上下文管理器協(xié)議的對(duì)象都可以用使用with操作，with啟動(dòng)了對(duì)象的上下文管理器

前言

在上下文管理器協(xié)議的過(guò)程中，涉及到兩個(gè)魔術(shù)方法__enter__方法和 __exit__方法

在python中所有實(shí)現(xiàn)了上下文管理器協(xié)議的對(duì)象都可以用使用with操作
with啟動(dòng)了對(duì)象的上下文管理器

上下文管理器協(xié)議：

__enter__方法: 進(jìn)入enter方法返回的結(jié)果被as后面的變量接收
exit: 退出with中所有的語(yǔ)句執(zhí)行完畢執(zhí)行執(zhí)行 exit

實(shí)現(xiàn)一個(gè)簡(jiǎn)單的文件操作來(lái)看下上下文管理器協(xié)議：

class MyOpen:
?? ?# 實(shí)例化
? ? def __init__(self, filename, mode, encoding):
? ? ? ? self.filename = filename
? ? ? ? self.mode = mode
? ? ? ? self.encoding = encoding

? ? def __enter__(self):
? ? ? ? print("---enter---方法")
? ? ? ? # 執(zhí)行文件打開(kāi)操作
? ? ? ? self.f = open(self.filename, self.mode, encoding=self.encoding)
? ? ? ? return self.f

? ? def __exit__(self, exc_type, exc_val, exc_tb):
? ? ? ? """
? ? ? ? :param exc_type: 異常類(lèi)型
? ? ? ? :param exc_val: 異常信息
? ? ? ? :param exc_tb: 異常溯源對(duì)象
? ? ? ? :return:
? ? ? ? """
? ? ? ? print('----enter---')
? ? ? ? self.f.close()
with MyOpen('hr.txt', 'w', encoding='utf-8') as f:
? ? print(f.write('當(dāng)前打開(kāi)了文件，寫(xiě)入了數(shù)據(jù)：23323232'))

用pymysql實(shí)現(xiàn)一個(gè)操作數(shù)據(jù)庫(kù)的類(lèi)，實(shí)現(xiàn)上下文管理器協(xié)議，實(shí)現(xiàn)退出上下文時(shí)，自動(dòng)關(guān)閉游標(biāo)，斷開(kāi)連接

todo:版本1

# todo:版本1：
class mysql_db(object):
? ? #實(shí)例化屬性
? ? def __init__(self):

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

? ? ? ? self.cou = pymysql.connect(
? ? ? ? ? ? host= "數(shù)據(jù)庫(kù)主機(jī)地址", ?
? ? ? ? ? ? port= 端口, ?
? ? ? ? ? ? user="登錄數(shù)據(jù)庫(kù)的賬號(hào)", ?
? ? ? ? ? ? password="登錄數(shù)據(jù)庫(kù)的密碼",?
? ? ? ? ? ? database="數(shù)據(jù)庫(kù)名稱(chēng)", ?
? ? ? ? ? ? charset='utf8', ? ? 編碼格式
? ? ? ? ? ? cursorclass=pymysql.cursors.DictCursor ? ? 將默認(rèn)的元組格式轉(zhuǎn)換成字典格式輸出
? ? ? ? )?

2.創(chuàng)建游標(biāo)

? ? ? ? self.cur = self.cou.cursor()
? ? def __enter__(self):
? ? ? ? return self.cur ? ? ? 返回cur對(duì)象
? ? def __exit__(self, exc_type, exc_val, exc_tb):
? ? ? ? """
? ? ? ? :param exc_type: 異常類(lèi)型
? ? ? ? :param exc_val: 異常信息
? ? ? ? :param exc_tb: 異常溯源對(duì)象
? ? ? ? :return:
? ? ? ? """
? ? ? ? #關(guān)閉游標(biāo)
? ? ? ? self.cur.close()
? ? ? ?# 關(guān)閉數(shù)據(jù)庫(kù)連接
? ? ? ? self.cou.close()
def Obtain_one_date():
? ? with mysql_db() as db:
? ? ? ? db.execute('select * from t_customer LIMIT 4') ? ? 使用execute方法進(jìn)行查詢(xún)語(yǔ)句
? ? ? ? content = db.fetchone() ?返回一條數(shù)據(jù)的查詢(xún)的結(jié)果
? ? ? ? print(content)

# 函數(shù)調(diào)用
Obtain_one_date()

todo:版本2

sql = 'select * from t_customer LIMIT 4'
def mysql_db1(**kwargs):
? ? return pymysql.connect(host=kwargs.get('host', 'xxxx'),
? ? ? ? ? ? ? ? ? ? ? ? ? ?user=kwargs.get("user",'xxxx'),
? ? ? ? ? ? ? ? ? ? ? ? ? ?passwd=kwargs.get("passwd",'xxxx'),
? ? ? ? ? ? ? ? ? ? ? ? ? ?database=kwargs.get("database",'xxxx'),
? ? ? ? ? ? ? ? ? ? ? ? ? ?port=kwargs.get('port', xxxx),
? ? ? ? ? ? ? ? ? ? ? ? ? ?charset=kwargs.get('charset', 'utf8'))

1.創(chuàng)建數(shù)據(jù)庫(kù)連接對(duì)象

cou = mysql_db1()

2.創(chuàng)建游標(biāo)

with cou.cursor() as cu:
? ? cu.execute(sql) ? ? ?使用execute方法進(jìn)行查詢(xún)語(yǔ)句
? ? commt = cu.fetchone() ? ? 返回一條數(shù)據(jù)的查詢(xún)的結(jié)果
? ? print(commt)

# 函數(shù)調(diào)用
mysql_db1()

到此這篇關(guān)于python上下文管理器協(xié)議的實(shí)現(xiàn)的文章就介紹到這了,更多相關(guān)python上下文管理器內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: