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
