qué - ¿Cómo funcionan__enter__ y__exit__ en las clases de decoradores de Python?

patron decorator python

Question

Estoy tratando de crear una clase decoradora que cuente cuántas veces se llama a una función, pero aparece un mensaje de error que dice:

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

y realmente no sé cómo le doy cuatro argumentos. Mi código se ve así:

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

¿Hay otros parámetros que debería (o no debería) pasar a la función def exit ? Cualquier consejo o idea sería apreciado.

Además, mi línea de código que dice "print ''f count ='', f.count" parece estar emitiendo la dirección de memoria en lugar del valor, pero ese es un problema completamente diferente.

Answer 1

El __exit__() debe aceptar información sobre las excepciones que surgen en el bloque with: Ver here

La siguiente modificación de tu código funciona:

def __exit__(self, exc_type, exc_value, tb):
    if exc_type is not None:
        traceback.print_exception(exc_type, exc_value, tb)
        # return False # uncomment to pass exception through

    return True

Luego, puede intentar generar una excepción en uno de sus bloques with: y quedará atrapado en __exit__() .