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 = "")