魔法代码

2020-02-15

with语法

会话管理，在Python 2.5中，为了代码利用定义了一个新的关键词 with 语句。会话控制在Python中不罕见(之前是作为库的一部分被实现)，直到 PEP343 被添加后。它被成为一级语言结构

原理

会话控制器通过包装一个 with 语句来设置和清理行为。回话控制器的行为通过两个魔术方法来定义: __enter__和__exit__
- __enter__(self)
  
  定义当使用 with 语句的时候会话管理器应该初始块被创建的时候的行为。注意 enter 的返回值被 with 语句的目标或者 as 后的名字绑定
- __exit__(self, exception_type, exception_value, traceback)
  
  定义当一个代码块被执行或者终止后会话管理器应该做什么。它可以被用来处理异常，清除工作或者做一些代码块执行完毕之后的日常工作。如果代码块执行成功， exception_type , exception_value , 和 traceback 将会是 None 。否则的话你可以选择处理这个异常或者是直接交给用户处理。如果你想处理这个异常的话，确认 exit 在所有结束之后会返回 True 。如果你想让异常被会话管理器处理的话，那么就这样处理,则可以像示例那样处理

使用示例

class Closer:
    '''通过with语句和一个close方法来关闭一个对象的会话管理器'''

    def __init__(self, obj):
        self.obj = obj

    def __enter__(self):
        return self.obj # bound to target

    def __exit__(self, exception_type, exception_val, trace):
        try:
            self.obj.close()
        except AttributeError: # obj isn't closable
            print('Not closable.')
            return True # exception handled successfully

if __name__ == '__main__':
    # file = open('queue_test.py')
    # wt = Closer(file)
    # print(wt.obj.read())
    # wt.obj.close()

    # 该用法和上面的代码用法等同，但是看起来更简洁with提高了代码精炼度
    with Closer(open('queue_test.py')) as ob:
        s = ob.read()
        print(s)

getitem

只要是在类中定义了这个方法，就可以用类对象通过[]运算符进行访问，例如数组就可以通过这种方式直接访问下标对应的元素。

class MyObject:
    def __getitem__(self, item):
        return '😘❤️'

# 当用[]取值的时候会触发__getitem__函数调用    
print(MyObject()['name'])

setitem

在对[]运算操作赋值时，会触发这个方法调用，比如我们直接对数组下标赋值，修改元素

class MyObject:
    def __getitem__(self, item):
        return '😘❤️'
    def __setitem__(self, key, value):
        print(key,value)

obj = MyObject()
print(obj['name'])
# 当给[]赋值时，会触发__setitem__调用
obj[89] = 'qwer'

len

一般用于需要计算长度的类会去重写这个方法

class MyObject:
    def __len__(self):
        return 45

obj = MyObject()
print(len(obj))