码迷,mamicode.com
首页 > 编程语言 > 详细

插件和过滤器装饰器开发中的感悟-python-django

时间:2014-10-02 18:33:53      阅读:300      评论:0      收藏:0      [点我收藏+]

标签:style   blog   color   io   os   使用   ar   for   strong   

    写这篇随笔是因为今天自己在写插件和过滤方法的过程中碰壁了,折腾了好久终于稍微发现些问题,在此记下,以作备忘。

在看了xadmin的插件机制后,笔者也想使用该思想来扩展kadmin中视图的方法。

例如,在一个登陆视图中,一般的逻辑如下:

class LoginView(BaseAdminView):
    ‘‘‘登陆视图‘‘‘
    auth_form=None#用于认证的表单类
    login_template=None
    title=""

    def update_login_params(self,defaults):
        ‘‘‘ 用于在执行login视图函数前修改参数的钩子‘‘‘
        return defaults
    @never_cache
    def get(self,request,*arg,**kwargs):
        from django.contrib.auth.views import login
        context=self.get_context(request,*arg,**kwargs)#获取父类的context
        context.update({
            title:self.title,
            app_path:request.get_full_path,
            REDIRECT_FIELD_NAME:request.get_full_path(),
        })
        defaults={
            extra_context:context,
            authentication_form:self.auth_form or AdminAuthenticationForm,
            template_name:self.login_template or kadmin/views/login.html,
        }
       defaults=self.update_login_params(defaults) 

return login(request,**defaults) @never_cache def post(self,request,*arg,**kwargs): return self.get(request,*arg,**kwargs)

我们希望能通过update_login_params方法,在执行login函数前更新context,就应该使用插件来接管该方法的执行以按需要进行一些修改。

这里有两种思路:

1、使用过滤器,过滤update_login_params的结果并返回

2、使用插件函数,控制update_login_params的执行(一般是活得update_login_params的执行结果 传入插件函数 由插件函数进行一些修改后再返回)

    也可以用插件函数接管update_login_params,并自行决定是否执行update_login_params,或是在update_login_params执行前或后进行一些行为

从上述两点看,插件更为灵活和自由,控制范围更大,二过滤器只能就update_login_params的返回值进行操作并返回。

1、过滤器

定义过滤器装饰器@filter_hook

def filter_hook(func):
    #过滤器装饰器
    #用于对被hook的方法返回值进行再处理
    func_name=func.__name__
    @wraps(func)
    def inner_func(self,*arg,**kwargs):
        if getattr(self,filters,None):
            filter_infos=[(getattr(fi,priority,10),fi) for fi in self.filters if fi.__name__==func_name]
            filters=[fi for p,fi in sorted(filter_infos,key=lambda x:x[0],reverse=False)]
            result=func(self,*arg,**kwargs)
            for fi in filters:
                result=fi(self,result,*arg,**kwargs)
            return result
        else:
            return func(self,*arg,**kwargs)
    return inner_func

源码说明:在视图对象self上遍历filters 得到过滤函数,过滤结果并返回结果

例如我们可以定义一个过滤器给say_hello的返回值加上括号

@set_attr(attr_name=__name__,value=say_hello)
def add_tag(self,result,o,*arg,**kwargs):
    return "(%s)"%result
class Person:
    name=akun
    sex="akun-male"
    plugins=[MalePlugin(sex_num) for sex_num in range(4)]
    filters=[add_tag]

    @filter_hook
    @plugin_hook
    def say_hello(self,go,*arg,**kwargs):
        print("-----------done--------")
        return "Hello i am %s"%self.name
if __name__=="__main__":
    p=Person()
    print(p.say_hello(go))

说明:@set_attr是一个装饰器,负责设置函数的属性值,因为filter_hook在查找过滤函数时是找被hook的同名函数,所以需要把add_tag的__name__设为say_hello

2.插件

定义插件装饰器:

def plugin_chain(funcs,token,func,*arg,**kwargs):
    if token==-1:
        return func()
    @wraps(func)
    def _inner_func():
        wrap_func=funcs[token]
        arg_specs=getargspec(wrap_func)[0]
        if len(arg_specs)==1:
            func()
            return wrap_func(*arg,**kwargs)
        elif len(arg_specs)>=2 and arg_specs[1]=="__":
            back=func
        else:
            back=func()
        return wrap_func(back,*arg,**kwargs)
    return plugin_chain(funcs,token-1,_inner_func,*arg,**kwargs)
            
def plugin_hook(func):
    func_name=func.__name__
    @wraps(func)
    def method(self,*arg,**kwargs):
        @wraps(func)
        def _inner_func():
            return func(self,*arg,**kwargs)
        if getattr(self,plugins,None):
            plugin_funcs=[(getattr(p,func_name),getattr(getattr(p,func_name),priority,10))
                      for p in self.plugins if getattr(p,func_name,None) ]
            print(plugin_funcs)
            #对插件方法按照priority升序排列a
            plugin_funcs=[p for p,priority in sorted(plugin_funcs,key=lambda x:x[1],reverse=False)]
            return plugin_chain(plugin_funcs,len(plugin_funcs)-1,_inner_func,*arg,**kwargs)
        else:
            return func(self,*arg,**kwargs)
    return method

源码说明:类似于@filter_hook只是,注意

        wrap_func=funcs[token]
        arg_specs=getargspec(wrap_func)[0]
        if len(arg_specs)==1:
            func()
            return wrap_func(*arg,**kwargs)
        elif len(arg_specs)>=2 and arg_specs[1]=="__":
            back=func
        else:
            back=func()
        return wrap_func(back,*arg,**kwargs)

如果若用于接收被hook方法的返回值的参数名为"__"则,__会被设置为被hook的方法,也就是可以通过插件来接管被hook的方法。

我们定义一个插件来在执行say_hello前打印出被hook的方法

class BeforeHook:
    @set_attr(attr_name=priority,value=6)
    def say_hello(self,__,o,*arg,**kwargs):
        print("the back is %s"%__)
        return __(*arg,**kwargs)

这里的__或被设置为被hook的方法

 

插件和过滤器装饰器开发中的感悟-python-django

标签:style   blog   color   io   os   使用   ar   for   strong   

原文地址:http://www.cnblogs.com/zkchen/p/4004248.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!