4.2. Decorator Types

  • Decorator function

  • Decorator method

  • Decorator class

4.2.1. Decorator Function

  • Decorator is a function which takes another object as an argument

  • Doesn't matter, whether this object is a function, class or method

Definition:

>>> def mydecorator(obj):
...     ...

Usage:

>>> @mydecorator
... def say_hello():
...     return 'hello'
>>> @mydecorator
... class Astronaut:
...     def say_hello():
...         return 'hello'
>>> class Astronaut:
...     @mydecorator
...     def say_hello():
...         return 'hello'

4.2.2. Decorator Method

  • Decorator is a method which takes instance and another object as an argument

  • Doesn't matter, whether this object is a function, class or method

Definition:

>>> class MyClass:
...     @staticmethod
...     def mydecorator(obj):
...         ...

Usage:

>>> @MyClass.mydecorator
... def say_hello():
...     return 'hello'
>>> @MyClass.mydecorator
... class Astronaut:
...     def say_hello():
...         return 'hello'
>>> class Astronaut:
...     @MyClass.mydecorator
...     def say_hello():
...         return 'hello'

4.2.3. Decorator Class

  • Decorator is a class which takes another object as an argument to __init__() method

  • Doesn't matter, whether this object is a function, class or method

Definition:

>>> class MyDecorator:
...     def __init__(self, obj):
...         ...

Usage:

>>> @MyDecorator
... def say_hello():
...     return 'hello'
>>> @MyDecorator
... class Astronaut:
...     def say_hello():
...         return 'hello'
>>> class Astronaut:
...     @MyDecorator
...     def say_hello():
...         return 'hello'