from tren import Tren


class TrenExtra(Tren):
    def __init__(self, idtren, velmax):
        super().__init__(idtren)
        self.vmax = velmax

    def quants_capacitat(self, capmax):
        n = 0
        for vag in self:
            if vag.capacitat() >= capmax:
                n = n + 1
        return n