python 装饰器应用

1 注册回调函数

下面这个示例展示了通过URL的路由来调用相关注册的函数示例：

class MyApp():

def __init__(self):

self.func_map = {}

def register(self, name):

def func_wrapper(func):

self.func_map[name] = func

return func

return func_wrapper

def call_method(self, name=None):

func = self.func_map.get(name, None)

if func is None:

raise Exception("No function registered against - " + str(name))

return func()

app = MyApp()

@app.register('/')

def main_page_func():

return "This is the main page."

@app.register('/next_page')

def next_page_func():

return "This is the next page."

print app.call_method('/')

print app.call_method('/next_page')

注意：
1）上面这个示例中，用类的实例来做decorator。
2）decorator类中没有__call__()，但是wrapper返回了原函数。所以，原函数没有发生任何变化。

2 打印日志

from functools import wraps

import inspect

def get_line_number():

return inspect.currentframe().f_back.f_back.f_lineno

def logger(loglevel):

def log_decorator(fn):

@wraps(fn)

def wrapper(*args, **kwargs):

ts = time.time()

result = fn(*args, **kwargs)

te = time.time()

print "function   = " + fn.__name__,

print "    arguments = {0} {1}".format(args, kwargs)

#print "    return    = {0}".format(result)

print "    time      = %.6f sec" % (te-ts)

if (loglevel == 'debug'):

print "    called_from_line : " + str(get_line_number())

return result

return wrapper

return log_decoratorspam(1,2,3)

3 调用追踪

import sys,os,linecache

def trace(f):

def globaltrace(frame, why, arg):

if why == "call": return localtrace

return None

def localtrace(frame=1, why=2, arg=4):

if why == "line":

# record the file name and line number of every trace

filename = frame.f_code.co_filename

lineno = frame.f_lineno

bname = os.path.basename(filename)

print "{}({}): {}".format(  bname,

lineno,

linecache.getline(filename, lineno)),

return localtrace

def _f(*args, **kwds):

sys.settrace(globaltrace)

result = f(*args, **kwds)

sys.settrace(None)

return result

return _f

@trace

def xxx():

a=1

print a

xxx() #调用

4 单例模式

# coding=utf-8

# 测试utf-8编码

# 单例装饰器

import sys

reload(sys)

sys.setdefaultencoding('utf-8')

# 使用装饰器实现简单的单例模式

def singleton(cls):

    instances = dict()  # 初始为空

    def _singleton(*args, **kwargs):

        if cls not in instances:  #如果不存在, 则创建并放入字典

            instances[cls] = cls(*args, **kwargs)

        return instances[cls]

    return _singleton

@singleton

class Test(object):

    pass

if __name__ == '__main__':

    t1 = Test()

    t2 = Test()

    # 两者具有相同的地址

    print t1

    print t2