Question

I was hoping this would work:

class A:
    def __enter__(self, code):
       print code

    def __exit__(..):
       pass

and then:

with A():
   f()
   g()

would print:

f()
g()

Specifically, my intention is to take this code and create a function out of it. So I can do: with runInThreads() or with runManyTimesTillSuccess(),..

Answer 1

Here’s how you could use a decorator with an argument:

>>> def manytimes(n):
    def decorate(fn):
        for i in range(n):
            fn()
    return decorate

>>> @manytimes(3)
def _():
    print("hello")


hello
hello
hello