前言

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

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

使用with語句結合自定義上下文管理器完成資料庫相應的操作，程式碼實現如下：

# 1. 匯入模組
import pymysql

# 建立自定義上下文管理器物件
class MyDatabase(object):
 # 接收引數並建立資料庫連線物件
 def __init__(self,host,port,user,passwd,database):
  self.__db = pymysql.Connection(host,database,charset='utf8')

 # 返回資料庫連線物件
 def __enter__(self):
  return self.__db

 # 關閉資料庫連線
 def __exit__(self,exc_type,exc_val,exc_tb):
  self.__db.close()


def main():
 # 使用with關鍵字接收enter返回的物件給db
 with MyDatabase('localhost',3306,'root','mysql','JDDB') as db:
  # 利用db建立遊標
  cur = db.cursor()
  sql = '''select * from %s'''
  cur.execute(sql,(goods,))
  result = cur.fetchall()
  for i in result:
   print(i)
  # 關閉遊標
  cur.close()

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

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

1.編寫__init__方法用來接收引數，並建立資料庫連線物件；

2.重寫__enter__方法，返回資料庫連線物件；

3.重寫__exit__方法，用來關閉資料庫連線；

with語句程式碼流程：

1.當將建立物件的語句放到with語句裡時不會建立物件，而是接受__enter__方法返回的物件並給物件起個別名；

2.使用接受到的物件即資料庫連線物件，建立遊標；

3.編寫SQL語句，並通過遊標執行SQL語句；

4.獲取SQL語句的查詢結果，並顯示出來；

5.關閉遊標；

6.當with語句內的程式碼執行完畢後自動執行__exit__方法關閉資料庫連線。

注意：

利用with結合自定義上下文類實現HTTP服務端：

# 1.匯入socket模組
import socket


class MySocket(object):
 # 2.編寫init方法接收port引數
 def __init__(self,port):
  self.__port = port
  # 3.建立socket物件
  self.__sk = socket.socket(socket.AF_INET,socket.SOCK_STREAM)

 # 4.編寫enter方法返回套接字物件
 def __enter__(self):
  # 設定埠複用
  self.__sk.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR,True)
  # 繫結埠
  self.__sk.bind(self.__port)
  # 設定埠監聽
  self.__sk.listen(128)
  # 返回套接字物件
  return self.__sk

 # 5.編寫exit方法關閉套接字物件
 def __exit__(self,exc_tb):
  self.__sk.close()


def main():
 # 使用with關鍵字 並接受返回的套接字物件給sk
 with MySocket(8000) as sk:
  # 等待客戶端連線
  clicent,ip_port = sk.accept()
  recv_data = clicent.recv(1024)
  print(recv_data.decode('utf-8'))
  # 編寫HTTP響應報文
  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)
  # 關閉客戶端連線
  clicent.close()

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

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

1.編寫__init__方法，用來接收引數並建立套接字物件；

2.編寫__enter__方法，並使用套接字物件設定埠複用、繫結埠、並設定監聽，然後返回套接字物件；

3.編寫__exit__方法，關閉套接字物件。

with語句程式碼解讀：

1.接收enter返回的套接字物件，並起個別名，

2.通過返回套接字物件等待客戶端連線，

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

4.使用客戶端套接字傳送HTTP響應報文

5.關閉客戶端連線

6.當with語句中的程式碼執行完畢後自動執行__exit__方法，關閉伺服器連線

總結

以上就是這篇文章的全部內容了，希望本文的內容對大家的學習或者工作具有一定的參考學習價值，謝謝大家對我們的支援。