python父类方法的装饰器_Python-在类声明中访问父类装饰器

Let's say I have this class:

class Foo:

@classmethod

def some_decorator(cls, ...):

...

And then I create a subclass which uses the parent class decorator:

class Bar(Foo):

@Foo.some_decorator(...)

def some_function(...)

...

How do I remove the need for Foo. before the decorator name? The below code doesn't work:

class Bar(Foo):

@some_decorator(...)

def some_function(...)

...

I believe this is possible, because the sly library does this.

See their example:

from sly import Lexer, Parser

class CalcLexer(Lexer):

...

@_(r'\d+')

def NUMBER(self, t):

t.value = int(t.value)

return t

...

As you can see, you can type in @_(...) instead of @Lexer._(...).

How do they accomplish this?

解决方案

This is done with a metaclass that implements a __prepare__ method. Excerpt from the docs:

3.3.3.4. Preparing the class namespace

Once the appropriate metaclass has been identified, then the class

namespace is prepared. If the metaclass has a __prepare__ attribute,

it is called as namespace = metaclass.__prepare__(name, bases, **kwds)

(where the additional keyword arguments, if any, come from the class

definition).

To put it in simple terms: You make your __prepare__ method return a dictionary that contains an entry for the decorator. Proof of concept:

class MyMeta(type):

def __prepare__(name, bases):

return {'x': 'foobar'}

class MyClass(metaclass=MyMeta):

print(x) # output: foobar