Registro automático de subclases

Introducción

En este proyecto, aprenderás a implementar una clase base en Python que registre automáticamente sus subclases. Esta funcionalidad puede ser útil en varios escenarios, como cuando se trabaja con modelos de base de datos o con arquitecturas basadas en plugins.

🎯 Tareas

En este proyecto, aprenderás:

• Cómo crear una metaclase para controlar el proceso de creación de clases
• Cómo implementar el método __init_subclass__ para registrar automáticamente las subclases
• Cómo hacer que una clase sea iterable implementando el método __iter__

🏆 Logros

Después de completar este proyecto, podrás:

• Comprender el concepto de metaclases y cómo se pueden utilizar para personalizar el comportamiento de las clases
• Implementar una clase base que registre automáticamente sus subclases
• Iterar sobre las subclases registradas de una clase base

Skills Graph

Implementando la clase base

En este paso, aprenderás a implementar la clase Base que registra automáticamente sus subclases.

1. Crea un archivo self_register.py en el directorio /home/labex/project.

cd /home/labex/project
touch self_register.py

2. Abre el archivo /home/labex/project/self_register.py en tu editor de código.
3. Crea una clase BaseMeta que herede de type. Esta clase se utilizará como metaclase para la clase Base.

class BaseMeta(type):
    def __iter__(self):
        return iter(self.subclasses)

El método __iter__ se implementa para hacer que la clase Base sea iterable, lo que te permite iterar sobre sus subclases registradas.

4. Crea la clase Base y configura su metaclase como BaseMeta.

class Base(metaclass=BaseMeta):
    subclasses = []

    def __init_subclass__(cls, **kwargs):
        super().__init_subclass__(**kwargs)
        Base.subclasses.append(cls)

El método __init_subclass__ se sobrescribe para agregar automáticamente cualquier subclase de Base a la lista subclasses.

Probando la clase base

En este paso, probarás la funcionalidad de la clase Base creando algunas subclases e iterando sobre ellas.

1. Al final del archivo self_register.py, agrega el siguiente código:

if __name__ == "__main__":
    class Txzy(Base):
        pass

    class SomeComplexCase(Base):
        pass

    assert Txzy in list(Base)
    assert SomeComplexCase in list(Base)

    for cls in Base:
        pass

Este código crea dos subclases de Base, Txzy y SomeComplexCase, y luego verifica que estén registradas en la clase Base. Finalmente, itera sobre la clase Base para asegurarse de que la iteración funcione como se espera.

2. Guarda el archivo self_register.py.
3. Ejecuta el script:

python /home/labex/project/self_register.py

Si el script se ejecuta sin errores, la clase Base está funcionando como se espera.

¡Felicidades! Has implementado la clase Base que registra automáticamente sus subclases. En los siguientes pasos, puedes explorar casos de uso más avanzados para esta funcionalidad.

Resumen

¡Felicidades! Has completado este proyecto. Puedes practicar más laboratorios en LabEx para mejorar tus habilidades.