데코레이터로 함수 비활성화

Question

파이썬 장식자를 사용하여 함수를 "비활성화"할 수 있습니까? 여기에 예 : 도서관에서 옵션 (하기 matplotlib 같은) 일부 종속성이 있으며, 이들은 다음과 같습니다

cond = False 

class C: 

    if cond: 
     def x(self): print "hi" 

    def y(self): print "ho" 

는이

class C: 

    @cond 
    def x(self): print "hi" 

    def y(self): print "ho" 

배경? :처럼 장식이 코드를 다시 작성하는 것이 가능 몇 가지 기능 만 필요합니다 (디버그 또는 정면). 이것은 어떤 시스템에서는 matplotlib가 다른 시스템에 설치되지 않았 음을 의미하지만 둘 다 (코어) 코드를 실행해야합니다. 그러므로 matplotlib가 설치되지 않은 경우 일부 기능을 비활성화하고 싶습니다. 그런 우아한 방법이 있습니까?

Answer 1

당신은 어떤 작전으로 기능을 설정 할 수있는 장식으로 (즉, 경고 로그) :

def conditional(cond, warning=None): 
    def noop_decorator(func): 
     return func # pass through 

    def neutered_function(func): 
     def neutered(*args, **kw): 
      if warning: 
       log.warn(warning) 
      return 
     return neutered 

    return noop_decorator if cond else neutered_function 

여기

conditional는 장식 공장입니다. 조건에 따라 두 개의 데코레이터 중 하나를 반환합니다.

하나의 데코레이터는 단순히 기능을 그대로 둡니다. 다른 데코레이터는 데코 레이팅 된 기능을 대체하여 경고를 대신 표시합니다.

사용 :

이

@conditional('matplotlib' in sys.modules, 'Please install matplotlib') 
def foo(self, bar): 
    pass 

Answer 2

Martijns이 noops에 기능을 tunring와 거래 대답, 당신이 실제로 클래스에서 제거하는 방법을 설명하겠습니다 - 아마 잔인한 사람입니다, 제가 정착 것 어떤 종류의 예외를 throw하는 Martijns 응답의 변형입니다. 하지만 어쨌든 :

클래스 데코레이터를 사용하여 영향을받은 함수를 클래스에서 제거 할 수 있습니다. 이 사람은 제거 부울 및 속성의 목록을 취

def rm_attrs_if(cond, attrs): 

    if not cond: 
     return lambda c: C#if the condition is false, don't modify the class 

    def rm_attrs(cls): 
     d = dict(cls.__dict__) #copy class dict 
     for attr in attrs: 
      del d[attr]  #remove all listed attributes 
     return type(cls.__name__, cls.__bases__, d) #create and return new class 

    return rm_attrs 

이처럼 사용

@rm_attrs_if(something == False, ["f1", "f2"]) 
class X(): 
    def f1(): pass 
    def f2(): pass 
    def f3(): pass