from IPython import get_ipython


def decorate_cell(line, cell):
    print("before")
    get_ipython().run_cell(cell)
    print("after")

get_ipython().register_magic_function(decorate_cell, "cell")