Dataclasses y protocolos

Teoría22 min20 XP

Escribir __init__, __repr__ y __eq__ a mano para cada clase de datos es tedioso y repetitivo. Las DATACLASSES los generan por ti con una sola línea: @dataclass. Y los PROTOCOLS traen a Python el "duck typing" tipado —definir qué debe TENER un objeto, sin importar de qué clase sea—, primo directo del tipado estructural que dominaste en TypeScript. Hoy cierras la POO con dos herramientas modernas que hacen tu código más limpio y flexible.

La fábrica de fichas automática

Dataclasses: menos código repetitivo

Con @dataclass sobre la clase y los campos declarados con sus TIPOS (nombre: str), Python genera automáticamente __init__ (para crear objetos), __repr__ (para imprimirlos bien) y __eq__ (para compararlos por contenido) —todo lo que antes escribías a mano—. Los campos pueden tener valores por defecto (categoria: str = "otros"). Es la forma moderna y recomendada de crear clases que principalmente GUARDAN datos: menos código, menos errores, misma funcionalidad. Puedes seguir agregando métodos propios como en cualquier clase.

Duck typing y protocolos

Si suena como pato, es un pato

Python practica el DUCK TYPING: "si algo camina como pato y suena como pato, trátalo como un pato". No importa de qué CLASE sea un objeto —importa qué puede HACER (qué métodos y atributos tiene)—. Una función que suma valores funciona con CUALQUIER objeto que tenga un .valor, sea un Gasto, un Ingreso o lo que sea:

protocolo.py

from typing import Protocol

# un PROTOCOL define QUÉ debe tener un objeto, no de qué clase es
class TieneValor(Protocol):
  valor: int

# esta función acepta CUALQUIER objeto con un atributo .valor
def total(items: list[TieneValor]) -> int:
  return sum(item.valor for item in items)

# funciona con distintas clases, sin herencia común:
class Gasto:
  def __init__(self, valor): self.valor = valor
class Ingreso:
  def __init__(self, valor): self.valor = valor

print(total([Gasto(4000), Ingreso(50000), Gasto(2800)]))   # 56800

Un Protocol (del módulo typing) define un CONTRATO por ESTRUCTURA: "cualquier objeto que tenga un atributo valor de tipo int". La función total acepta cualquier objeto que cumpla ese contrato, sin importar su clase ni que hereden de algo común. Es EXACTAMENTE el tipado estructural de TypeScript (SC-03): importa la FORMA, no el nombre del tipo. Los protocols te dan la flexibilidad del duck typing CON la seguridad de los tipos —tu editor y mypy verifican que los objetos cumplan la forma—.

Cuándo cada herramienta

Un mapa rápido de lo que viste en el módulo:

Clase normal: cuando el objeto tiene comportamiento rico (métodos que hacen cosas, estado que cambia con validación) —una Cuenta, un Bolsillo—.
@dataclass: cuando el objeto es principalmente DATOS (una ficha: Gasto, Usuario, Punto). Te ahorra el boilerplate. Puedes agregarle métodos igual.
Protocol: cuando quieres que una función acepte "cualquier cosa que tenga X", sin atarla a una clase concreta —flexibilidad tipada—.
Herencia: cuando hay una relación "ES UN" clara y quieres reutilizar/especializar. Con moderación.

Con esto tienes el kit completo de la POO en Python: clases para modelar, herencia para especializar, dunder methods para comportamiento nativo, properties para proteger, dataclasses para datos, y protocols para contratos flexibles.

¿Qué genera @dataclass y qué es un Protocol?

Mini-reto

Moderniza Mi Bolsillo: 1) convierte tu clase Gasto en una @dataclass con campos nombre: str, valor: int y categoria: str = "otros", y comprueba que print y == funcionan solos; 2) crea una lista de gastos como dataclasses y suma sus valores con una comprehension; 3) (en papel o probándolo) piensa en una función total(items) que acepte tanto Gasto como cualquier objeto con .valor —el duck typing en acción—. Nota cuánto código te ahorró @dataclass.

Qué sigue

¡Módulo 4 casi cerrado! Dominas la POO en Python: clases, atributos, herencia, dunder methods, properties, dataclasses y protocols. Ya modelas el mundo con objetos. El reto del módulo junta todo en un modelo completo de Mi Bolsillo con clases. Después, el módulo 5 traerá el Python INTERMEDIO —excepciones, archivos, context managers— para escribir programas robustos que interactúan con el mundo real.