我们上篇分析了上下文压入栈，出栈的大致流程，现在我们要问，flask是怎么实现的呢？准备来说，Local是怎么管理不同请求对象的，LocalProxy是怎么代理的。因为flask源码太复杂，今天我们自己一步一步实现类似于Local,LocalProxy的代码。通过自己实现，我们学会用代理模式，学会python的魔法方法setattr,getattr。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
| class A(object):
def spam(self, x):
print('A.spam')
def foo(self):
print('A.foo')
class AProxy(object):
def __init__(self):
self._a = A()
def spam(self,x):
print('B1.spam')
return self._a.spam(x)
def foo(self):
print('B1.foo')
return self._a.foo()
def bar(self):
print('B1.bar')
b1 = AProxy()
b1.spam(1)
b1.foo()
b1.bar()
|
这里的b1,是一个对象，我们访问b1.spam(1),它会返回类A的一个实例的方法。b1.foo（）也同理。
这个代理的问题是，我们写代理的时候，要先知道被代理的类A有哪些方法，下面我们实现一种更通用的代理。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
| class A(object):
def spam(self, x):
print('A.spam')
def foo(self):
print('A.foo')
class AProxy(object):
def __init__(self):
self._a = A()
self.name = ''
def __getattr__(self,name):
return getattr(self._a,key)
def bar(self):
print('B1.bar')
b1 = AProxy()
b1.spam(1)
b1.foo()
b1.bar()
|
上面这个代理，比第一个就抽象了，这晨用getattr实现了什么呢？只要你访问AProxy没有的方法，它就会去getattr中找，这个函数实现了你给一个函数名，它会给你被代理对象的函数。这里是一个抽象。
上面这个函数，只是实现了属性，方法的访问，却不能给它赋值。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
| class A(object):
def __init__(self):
self.log = ''
def spam(self, x):
print('A.spam')
def foo(self):
print('A.foo')
class AProxy(object):
def __init__(self):
self._a = A()
self.name = ''
def __getattr__(self,name):
return getattr(self._a,key)
def __setattr__(self,name,value):
return setattr(self._a,name,value)
def bar(self):
print('B1.bar')
b1 = AProxy()
b1.name = '我是AProxy的一个实例'
b1.log = '我是A的一个实例'
|
这个代理就比前一个更好用了，因为它可以给A的实例间接赋值。我们调用b1.log,这个时候AProxy没有这个属性，这时就去setattr中找，把log,’我是A的一个实例’作为实参，传给setattr。
当然这个代理也有些问题，就是它还不够抽象，它现在只能代理类A,如果我们希望它还能代理其它类怎么办呢？
改进如下：
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
| class A(object):
def __init__(self):
self.log = ''
def spam(self, x):
print('A.spam')
def foo(self):
print('A.foo')
class AProxy(object):
def __init__(self，object):
self._a = object
self.name = ''
def __getattr__(self,name):
return getattr(self._a,key)
def __setattr__(self,name,value):
return setattr(self._a,name,value)
def bar(self):
print('B1.bar')
a1 = A
b1 = AProxy(a1)
b1.name = '我是AProxy的一个实例'
b1.log = '我是A的一个实例'
|
这里通过一个给AProxy传参初始化，可以拿到不同的实例代理。
有了上面这些基础，现在我们模仿Local,LocalProxy来写一个代理：
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
| class Request(object):
def __init__(self):
self.url = 'tanliang.com'
class User(object):
def __init__(self):
self.owner = 'tanliang'
class Local:
def __init__(self):
self._objs = {}
self._objs['request'] = Request()
self._objs['user'] = User()
def __getattr__(self, name):
print('Local.__getattr__')
return self._objs[name]
def __call__(self, name):
return LocalProxy(self, name)
class LocalProxy(object):
def __init__(self, local, name):
self._local = local
self._name = name
def _get_current_obj(self):
print('_get_current_obj')
return getattr(self._local, self._name)
def __getattr__(self, name):
print('SpamProxy.__getattr__')
return getattr(self._get_current_obj(), name)
def __setattr__(self, name ,value):
if name.startswith('_'):
super().__setattr__(name, value)
else:
setattr(self._obj, name, value)
def __delattr__(self, name):
print('SpamProxy.__delattr__')
if name.startswith('_'):
super().__delattr__(name)
else:
print('SpamProxy', name)
delattr(self._obj, name)
l = Local()
request = l('request')
print(request)
print(request.url)
user = l('user')
print(user.owner)
|
这里l是一个Local实例，request是一个代理对象，LocalProxy的实例，因为Local实现了call方法。当我们访问request.url时，会发生什么呢？先找到getattr,然后把url作为参数传给getattr。可是这个时候还没有拿到“当前对象”。self.get_current_obj()就是拿当前对象的，最后能过Local的getattr__拿到request对象。