实际案例：
我们实现了一个telnet客户端TelnetClient，调用实例的start()方法启动客户端与服务器交互，交互完毕后需调用cleanup()方法，关闭已连接的socket，以及将操作历史记录写入文件并关闭。
能否让TelnetClient的实例支持上下文管理协议，从而替代手工调用cleanup()方法。
解决方案：
实现上下文管理协议，需定义实例的__enter__，__exit__方法，它们分别在with开始和结束时被调用。
2、代码演示（1）上下文管理的例子
'''
经常在对文件操作的时候使用上下文管理使用with语句，
使用上下文管理的好处就是在with语句结束以后不用显示的close去关闭文件，
这样一个清理工作会交给上下文管理协议，让它们自动帮我们完成。
'''
with open('demo.txt', 'w') as f:
f.write('abcdef')
f.writelines(['xyzn', '123n'])
# f.close()
（2）实现一个类的对象支持上下文管理协议
from telnetlib import Telnet
from sys import stdin, stdout
from collections import deque
class TelnetClient(object):
def __init__(self, addr, port=23):
self.addr = addr
self.port = port
self.tn = None
def start(self):
# 人为显示抛出一个异常,用来证明即使有异常__exit__同样会被调用
# raise Exception('Test')
# 进行登录操作输入用户名和密码
# user
t = self.tn.read_until(b'login: ')
stdout.write(t)
user = stdin.readline()
self.tn.write(user)
# password
t = self.tn.read_until(b'password: ')
if t.startswith(user[:-1]):
t = t[len(user) + 1:]
stdout.write(t)
self.tn.write(stdin.readline())
# 登录到服务器的shell当中
t = self.tn.read_until('$ ')
stdout.write(t)
while True:
uinput = stdin.readline()
if not uinput:
break
self.history.append(uinput)
self.tn.write(uinput)
t = self.tn.read_until('$ ')
stdout.write(t[len(uinput) + 1:])
def cleanup(self):
pass
def __enter__(self):
# Telnet连接对象
self.tn = Telnet(self.addr, self.port)
# 创建队列存储用户的操作历史记录
self.history = deque()
# __enter__()返回的才是with中as的对象，self自身
return self
# 3个参数分别为异常类型、异常值、跟踪栈信息
# 无异常的情况下三个参数都是None
def __exit__(self, exc_type, exc_val, exc_tb):
# print('In __exit__')
# 关闭服务器连接的socket
self.tn.close()
self.tn = None
with open(self.addr + 'history.txt', 'w') as f:
f.writelines(self.history)
# 默认返回None
# return True # 不抛出异常
with TelnetClient('127.0.0.1') as client:
client.start()
print('END')
