from Bus import Bus
bus1 = Bus(int(input('Enter number of seats in the bus: ')))
print(bus1)
action = int(
input('TAKE Action!\n1 - Passenger in\n2 - Passenger out\n0 - END\n'))
print(bus1)
while action != 0:
if action == 1:
bus1.get_on(input('Enter passenger name to get on: '))
elif action == 2:
bus1.get_off(input('Enter passenger name to get off: '))
else:
print('invalid action.')
print(bus1)
action = int(
input('TAKE Action!\n1 - Passenger in\n2 - Passenger out\n0 - END\n'))
print(bus1)
