데코레이터 함수에 현재 클래스 전달

Question

함수에 전달 된 인수 유형을 확인하는 데코레이터 메소드가 있습니다. 난 그냥 WHATTOPUTHERE의 장소에 ClassA을 넣을 수 있습니다 다른 클래스에서

class ClassA: 
    @accepts(WHATTOPUTHERE) 
    def doSomething(otherObject): 
     # Do something 

: 클래스에서 지금

def accepts(*types): 
    def check_accepts(f): 
     assert len(types) == f.func_code.co_argcount 
     def new_f(*args, **kwds): 
      for (a, t) in zip(args, types): 
       assert isinstance(a, t), \ 
         "arg %r does not match %s" % (a,t) 
      return f(*args, **kwds) 
     new_f.func_name = f.func_name 
     return new_f 
    return check_accepts 

은 (classA.py에), 나는 방법은 동일한 클래스의 인수를 허용하려면 그러나 classA.py 안에, ClassA는 알려져 있지 않습니다. 어떻게 현재 클래스를 @accepts() 함수에 전달할 수 있습니까?

Answer 1

데코레이터의 기능 기반 버전을 사용하고 클래스 정의 후 적용 :

class ClassA: 
    def doSomething(otherObject): 
     # Do something 
ClassA.doSomething = accepts(ClassA)(ClassA.doSomething) 

또 다른 방법은 자동으로 클래스를 생성 한 후이를 적용 할 것이라고 Metaclass 쓰기 될 것이다 :

class Meta(type): 

    def __new__(cls, clsname, bases, dct): 
     fields = ('doSomething',) #Fields on which you want to apply the decorator 
     for name, val in dct.items(): 
      if name in fields: 
       dct[name] = accepts(cls)(val) 
     return type.__new__(cls, clsname, bases, dct) 


class ClassA(object): 
    __metaclass__ = Meta 
    def doSomething(otherObject): 
     pass 

---

new_f.func_name = f.func_name과 같은 수동 작업 대신 functools.wraps을 사용하십시오. 이것은 문서 문자열, 인수 목록 등을 보존 할 것입니다.

from functools import wraps 
def accepts(*types): 
    def check_accepts(f): 
     print "inside" 
     assert len(types) == f.func_code.co_argcount 
     @wraps(f) 
     def new_f(*args, **kwds):