Gestores de contexto

Parte de la sección Programación orientada a objetos del Journey de Python de Coddy. Lección 41 de 64.

Los administradores de contexto te permiten asignar y liberar recursos justo cuando los necesitas. Garantizan una limpieza adecuada incluso si se producen errores.

Aquí tienes el ejemplo más común de uso de la instrucción with:

with open('example.txt', 'w') as file:
    file.write('Hello, world!')
# El archivo se cierra automáticamente aquí

El archivo se cierra automáticamente después del bloque, incluso si se produce una excepción.

Crea tu propio gestor de contexto implementando los métodos __enter__ y __exit__:

class MyContext:
    def __enter__(self):
        print("Entering the context")
        return self
    
    def __exit__(self, exc_type, exc_val, exc_tb):
        print("Exiting the context")
        return False  # No suprimas las excepciones

Usa tu administrador de contexto personalizado:

with MyContext() as ctx:
    print("Inside the context")

Salida:

Entering the context
Inside the context
Exiting the context

Crea un administrador de contexto más práctico para las conexiones a bases de datos:

class DatabaseConnection:
    def __init__(self, db_name):
        self.db_name = db_name
        self.connection = None
    
    def __enter__(self):
        print(f"Connecting to {self.db_name}")
        self.connection = f"Connection to {self.db_name}"
        return self.connection
    
    def __exit__(self, exc_type, exc_val, exc_tb):
        print(f"Closing connection to {self.db_name}")
        self.connection = None

with DatabaseConnection("users_db") as conn:
    print(f"Using {conn}")
    print("Performing database operations...")

Maneja las excepciones en los administradores de contexto:

class SafeContext:
    def __enter__(self):
        print("Setting up resources")
        return self
    
    def __exit__(self, exc_type, exc_val, exc_tb):
        print("Cleaning up resources")
        if exc_type:
            print(f"An exception occurred: {exc_val}")
        return False  # No suprimas la excepción

with SafeContext():
    print("Working with resources")
    # raise ValueError("Something went wrong")  # Descomenta para probar

Salida:

Connecting to users_db
Using Connection to users_db
Performing database operations...
Closing connection to users_db
Setting up resources
Working with resources
Cleaning up resources

El método __exit__ recibe tres parámetros:

exc_type: Tipo de excepción (o None)
exc_val: Valor de la excepción (o None)
exc_tb: Traza de la excepción (o None)

Punto clave: Los gestores de contexto usan los métodos __enter__ y __exit__ para gestionar recursos. La instrucción with llama automáticamente a estos métodos, lo que garantiza una configuración y una limpieza adecuadas. Esto resulta especialmente útil para archivos, conexiones a bases de datos y otros recursos que requieren una limpieza garantizada.

Desafío

Fácil

En este desafío, implementarás una clase de administrador de contexto que mide el tiempo transcurrido entre la entrada y la salida de un bloque de contexto.

timer.py - Contiene la implementación de la clase Timer (este es el archivo que editarás)
driver.py - Contiene casos de prueba exhaustivos (no lo modifiques)

Implementa el administrador de contexto Timer en timer.py siguiendo los comentarios TODO. La clase debe:

Registrar la hora de inicio al entrar en un contexto
Registrar la hora de finalización al salir
Calcular y mostrar el tiempo transcurrido

Pruébalo tú mismo

from timer import Timer
import time

# Gestor de casos de prueba
test_case = input()

if test_case == "basic_test":
    # Prueba de funcionalidad básica
    with Timer():
        # Simular algo de trabajo
        time.sleep(2)  # Esto hace que el programa espere 2 segundos

elif test_case == "nested_contexts":
    # Probar gestores de contexto anidados
    with Timer():
        time.sleep(1)
        with Timer():
            time.sleep(1)

elif test_case == "exception_handling":
    # Probar el manejo de excepciones dentro del contexto
    try:
        with Timer():
            time.sleep(1)
            raise ValueError("Test exception")
    except ValueError as e:
        print(f"Caught exception: {e}")

elif test_case == "zero_sleep":
    # Probar con una pausa mínima
    with Timer():
        time.sleep(0.001)

elif test_case == "multiple_timers":
    # Probar varios temporizadores en secuencia
    with Timer():
        time.sleep(1)
    
    with Timer():
        time.sleep(0.5)
    
    with Timer():
        time.sleep(2)

elif test_case == "as_decorator":
    # Probar el temporizador en el contexto de una función
    def timed_function():
        with Timer():
            time.sleep(2)
    
    timed_function()

else:
    # Prueba predeterminada - funcionalidad básica
    with Timer():
        # Simular algo de trabajo
        time.sleep(2)  # Esto hace que el programa espere 2 segundos