def main(): # Gather and validate input (user provides names of start/end stations) # Assumption: User is not transferring start_station = input("Enter a starting T station\n") while not mbta.is_valid_station(start_station): start_station = input("I've never heard of that station, " "please enter again\n") end_station = input("Enter your destination T station\n") while not mbta.is_valid_station(end_station): end_station = input("I've never heard of that station, " "please enter again\n") # Figure out the direction to go on that line (final station # in the direction from start to end) direction = mbta.get_direction(start_station, end_station) num_stops = mbta.get_num_stops(start_station, end_station) # Report the result, assuming we"re actually going somewhere if num_stops == 0: print(start_station, " to ", end_station, "? " "I can't help you with that one!", sep = "") else: print("Get on at ", start_station, " toward ", direction, ".\n" "Take the train for ", num_stops, " stops and arrive at ", end_station, ".", sep = "")
def main(): commands = sys.argv if len(commands) > 1: end_station = sys.argv[2] start_station = sys.argv[1] else: start_station = input("Enter a starting T station\n") while not mbta.is_valid_station(start_station): start_station = input("Never heard of that station, " "please enter again\n") end_station = input("Enter your destination T station\n") while not mbta.is_valid_station(end_station): end_station = input("I've never heard of that station, " "please enter again\n") direction = mbta.get_direction(start_station, end_station) num_stops = mbta.get_num_stops(start_station, end_station) if num_stops == 0: print(start_station, " to ", end_station, "?" "I can't help you with that one!", sep = "") else: print("Get on at ", start_station, " toward ", direction, ".\n" "Take the train for ", num_stops, " stops and arrive at ", end_station, ".", sep = "")