Как создать декоратор в классе на Python?

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

from functools import wraps # Импортируем из functools модуль для работы с декораторами

class Replicator:
    '''
    Данный декоратор принимает в аргументы переменную int - количество повторов функции
    '''

    def __init__(self, count):
        self.count = count

    def __call__(self, f): # Создаем функцию, которая будет срабатывать при вызове класса
        @wraps(f)

        def wrapper(*args, **kwargs):
            for _ in range(self.count):
                f(*args, **kwargs)

        return wrapper

@Replicator(5) # Над функцией обозначаем декоратор, в аргументах которого стоит число - это то, сколько раз выполнится функция
# Т.К это декоратор, то аргумент f(саму функцию) он получит автоматически
def GET(url):
    print(url)

if __name__ == "__main__":
    GET("dropcode.ru")

# Вывод :
# dropcode.ru
# dropcode.ru
# dropcode.ru
# dropcode.ru
# dropcode.ru

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12

def my_decorator(cls):
    class NewClass:
        def __init__(self, *args, **kwargs):
            self.instance = cls(*args, **kwargs)

        def __getattr__(self, name):
            return getattr(self.instance, name)

        def new_method(self):
            print("This is a new method.")

    return NewClass

1
2
3
4

@my_decorator
class MyClass:
    def my_method(self):
        print("This is my method.")

1
2

my_object = MyClass()
my_object.new_method()  # "This is a new method."