zoukankan      html  css  js  c++  java
  • python django 自定义 装饰器

    # -*-coding:utf-8-*-
    __author__ = "GILANG (pleasurelong@foxmail.com)"
    """
    django 自定义用于view的装饰器
    """
    from functools import wraps
    
    def object_does_not_exist(func):
        """
        不带参数的装饰器
        """
        @wraps(func)
        def returned_wrapper(request, *args, **kwargs):
            try:
                return func(request, *args, **kwargs)
            except ObjectDoesNotExist:
                raise Http404()
        return returned_wrapper
    
    @object_does_not_exist
    def detail(request):
        """
        用法
        """
        pass
    
    ############################################################################
    
    
    
    def object_does_not_exist(redirect=None):
        """
        第一种写法:带参数的装饰器
        """
        def decorator(func):
            @wraps(func)
            def returned_wrapper(request, *args, **kwargs):
                try:
                    return func(request, *args, **kwargs)
                except ObjectDoesNotExist:
                    if redirect:
                        return HttpResponseRedirect(redirect)
                    else:
                        raise Http404()
            return returned_wrapper
        return decorator
    
    
    @object_does_not_exist(redirect='/')
    def detail(request):
        pass
    
    # 记得加个闭合括号,否则会出现类似 takes exactly 1 argument (0 given) 的错误
    @object_does_not_exist()
    def foo(request):
        pass
    
    
    ############################################################################
    
    def object_does_not_exist(func=None, redirect=None):
        """
        第二种写法:带参数的装饰器
        第二种方法可以解决 got an unexpected keyword argument 错误。
        """
        def decorator(func):
            @wraps(func)
            def returned_wrapper(request, *args, **kwargs):
                try:
                    return func(request, *args, **kwargs)
                except ObjectDoesNotExist:
                    if redirect:
                        return HttpResponseRedirect(redirect)
                    else:
                        raise Http404()
            return returned_wrapper
    
        if not func:
            def foo(func):
                return decorator(func)
            return foo
    
        else:
            return decorator(func)
        
        
    @object_does_not_exist(redirect='/')
    def detail(request):
        pass
    
    @object_does_not_exist
    def foo(request):
        pass
    

      

  • 相关阅读:
    Alpha冲刺
    Alpha冲刺
    Alpha冲刺
    Alpha冲刺
    抽奖系统(记一次未完成的教训)
    Alpha冲刺
    Alpha冲刺
    Alpha冲刺 (2/10)
    Alpha 冲刺 (1/10)
    软工 团队第三次作业
  • 原文地址:https://www.cnblogs.com/zhaojia-dream/p/4673779.html
Copyright © 2011-2022 走看看