Question

I thought I understood decorators but not anymore. Do decorators only work when the function is created?

I wanted to create a series of functions that all have a required argument called ‘ticket_params’ that is a dictionary. and then decorate them with something like @param_checker(['req_param_1', 'req_param_2']) and then if ‘req_param_1’ and ‘req_param_2’ aren’t in the dictionary, raise a custom Exception subclass. Am I thinking of this all wrong?

It would be something like this in the calling code:

@param_checker(['req_param_1', 'req_param_2'])
def my_decorated_function(params):
    # do stuff

params = {'req_param_1': 'Some Value'}
my_decorated_function(params)

# exception would be raised here from decorator.

Answer 1

A decorator is applied immediately after the def statement; the equivalence is:

@param_checker(['req_param_1', 'req_param_2'])
def my_decorated_function(params):
    # do stuff

is exactly the same thing as:

def my_decorated_function(params):
    # do stuff
my_decorated_function = param_checker(['req_param_1', 'req_param_2'])(my_decorated_function)

So the job of param_checker is to return a function that takes as its argument the function to be decorated and returns yet another function which does what you require. OK so far?

Edit: so, here’s one implementation…:

import functools

def param_checker(reqs):
  reqs = set(reqs)
  def middling(f):
    @functools.wraps(f)
    def wrapper(params):
      missing = reqs.difference(params)
      if missing:
        raise TypeError('Missing parms: %s' % ', '.join(sorted(missing)))
      return f(params)
    return wrapper
  return middling