Werkzeug之Local源码解析

Posted daviddd

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Werkzeug之Local源码解析相关的知识,希望对你有一定的参考价值。

Werkzeug之Local源码解析

原博客地址

http://liuyajing.coding.me/blogs/python/2018/werkzeug-local/

一、引入

最近在阅读 Flask 的源码,遇到三个概念:Local 、 LocalStack 和 LocalProxy ,本文主要就针对 Local 概念及其源码进行原理剖析。

二、Local

Local 是一个类,源码位置:site-packages/werkzeug/local.py

在模块的开头,有以下代码:

# 由于每个线程都有自己的greenlet,我们可以将它们用作上下文的标识符。 
# 如果greenlet不可用,我们将根据它的位置回退到当前的线程标识。
try:
    from greenlet import getcurrent as get_ident
except ImportError:
    try:
        from thread import get_ident
    except ImportError:
        from _thread import get_ident

1. 定义

class Local(object):
    # 定义此类允许绑定的属性名称 tuple 
    __slots__ = ('__storage__', '__ident_func__')

2. init

def __init__(self):
    object.__setattr__(self, '__storage__', {})
    object.__setattr__(self, '__ident_func__', get_ident)
'''
调用object 的__setattr__ 方法设置类Local的两个属性值, 即:
self.__storage__ = {}, self.__ident__func__ = get_ident
'''

3. iter

def __iter__(self):
    return iter(self.__storage__.items())
'''
__iter__方法返回一个迭代器,说明Local实例是一个可迭代对象
'''

4. call

def __call__(self, proxy):
    """Create a proxy for a name."""
    return LocalProxy(self, proxy)
'''
在类中实现了 __call__ 方法,那么实例对象也将成为一个可调用对象,那就可以像函数一样调用它。
'''

5. getattr

def __getattr__(self, name):
    try:
        return self.__storage__[self.__ident_func__()][name]
    except KeyError:
        raise AttributeError(name)
'''
如果 name 被访问,同时它不存在的时候,此方法被调用。
此方法实现了通过直接访问类属性的方式间接获取字典 self.__storage__[self.__ident_func__()]中key为name的值,而name并不属于类的属性,而此函数一定会被调用,很巧妙的        写法。
'''

6. setattr

def __setattr__(self, name, value):
    ident = self.__ident_func__()
    storage = self.__storage__
    try:
        storage[ident][name] = value
    except KeyError:
        storage[ident] = {name: value}
'''
如果要给name赋值,就调用这个方法。
此方法实现了通过直接给类属性赋值达到间接往字典self.__storage__[self.__ident_func__()]中插入{name: value}的目的。
'''

7. delattr

def __delattr__(self, name):
    try:
        del self.__storage__[self.__ident_func__()][name]
    except KeyError:
        raise AttributeError(name)

'''
如果要删除name,这个方法就被调用。
此方法实现了通过直接删除类属性达到间接从字典 self.__storage__[self.__ident_func__()]中删除key的目的。
'''

8. release_local

def __release_local__(self):
    self.__storage__.pop(self.__ident_func__(), None)
'''
此方法是为了删除字典self.__storage__中key为self.__ident_func__()及对应的value值。
'''

9. 总结

'''
- Local 类实际是对 dict __storage__ 的封装,而这个dict中 的 key 使用的就是get_indent 函数获取的 id (当有 greenlet时使用 greenlet id,没有则使用 thread id)
- dict __storage__ 中的 value 也是一个 dict,这个 dict 就是该 greenlet (或者thread) 对应的 local 存储空间
- 通过重新实现 __getattr__、__setattr__、__delattr__ 等方法,我们在 greenlet 或者 thread 中使用 Local 实例对象时,可以通过访问类属性的方式 (会自动获取greenlet
- id (或者 thread id)) ,访问到对应的 dict 存储空间中真正存储的对象。
- 这个技巧在实际编写线程安全或协程安全的代码时是非常有用的,即通过 greenlet id ( thread id ) 来分别存储数据。
- 当需要释放 greenlet ( 或 thread ) 对应的存储空间时,可以通过调用__release_local__() 函数来实现。
'''

以上是关于Werkzeug之Local源码解析的主要内容,如果未能解决你的问题,请参考以下文章

Flask -- 01. werkzeug请求与响应以及源码的解析

Flask之WSGI:Werkzeug

1.flask 源码解析:简介

Werkzeug库:local模块

flask 中的 werkzeug Local,LocalStack 和 LocalProxy 技术应用

werkzeug(flask)中的local,localstack,localproxy探究