Python 3: преимущества и недостатки языка Python интерпретируемый язык программирования

def first_test():

print("Test function 1")

first_test_wrapped = simple_decore(first_test)

first_test = first_test_wrapped

Можно написать вот так:

1. 
   @simple_decore
   
2. 
   def first_test():
   
3. 
   print("Test function 1")
   

@simple_decore – это и есть декоратор функции.

Передача аргументов в функцию через декоратор

Если функция в своей работе требует наличие аргумента, то его можно передать через декоратор. Создадим декоратор, который принимает аргумент и выводит информацию о декорируемой функции и ее аргументе.

1. 
   def param_transfer(fn):
   
2. 
   def wrapper(arg):
   
3. 
   print("Run function: " + str(fn.__name__) + "(), with param: " + str(arg))
   
4. 
   fn(arg)
   
5. 
   return wrapper
   

Для демонстрации ее работы создадим функцию, которая выводит квадратный корень переданного ей числа, в качестве декоратора, укажем только что созданный param_transfer.

1. 
   @param_transfer
   
2. 
   def print_sqrt(num):
   
3. 
   print(num**0.5)
   

Выполним эту функцию с аргументом 4.

1. 
   >>> print_sqrt(4)
   
2. 
   Run function: print_sqrt(), with param: 4
   
3. 
   2.0
   

Декораторы для методов класса

Методы классов также можно объявлять с декоратором. Модифицируем декоратор param_transfer.

1. 
   def method_decor(fn):
   
2. 
   def wrapper(self):
   
3. 
   print("Run method: " + str(fn.__name__))
   
4. 
   fn(self)
   
5. 
   return wrapper
   

Создадим класс для представления двумерного вектора (из математики). В этом классе определим метод norm(), который выводит модуль вектора.

1.