Python - могу ли я использовать декоратор внутри функции?

Question

У меня есть 2 вопроса. Я создал декоратор, который проверяет, имеет ли словарь ключ или нет? Вот.

def check_has_key(func): 
    def inner(x,y): # inner function needs parameters 
     dictionary = {"add" : "true", "subtract" : "true"} 
     if dictionary.has_key("add") : 
      return func(x,y) 
     return "Add not allowed" 
    return inner # return the inner function (don't call it) 

@check_has_key 
def add(x,y): 
    return x+y 

print add(1,2) 

1) Могу ли я передать ключ в качестве аргумента в обертке, а затем проверить, существует ли он или нет? Например: например, я просто передаю ключ как @check_has_ket("subtact").

2) Могу ли я использовать декоратор внутри функции? как в случае, если мне нужно проверить, есть ли у словаря ключ или нет, вглубь функции?

EDIT

Я получил ответ на 1-й вопрос.

def abc(a): 
    def check_has_key(func): 
     def inner(x,y): # inner function needs parameters 
      dictionary = {"add" : "true", "subtract" : "true"} 
      if dictionary.has_key(a) : 
       return func(x,y) 
      return "Add not allowed" 
     return inner # return the inner function (don't call it) 
    return check_has_key 

@abc("subtract") 
def add(x,y): 
    return x+y 

print add(1,2) 

Но мои сомнения по-прежнему остаются я могу использовать декоратор в глубине функции? Значение, если мне нужно проверить, существует ли ключ в словаре или не углубляться в эту функцию, могу ли я использовать декоратор для этой цели или мне нужно будет использовать только условие if?

Answer 1

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

class decorate: 

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

    def __call__(self, func): 
     def inner(x,y): 
      # do something, probably using self.decorator_arg 
      return func(x,y) 
     return inner 

@decorate("subtract") 
def add(x,y): 
    return x+y 

Для (2), да, если у вас есть другая функция для украшения внутри вашей функции. Если вам нужно сделать такую ​​вещь, вам, вероятно, просто нужна функция вместо декоратора.