forked from faif/python-patterns
-
Notifications
You must be signed in to change notification settings - Fork 0
/
decorator.py
52 lines (36 loc) · 997 Bytes
/
decorator.py
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
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""http://stackoverflow.com/questions/3118929/implementing-the-decorator-pattern-in-python"""
class foo_decorator(object):
def __init__(self, decoratee):
self._decoratee = decoratee
def f1(self):
print("decorated f1")
self._decoratee.f1()
def __getattr__(self, name):
return getattr(self._decoratee, name)
class undecorated_foo(object):
def f1(self):
print("original f1")
def f2(self):
print("original f2")
@foo_decorator
class decorated_foo(object):
def f1(self):
print("original f1")
def f2(self):
print("original f2")
def main():
u = undecorated_foo()
v = foo_decorator(u)
# The @foo_decorator syntax is just shorthand for calling
# foo_decorator on the decorated object right after its
# declaration.
v.f1()
v.f2()
if __name__ == '__main__':
main()
### OUTPUT ###
# decorated f1
# original f1
# original f2