Resumen de pedidos en XML con ElementTree

1import sys
2import xml.etree.ElementTree as ET
3from decimal import Decimal, InvalidOperation
4
5ESTADOS = ("pendiente", "enviado", "entregado", "cancelado")
6
7
8def leer_pedido(nodo):
9    """Comprueba un <pedido> y devuelve su id, cliente, estado e importe; lanza ValueError con el motivo."""
10    for atributo in ("id", "cliente", "estado"):
11        if not nodo.get(atributo):
12            raise ValueError(f"falta el atributo {atributo}")
13    if nodo.get("estado") not in ESTADOS:
14        raise ValueError(f"estado desconocido {nodo.get('estado')}")
15    lineas = nodo.findall("linea")
16    if not lineas:
17        raise ValueError("no tiene líneas")
18    importe = Decimal(0)
19    unidades = {}
20    for linea in lineas:
21        ref, cantidad, precio = linea.get("ref"), linea.get("cantidad", ""), linea.get("precio", "")
22        if not ref or not cantidad.isdigit() or int(cantidad) == 0:
23            raise ValueError(f"línea {ref or '?'} con una cantidad no válida")
24        try:
25            precio = Decimal(precio)
26        except InvalidOperation:
27            raise ValueError(f"línea {ref} con un precio no válido")
28        importe += int(cantidad) * precio
29        unidades[ref] = unidades.get(ref, 0) + int(cantidad)
30    return nodo.get("id"), nodo.get("cliente"), nodo.get("estado"), importe, unidades
31
32
33def main():
34    try:
35        raiz = ET.fromstring(sys.stdin.read())
36    except ET.ParseError as e:
37        print(f"XML mal formado (línea {e.position[0]})")
38        return
39    if raiz.tag != "pedidos":
40        print("El elemento raíz debe ser <pedidos>")
41        return
42
43    por_cliente = {}      # cliente → [pedidos, importe]
44    por_estado = {e: 0 for e in ESTADOS}
45    vendidas = {}
46    total = Decimal(0)
47    validos = 0
48    for nodo in raiz.findall("pedido"):
49        try:
50            id_, cliente, estado, importe, unidades = leer_pedido(nodo)
51        except ValueError as e:
52            print(f"Aviso: pedido {nodo.get('id', '?')} ignorado: {e}")
53            continue
54        validos += 1
55        por_estado[estado] += 1
56        if estado == "cancelado":
57            continue          # se cuenta, pero no suma importe ni unidades
58        datos = por_cliente.setdefault(cliente, [0, Decimal(0)])
59        datos[0] += 1
60        datos[1] += importe
61        total += importe
62        for ref, n in unidades.items():
63            vendidas[ref] = vendidas.get(ref, 0) + n
64
65    # El resumen se construye como un árbol y se serializa con sangría
66    resumen = ET.Element("resumen", fecha=raiz.get("fecha", ""), pedidos=str(validos), importe=f"{total:.2f}")
67    estados = ET.SubElement(resumen, "estados")
68    for estado in ESTADOS:
69        ET.SubElement(estados, "estado", nombre=estado, pedidos=str(por_estado[estado]))
70    clientes = ET.SubElement(resumen, "clientes")
71    for cliente, (n, importe) in sorted(por_cliente.items(), key=lambda c: (-c[1][1], c[0])):
72        ET.SubElement(clientes, "cliente", nombre=cliente, pedidos=str(n), importe=f"{importe:.2f}")
73    if vendidas:
74        ref, n = min(vendidas.items(), key=lambda v: (-v[1], v[0]))
75        ET.SubElement(resumen, "masVendido", ref=ref, unidades=str(n))
76    ET.indent(resumen, space="  ")
77    print(ET.tostring(resumen, encoding="unicode"))
78
79
80main()