Как __enter__ и __exit__ работают в классах декоратора Python?

Я пытаюсь создать класс декоратора, который подсчитывает, сколько раз вызывается функция, но я получаю сообщение об ошибке:

    "TypeError: __exit__() takes exactly 1 argument (4 given)"

и я действительно не знаю, как я даю ему четыре аргумента. Мой код выглядит следующим образом:

class fcount2(object):
    __instances = {}
    def __init__(self, f):
        self.__f = f
        self.__numcalls = 0
        fcount2.__instances[f] = self

    def __call__(self, *args, **kwargs):
        self.__numcalls += 1
        return self.__f(*args, **kwargs)

    def __enter__(self):
        return self

    def __exit__(self):
        return self

    @staticmethod
    def count(f):
        return fcount2.__instances[self.__f].__numcalls


@fcount2
def f(n):
    return n+2

for n in range(5):
    print f(n)   
print 'f count =',f.count

def foo(n):
    return n*n

with fcount2(foo) as g:
    print g(1)
    print g(2)
print 'g count =',g.count
print 'f count =',f.count

with fcount2(f) as g:
    print g(1)
    print g(2)
print 'g count =',g.count
print 'f count =',f.count

with f:
    print f(1)
    print g(2)
print 'g count =',g.count
print 'f count =',f.count

Есть ли какие-то другие параметры, которые я должен (или не должен) переходить в функцию def exit? Любые советы или идеи будут оценены.

В стороне, моя строка кода, в которой указано "print" f count = ', f.count ", как представляется, выводит адрес памяти, а не значение, но это целая проблема.

Как enter и exit работают в классах декоратора Python?

Ответ 1