Skip to content

Commit a84fd77

Browse files
Chore: minor code cleanup for readability and functions return boolean (True/False) where needed for unittesting.
1 parent 8015e2c commit a84fd77

File tree

2 files changed

+8
-3
lines changed

2 files changed

+8
-3
lines changed

Travel_Itinerary_Planner/main.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -313,11 +313,13 @@ def run_app():
313313
else:
314314
for flight in itinerary["flights"]:
315315
flight_options.append(flight["flight name"])
316+
316317
# Checks if 'while True' statement should be broken
317318
if not multiple_flights:
318319
print("Returning to main menu...")
319320
break
320321
flight_id, itinerary_index = pick(flight_options, flight_prompt)
322+
321323
if delete_itinerary_item(itineraries, selected_type, itinerary_option, flight_id):
322324
print(f"Flight '{flight_id}' has been deleted.")
323325
else:
@@ -351,6 +353,7 @@ def run_app():
351353
print("Returning to main menu...")
352354
break
353355
attraction_id, itinerary_index = pick(attraction_options, attraction_prompt)
356+
354357
if delete_itinerary_item(itineraries, selected_type, itinerary_option, attraction_id):
355358
print(f"Attraction '{attraction_id}' has been deleted.")
356359
else:

Travel_Itinerary_Planner/src/manage_itineraries.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -168,6 +168,7 @@ def add_new_flight(itinerary_list, itinerary_name, new_flights):
168168
else:
169169
print(f"Duplicate flight detected: {flight["flight name"]}!")
170170
print("This flight will not be added.")
171+
return False
171172
save_itineraries(itinerary_list)
172173
# Uncomment to print for itinerary_list validation:
173174
# print(itinerary_list)
@@ -183,6 +184,7 @@ def add_new_attraction(itinerary_list, itinerary_name, new_attractions):
183184
else:
184185
print(f"Duplicate attraction detected: {attraction["attraction name"]}!")
185186
print("This attraction will not be added.")
187+
return False
186188
save_itineraries(itinerary_list)
187189
# Uncomment to print for itinerary_list validation:
188190
# print(itinerary_list)
@@ -239,12 +241,12 @@ def print_table(trips):
239241
trip_table = Table(title="Itineraries", show_lines=True)
240242

241243
trip_table.add_column("Trip Name", justify="center", no_wrap=True)
242-
trip_table.add_column("Location", justify="center")
243-
trip_table.add_column("Description", justify="center")
244+
trip_table.add_column("Location", justify="center", no_wrap=True)
245+
trip_table.add_column("Description", justify="center", no_wrap=False)
244246
trip_table.add_column("Start Date", justify="center", no_wrap=True)
245247
trip_table.add_column("End Date", justify="center", no_wrap=True)
246248
trip_table.add_column("Flights", justify="left", no_wrap=True)
247-
trip_table.add_column("Attractions", justify="left", style="bold")
249+
trip_table.add_column("Attractions", justify="left", style="bold", no_wrap=False)
248250

249251
for trip in trips:
250252
flight_list = ''

0 commit comments

Comments
 (0)