from aviores import AvioRes
class AvioCat(AvioRes):
    def __init__(self, ide, nfil, nfp):
        super().__init__(ide, nfil)
        self.nfilprim = nfp
        
    def seients_categoria(self, cat):
        if cat == 1:
            return self.nfilprim * 6
        else:
            return (len(self)-self.nfilprim)*6

    def seients_lliures(self):
        n1 = 0
        for i in range(self.nfilprim):
            for lletra in 'ABCDEF':
                if self[i, lletra] == 'Lliure':
                    n1 = n1 + 1
        n2 = 0
        for i in range(self.nfilprim, len(self)):
            for lletra in 'ABCDEF':
                if self[i, lletra] == 'Lliure':
                    n2 = n2 + 1                    

        return n1, n2