E-commerce #11 : Historique des commandes

from decimal import Decimal
p = Produit('A')
h = HistoriqueCommandes()
h.enregistrer(Commande({p: 1}, '50.00'))
h.enregistrer(Commande({p: 2}, '100.00'))
assert h.total_revenus() == Decimal('150.00')

from decimal import Decimal
p = Produit('A')
h = HistoriqueCommandes()
h.enregistrer(Commande({p: 1}, '10', 'payee'))
h.enregistrer(Commande({p: 1}, '10', 'annulee'))
assert len(h.commandes_par_statut('payee')) == 1

Solution officielle

from decimal import Decimal
from collections import defaultdict

class Produit:
    def __init__(self, nom):
        self.nom = nom

class Commande:
    def __init__(self, items, total, statut='payee'):
        self.items = items
        self.total = Decimal(str(total))
        self.statut = statut

class HistoriqueCommandes:
    def __init__(self):
        self.commandes = []

    def enregistrer(self, commande):
        self.commandes.append(commande)

    def total_revenus(self):
        return sum((c.total for c in self.commandes), Decimal('0'))

    def commandes_par_statut(self, statut):
        return [c for c in self.commandes if c.statut == statut]

    def produit_le_plus_vendu(self):
        ventes = defaultdict(int)
        for commande in self.commandes:
            for produit, qte in commande.items.items():
                ventes[produit.nom] += qte
        if not ventes:
            return None
        return max(ventes, key=lambda n: ventes[n])