使用可选参数装饰 python 方法

Question

我有一个 Python 类，其中包含许多带有签名的方法：

def select_xxx(self, arg1 , arg2 , .. argn, intersect = False)

即这些方法具有不同 (1-3) 数量的位置参数，并且可选参数与默认值 False 相交。我想用一个装饰器来装饰所有这些方法，该装饰器将检查 intersect 参数的值并相应地采取不同的操作。我目前的做法是这样的：

def select_decorator(select_method):

  def select_wrapper( self , *args, intersect = False , **kwargs)
     if intersect:
        # Special init code for intersect == True
        select_method( self , *args , **kwargs)
     else:
        # Normal call path for intersect == False 
        select_method( self , *args , **kwargs)

  return select_wrapper  

@select_decorator
select_xxx( self , arg1 , arg2 , intersect = False)

但是让可选参数与装饰器内部的 *args 和 **kwargs 组合相交目前并不令人愉快。如果这使问题更容易解决，我可以牺牲 **kwargs 功能。有什么建议吗？

乔金

Answer 1

我假设 intersect 将始终作为关键字参数传递。在这种情况下，您可以简单地在您的装饰器中执行此操作

def select_decorator(select_method):

  def select_wrapper( self , *args, **kwargs):
     intersect = kwargs.has_key('intersect') and kwargs['intersect']
     if intersect:
        # Special init code for intersect == True
        select_method( self , *args , **kwargs)
     else:
        # Normal call path for intersect == False 
        select_method( self , *args , **kwargs)
  return select_wrapper