06 #### `__enter__、__exit__` 用来对上下文管理，可以使用这两个方法

# __enter__、__exit__ 用来对上下文管理
class content:

    def __init__(self, filepath, mode):
        self.filepath = filepath
        self.mode = mode
        self.encoding = 'utf-8'
        self.fileobject = None

    def __enter__(self):
        # 可以用来：打开文件、链接、数据操作
        self.fileobject = open(self.filepath, self.mode, encoding=self.encoding)  # 得到一个文件对象,然后进行返回
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        # 关闭 or 结束
        self.fileobject.close()  # 操作完成后，进行关闭文件

    def send(self):
        self.fileobject.write('发送\n')  # 通过获取的文件对象进行写操作

    def read(self):
        self.fileobject.write('发送222\n')


# 写法1
obj = content('test', 22)

# 如果需要支持 with,需要在类方法中，增加__enter、__exit__方法，否则报错

with obj as xxx:  # with的是一个对象，只要with这个对象时，自动执行__enter 方法,

    print(xxx)  # 如果enter中return 123，此时as xxx 中的：xxx  就等于123

# 写法2
# 也支持下面的这种写法：
with content('test.txt', 'a') as obj:
    print(obj)  # 当with里面的代码执行完后，自动执行 __exit__ 方法
    # 此时执行的 obj.send/obj.read 都 在 __enter__ 之后
    obj.send()
    obj.read()