深入类与对象-上下文管理器

mixin继承

多继承会造成关系混乱，导致MRO算法出现许多预料不到的问题，一般都是推荐MIXIN继承，最小继承，只继承一个类。

上下文管理器

with上下文管理器，这个用过很多次，比如在执行mysql语句的时候要先链接数据库，获取游标，执行sql语句，关闭连接。又或者是文本内容的读写，打开文本写入内容关闭文本。

如果每次执行一条语句都要做这么多操作，就会产生大量重复的代码，这个时候使用上下文管理器即可美观又轻松的解决这个问题。

with管理上下文的作用是对一些重复的代码简单化，并且能优化try/except/finally的写法。

上下文的实现是通过两个魔法函数enter和exit实现，后来更新再造后，使用contextlib提供的API可以更加方便的完成。

概念和功能都明白后，最重要的还是如何实现。

举个例子：shuke去给猫买饼干，每次买的时候都要做这些动作，掏出钱包，花出x元，收回钱包。每次都要重复掏钱包收钱包这个动作有些麻烦，并且万一忘了掏钱包就不能付钱，忘了收钱包的话，钱包就掉了。每次掏钱包收钱包都要写代码，有啥简介的方式嘛？

使用enter和exit实现(1)

#! /usr/bin/env python
# -*- coding: utf-8 -*-
"""
@author: shuke
@file: 0325.py 
@time: 2019/03/25 16:28
@contact: zhaofengfeng@shuke.net
@software:  learn
"""


class wallet(object):
    def __init__(self, man):
        self.man = man

    def __enter__(self):
        print(self.man + '放心大胆的掏出了钱包')

    def __exit__(self, exc_type, exc_val, exc_tb):
        print(self.man + '小心翼翼的收起来钱包')


def use_money(man):
    return wallet(man)


with use_money('shuke') as a:
    print('花了600块钱')

返回结果：

1
2
3

shuke放心大胆的掏出了钱包
花了600块钱
shuke小心翼翼的收起来钱包

使用enter和exit实现(2)

class wallet(object):
    def __init__(self, man):
        self.man = man

    def __enter__(self):
        print(self.man + '放心大胆的掏出了钱包')
        # return self 这一步非常重要，作用是返回实例
        return self

    def __exit__(self, exc_type, exc_val, exc_tb):
        print(self.man + '小心翼翼的收起来钱包')

    def use_money(self, money):
        print(self.man + '花了' + money + '元')


with wallet('shuke02')as a:
    a.use_money('1000')

运行结果:

1
2
3

shuke02放心大胆的掏出了钱包
shuke02花了1000元
shuke02小心翼翼的收起来钱包

使用contextlib实现

import contextlib


@contextlib.contextmanager
def use_money(man):
    try:
        print(man + '十分放心大胆的掏出钱包')
        yield None
        # yield生成器，运行到这里会返回一个值(你随便写一个就行)
    finally:
        print(man + '万分谨慎的收起了钱包')


with use_money('shuke03')as a:
    print('花了100块钱')

返回结果：

1
2
3

shuke03十分放心大胆的掏出钱包
花了100块钱
shuke03万分谨慎的收起了钱包

这只是一个创建上下文管理器的方法，按照固定的格式书写就行.

refs

深入类与对象-下