7-3 如何让对象支持上下文管理协议，需要定义__enter__, __exit__方法 分别在with开始和结束时候被调用

7-3 如何让对象支持上下文管理
解决方案:
实现上下文管理协议，需要定义enter, exit方法
他们分别在with开始和结束时候被调用

with TelnetClient2('127.0.0.1') as client:
    client.start()      

对象client是来自于__enter__函数的返回
最后退出的时候不管是否存在异常，都将会进入到__exit__处理

with open('test.txt','w') as fw:
    fw.write('hello,world')
    fw.writelines(['xyz\n',123\n'])
fw.close()

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):

        #user
        t = self.tn.read_unt('login: ')
        stdout.write(t)
        user = stdin.readline()
        self.tn.write(user)

        # password
        t = self.tn.read_until("Password: ")
        if t.startswith(user[:-1]): t =t[len(user) + 1:]
        stdout.write(t)
        self.tn.write(stdin.readline())
        t = self.tn.read_until('> ')
        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):
        self.tn.close()
        self.tn = None
        with open(self.addr + '_history.txt', 'w') as f:
            f.writeline(self.history)


class TelnetClient2(object):
    def __init__(self,addr,port=23):
        self.addr = addr
        self.port = port
        self.tn = None

    def start(self):      
        #user
        t = self.tn.read_until('login: ')
        stdout.write(t)
        user = stdin.readline()
        self.tn.write(user)

        # password
        t = self.tn.read_until("Password: ")
        if t.startswith(user[:-1]): t =t[len(user) + 1:]
        stdout.write(t)
        self.tn.write(stdin.readline())
        t = self.tn.read_until('> ')
        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 __enter__(self):
        self.tn = Telnet(self.addr,self.port)
        self.history = deque() 
        return self

    def __exit__(self,exc_type,exc_val,exc_tb):
        self.tn.close()
        self.tn = None
        with open(self.addr + '_history.txt', 'w') as f:
            f.writelines(self.history)

with TelnetClient2('127.0.0.1') as client:
    client.start()