def createFamily(size: int): if size == 3: a, b, c = Passenger('a', 'b', 'c', 0), Passenger('a', 'b', 'c', 0), Passenger('a', 'b', 'c', 0) a.category = "Family" b.category = "Family" c.category = "Family" return a, b, c elif size == 4: a, b, c, d = Passenger('a', 'b', 'c', 0), Passenger('a', 'b', 'c', 0), Passenger( 'a', 'b', 'c', 0), Passenger('a', 'b', 'c', 0) a.category = "Family" b.category = "Family" c.category = "Family" d.category = "Family" return a, b, c, d else: a, b, c, d, e = Passenger('a', 'b', 'c', 0), Passenger( 'a', 'b', 'c', 0), Passenger('a', 'b', 'c', 0), Passenger('a', 'b', 'c', 0), Passenger('a', 'b', 'c', 0) a.category = "Family" b.category = "Family" c.category = "Family" d.category = "Family" e.category = "Family" return a, b, c, d, e
def createTourists(): x, y = Passenger(a, b, c, d), Passenger(b, a, c, d) x.category = "Tourist" y.category = "Tourist" return x, y
def createBusiness(): x = Passenger(a, b, c, d) x.category = "Business" return x