Tự động đăng ký các phương thức lớp bằng cách sử dụng trang trí

Question

Tôi muốn có thể tạo trình trang trí python tự động "đăng ký" các phương thức lớp trong kho lưu trữ toàn cục (với một số thuộc tính).

Ví dụ mã:

class my_class(object): 

    @register(prop1,prop2) 
    def my_method(arg1,arg2): 
     # method code here... 

    @register(prop3,prop4) 
    def my_other_method(arg1,arg2): 
     # method code here... 

Tôi muốn rằng khi tải xong, ở đâu đó sẽ có một dict chứa:

{ "my_class.my_method"  : (prop1, prop2) 
    "my_class.my_other_method" : (prop3, prop4) } 

Đây có phải là có thể?

Answer 1

Không chỉ với một người trang trí, không. Nhưng một metaclass có thể tự động làm việc với một class sau khi nó được tạo ra. Nếu register trang trí của bạn chỉ làm cho ghi chú về những gì metaclass nên làm, bạn có thể làm như sau:

registry = {} 

class RegisteringType(type): 
    def __init__(cls, name, bases, attrs): 
     for key, val in attrs.iteritems(): 
      properties = getattr(val, 'register', None) 
      if properties is not None: 
       registry['%s.%s' % (name, key)] = properties 

def register(*args): 
    def decorator(f): 
     f.register = tuple(args) 
     return f 
    return decorator 

class MyClass(object): 
    __metaclass__ = RegisteringType 
    @register('prop1','prop2') 
    def my_method(arg1,arg2): 
     pass 

    @register('prop3','prop4') 
    def my_other_method(arg1,arg2): 
     pass 

print registry 

in

{'MyClass.my_other_method': ('prop3', 'prop4'), 'MyClass.my_method': ('prop1', 'prop2')} 

Answer 2

No. Người trang trí sẽ nhận được chức năng trước khi nó trở thành một phương pháp, vì vậy bạn không biết nó đang ở trên lớp nào.

Answer 3

Không dễ dàng, nhưng nếu bạn đang sử dụng Python 3 này nên làm việc:

registry = {} 
class MetaRegistry(type): 
    @classmethod 
    def __prepare__(mcl, name, bases): 
     def register(*props): 
      def deco(f): 
       registry[name + "." + f.__name__] = props 
       return f 
      return deco 
     d = dict() 
     d['register'] = register 
     return d 
    def __new__(mcl, name, bases, dct): 
     del dct['register'] 
     cls = super().__new__(mcl, name, bases, dct) 
     return cls 

class my_class(object, metaclass=MetaRegistry): 
    @register('prop1','prop2') 
    def my_method(arg1,arg2): 
     pass # method code here... 

    @register('prop3','prop4') 
    def my_other_method(arg1,arg2): 
     pass # method code here... 

print(registry) 

Lưu ý rằng bạn không thể có tên phương pháp tương đương với tên trang trí trong lớp meta-gõ, vì họ sẽ được tự động xóa bởi lệnh del trong phương thức __new__ của metaclass.

Đối với Python 2.6 Tôi nghĩ bạn sẽ phải nói rõ ràng cho người trang trí tên lớp để sử dụng.

Answer 4

Dưới đây là một tình yêu nhỏ cho trang trí lớp học. Tôi nghĩ cú pháp hơi đơn giản hơn so với yêu cầu đối với metaclasses.

def class_register(cls): 
    cls._propdict = {} 
    for methodname in dir(cls): 
     method = getattr(cls, methodname) 
     if hasattr(method, '_prop'): 
      cls._propdict.update(
       {cls.__name__ + '.' + methodname: method._prop}) 
    return cls 


def register(*args): 
    def wrapper(func): 
     func._prop = args 
     return func 
    return wrapper 


@class_register 
class MyClass(object): 

    @register('prop1', 'prop2') 
    def my_method(self, arg1, arg2): 
     pass 

    @register('prop3', 'prop4') 
    def my_other_method(self, arg1, arg2): 
     pass 

myclass = MyClass() 
print(myclass._propdict) 
# {'MyClass.my_other_method': ('prop3', 'prop4'), 'MyClass.my_method': ('prop1', 'prop2')} 

Answer 5

Không như đẹp hay thanh lịch, nhưng có lẽ cách đơn giản nhất nếu bạn chỉ cần điều này trong chỉ có một lớp:

_registry = {} 
class MyClass(object): 
    def register(*prop): 
     def decorator(meth): 
      _registry[MyClass.__name__ + '.' + meth.__name__] = prop 
     return decorator 

    @register('prop1', 'prop2') 
    def my_method(self, arg1, arg2): 
     pass 
    @register('prop3', 'prop4') 
    def my_other_method(self, arg1, arg2): 
     pass 

    del register 

Answer 6

Nếu bạn cần tên lớp, sử dụng giải pháp của Matt. Tuy nhiên, nếu bạn là ok với chỉ có tên các phương pháp - hoặc một tham chiếu đến phương pháp này - trong registry, đây có thể là một cách đơn giản hơn để làm việc đó:

class Registry: 
    r = {} 

    @classmethod 
    def register(cls, *args): 
     def decorator(fn): 
      cls.r[fn.__name__] = args 
      return fn 
     return decorator 

class MyClass(object): 

    @Registry.register("prop1","prop2") 
    def my_method(arg1,arg2): 
     pass 

    @Registry.register("prop3","prop4") 
    def my_other_method(arg1,arg2): 
     pass 

print Registry.r 

in

{'my_other_method': ('prop3', 'prop4'), 'my_method': ('prop1', 'prop2')}