快捷導(dǎo)航

python中with語句結(jié)合上下文管理器操作詳解

更新時間：2019年12月19日 10:42:17 作者：老王同鞋

這篇文章主要給大家介紹了關(guān)于python中with語句結(jié)合上下文管理器操作的相關(guān)資料，文中通過示例代碼介紹的非常詳細(xì)，對大家學(xué)習(xí)或者使用python具有一定的參考學(xué)習(xí)價值，需要的朋友們下面來一起學(xué)習(xí)學(xué)習(xí)吧

前言

所謂上下文管理器即在一個類中重寫了__enter__方法和__exit__方法的類就可以成為上下文管理器類。

我們可以通過with語句結(jié)合上下文管理器簡化一些操作。

使用with語句結(jié)合自定義上下文管理器完成數(shù)據(jù)庫相應(yīng)的操作，代碼實現(xiàn)如下：

# 1. 導(dǎo)入模塊
import pymysql

# 創(chuàng)建自定義上下文管理器對象
class MyDatabase(object):
 # 接收參數(shù)并創(chuàng)建數(shù)據(jù)庫連接對象
 def __init__(self, host, port, user, passwd, database):
  self.__db = pymysql.Connection(host, port, user, passwd, database, charset='utf8')

 # 返回數(shù)據(jù)庫連接對象
 def __enter__(self):
  return self.__db

 # 關(guān)閉數(shù)據(jù)庫連接
 def __exit__(self, exc_type, exc_val, exc_tb):
  self.__db.close()


def main():
 # 使用with關(guān)鍵字接收enter返回的對象給db
 with MyDatabase('localhost', 3306, 'root', 'mysql', 'JDDB') as db:
  # 利用db創(chuàng)建游標(biāo)
  cur = db.cursor()
  sql = '''select * from %s'''
  cur.execute(sql, (goods,))
  result = cur.fetchall()
  for i in result:
   print(i)
  # 關(guān)閉游標(biāo)
  cur.close()

# 程序入口
if __name__ == '__main__':
 main()

上下文管理器類的代碼流程：

1.編寫__init__方法用來接收參數(shù)，并創(chuàng)建數(shù)據(jù)庫連接對象；

2.重寫__enter__方法，返回數(shù)據(jù)庫連接對象；

3.重寫__exit__方法，用來關(guān)閉數(shù)據(jù)庫連接；

with語句代碼流程：

1.當(dāng)將創(chuàng)建對象的語句放到with語句里時不會創(chuàng)建對象，而是接受__enter__方法返回的對象并給對象起個別名；

2.使用接受到的對象即數(shù)據(jù)庫連接對象，創(chuàng)建游標(biāo)；

3.編寫SQL語句，并通過游標(biāo)執(zhí)行SQL語句；

4.獲取SQL語句的查詢結(jié)果，并顯示出來；

5.關(guān)閉游標(biāo)；

6.當(dāng)with語句內(nèi)的代碼執(zhí)行完畢后自動執(zhí)行__exit__方法關(guān)閉數(shù)據(jù)庫連接。

注意：with MyDatabase() as db ---> db = MyDatabase().__enter__()

利用with結(jié)合自定義上下文類實現(xiàn)HTTP服務(wù)端：

# 1.導(dǎo)入socket模塊
import socket


class MySocket(object):
 # 2.編寫init方法接收port參數(shù)
 def __init__(self, port):
  self.__port = port
  # 3.創(chuàng)建socket對象
  self.__sk = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

 # 4.編寫enter方法返回套接字對象
 def __enter__(self):
  # 設(shè)置端口復(fù)用
  self.__sk.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, True)
  # 綁定端口
  self.__sk.bind(self.__port)
  # 設(shè)置端口監(jiān)聽
  self.__sk.listen(128)
  # 返回套接字對象
  return self.__sk

 # 5.編寫exit方法關(guān)閉套接字對象
 def __exit__(self, exc_type, exc_val, exc_tb):
  self.__sk.close()


def main():
 # 使用with關(guān)鍵字 并接受返回的套接字對象給sk
 with MySocket(8000) as sk:
  # 等待客戶端連接
  clicent, ip_port = sk.accept()
  recv_data = clicent.recv(1024)
  print(recv_data.decode('utf-8'))
  # 編寫HTTP響應(yīng)報文
  http_line = 'HTTP/1.1 GET 200 OK\r\n'
  http_header = 'Server PWS/1.0\r\n'
  http_body = 'Welcome to index!\r\n'
  send_data = (http_line + http_header + '\r\n' + http_body).encode('utf-8')
  clicent.send(send_data)
  # 關(guān)閉客戶端連接
  clicent.close()

# 編寫程序入口
if __name__ == '__main__':
 main()

自定義上下文管理器類的代碼解讀：

1.編寫__init__方法，用來接收參數(shù)并創(chuàng)建套接字對象；

2.編寫__enter__方法，并使用套接字對象設(shè)置端口復(fù)用、綁定端口、并設(shè)置監(jiān)聽，然后返回套接字對象；

3.編寫__exit__方法，關(guān)閉套接字對象。

with語句代碼解讀：

1.接收enter返回的套接字對象，并起個別名，

2.通過返回套接字對象等待客戶端連接，

3.接收客戶端連接成功后會返回一個新的套接字和IP端口號，

4.使用客戶端套接字發(fā)送HTTP響應(yīng)報文

5.關(guān)閉客戶端連接

6.當(dāng)with語句中的代碼執(zhí)行完畢后自動執(zhí)行__exit__方法，關(guān)閉服務(wù)器連接

總結(jié)