class Create_Menu:
'''Menu for Create options
This class allows the user to choose what to create.
-----------------------------------------------------
-create_employee = indicates if we would like to create new employee on the system
-create_destination = indicates if we would like to create new destination on the system
-create_airplane = indicates if we would like to create new airplane on the system
-create__voyage = indicates if we woule like to create new voyage on the system
'''
def __init__(self):
self.__llapi = LLAPI()
def create_menu(self):
action = ""
while (action != "b"):
print(header_string("CREATE", 50))
print_create_menu()
action = input("Choose an option: ").lower()
while action not in ["1", "2", "3", "4", "b"]:
print("Invalid input. Please choose from the list")
if action == "1":
self.__create_employee()
elif action == "2":
self.__create_destination()
elif action == "3":
self.__create_airplane()
elif action == "4":
self.__new_voyage()
elif action == "b":
pass
def __create_employee(self):
'''Takes no input. Prints on the screen and asks for input to create
an employee in the database. If all input is there and correctly
typed it is saved to the employee.csv data file'''
print(header_string("CREATE EMPLOYEE", 50))
occupation_str = self.__llapi.choose_occupation()
if occupation_str:
print(please_fill_info())
print("Occupation: ", occupation_str)
name_str = get_string("Name")
SO_str = input("Social Security Number: ")
while not (self.__llapi.is_ssn_valid(SO_str)):
print("Please insert a valid 10-digit social security number.")
SO_str = input("Social Security Number: ")
if self.__llapi.check_if_ssn_unique(SO_str):
address_str = get_address()
home_phone_str = self.__llapi.get_phone("Home")
cell_phone_str = self.__llapi.get_phone("Cell")
email_str = get_email()
if occupation_str in ["C", "P"]:
print("")
print('List of airplane models')
airplanes = self.__llapi.get_airplanes()
print_airplane_models(airplanes)
airplane_license_str = self.__llapi.get_airplane_model()
else:
airplane_license_str = "N/A"
print("")
if is_correct():
new_employee = Employee(occupation_str, name_str, SO_str,
address_str, home_phone_str,
cell_phone_str, email_str,
airplane_license_str)
if self.__llapi.add_employee(new_employee):
print(header_string("SUCCESS!", 50))
press_enter()
else:
print(
"Oh-oh something went wrong! Please fill in all information"
)
try_again()
self.__create_employee()
else:
self.__create_employee()
else:
print("The SSN already exists!")
press_enter()
def __create_destination(self):
'''Takes no input. Prints on the screen and asks for input to create
a destination in the database. If all input is there and correctly
typed it is saved to the destination.csv data file'''
print(header_string("CREATE DESTINATION", 50))
print(please_fill_info())
country_str = get_string("Country")
airport_str = get_string("Airport")
if self.__llapi.is_airport_unique(airport_str):
duration_str = self.__llapi.get_destination_duration()
distance_str = get_number("Distance from Iceland (km)")
contact_name_str = get_string("Contact name")
contact_phone_nr_str = self.__llapi.get_phone("Contact")
if is_correct():
new_destination = Destination(country_str, airport_str,
duration_str, distance_str,
contact_name_str,
contact_phone_nr_str)
if self.__llapi.add_destination(new_destination):
print(header_string("SUCCESS!", 50))
press_enter()
else:
print("Oh no something went wrong! Please try again.")
try_again()
self.__create_destination()
else:
self.__create_destination()
else:
print("Airport already exists.")
press_enter()
def __create_airplane(self):
'''Takes no input. Prints on the screen and asks for input to create
a new airplane in the database. If all input is there and correctly
typed it is saved to the airplane.csv data file'''
print(header_string("CREATE AIRPLANE", 50))
print(please_fill_info())
name_str = input("Name: ")
if self.__llapi.is_airplane_unique(name_str):
model_str = input("Model: ")
producer_str = input("Producer: ")
number_of_seats_str = get_number("Number of seats")
print("")
if is_correct():
print(header_string("SUCCESS!", 50))
new_airplane = Airplane(name_str, model_str, producer_str,
number_of_seats_str)
self.__llapi.add_airplane(new_airplane)
press_enter()
else:
self.__create_airplane()
else:
print("Airplane already exists")
press_enter()
def __new_voyage(self):
'''Takes no input. Prints on the screen and asks for input to create
a new voyage in the database. First step is to input airport name
and then it calls copy_voyage function if user wants to copy an older
voyage already in the system. If not it calls the create_voyage
function.'''
print(header_string("CREATE VOYAGE", 50))
print(please_fill_info())
airport = self.__llapi.get_destination()
print_airport(airport)
destination_str = self.__llapi.get_voyage_airport()
copy_voyage = input(
"\nDo you want to copy an existing voyage? (Y/N): ").lower()
while copy_voyage != "y" and copy_voyage != "n":
print("Wrong input. Please choose Y or N")
copy_voyage = input(
"\nDo you want to copy an existing voyage? (Y/N): ").lower()
if copy_voyage == "y":
self.__copy_voyage(destination_str)
else:
self.__create_voyage(destination_str)
def __copy_voyage(self, airport):
'''Takes in a name of the airport that the voyage is scheduled for. Asks for if
user wants to man voyage if needed and if user wants to change employees on
the voyage. Then it writes the new voyage to the voyage.csv file.'''
print("\nWhat date are you looking for? (only use numbers)")
airport_str = airport
copy_year_str, copy_month_str, copy_day_str = self.__llapi.get_voyage_date(
)
voyages = self.__llapi.get_voyage_destination(airport,
int(copy_year_str),
int(copy_month_str),
int(copy_day_str))
if print_voyages_destination(voyages, airport):
flight_number = input(
"Please insert flight number for the voyage: ").upper()
the_voyage = self.__llapi.get_the_voyage(airport,
int(copy_year_str),
int(copy_month_str),
int(copy_day_str),
flight_number)
copy_voyage = the_voyage[0]
new_year_str, new_month_str, new_day_str = self.__llapi.get_voyage_date(
)
print("")
availableplanes = self.__llapi.get_airplane_status(
int(new_year_str), int(new_month_str), int(new_day_str))
new_hour_str, new_minutes_str = self.__llapi.get_voyage_time()
new_departure_time = datetime.datetime(int(new_year_str),
int(new_month_str),
int(new_day_str),
int(new_hour_str),
int(new_minutes_str),
0).isoformat()
airplane_str = copy_voyage.airplane
if copy_voyage.captain == "N/A":
man_voyage = input(
"Would you like to man the voyage at this time? (Y/N): "
).lower()
while man_voyage != "y" and man_voyage != "n":
print("Wrong input. Please choose Y or N")
man_voyage = input(
"Would you like to man the voyage at this time? (Y/N): "
).lower()
if man_voyage == "y":
self.__man_voyage(airport_str, new_departure_time,
airplane_str, new_year_str, new_month_str,
new_day_str, new_hour_str, new_minutes_str)
else:
new_voyage = Voyage(airport_str, new_departure_time,
airplane_str)
self.__llapi.add_voyage(new_voyage)
else:
change_employees = input(
"Would you like to change employees for the voyage? (Y/N): "
).lower()
while change_employees != "y" and change_employees != "n":
print("Wrong input. Please choose Y or N")
change_employees = input(
"Would you like to change employees for the voyage? (Y/N): "
).lower()
if change_employees == "y":
self.__man_voyage(airport_str, new_departure_time,
airplane_str, new_year_str, new_month_str,
new_day_str, new_hour_str, new_minutes_str)
else:
new_voyage = Voyage(airport_str, new_departure_time,
airplane_str, copy_voyage.captain,
copy_voyage.pilot, copy_voyage.fsm,
copy_voyage.flight_attendant)
self.__llapi.add_voyage(new_voyage)
def __create_voyage(self, destination):
'''Takes in a name of the airport that the voyage is scheduled for. Asks for if
user wants to man voyage. Then it writes the new voyage to the voyage.csv file.'''
destination_str = destination
year_str, month_str, day_str = self.__llapi.get_voyage_date()
hour_str, minutes_str = self.__llapi.get_voyage_time()
new_departure_time = datetime.datetime(int(year_str), int(month_str),
int(day_str), int(hour_str),
int(minutes_str),
0).isoformat()
all_planes = self.__llapi.get_airplane(int(year_str), int(month_str),
int(day_str), int(hour_str),
int(minutes_str))
temp_lst = []
for item in all_planes:
if item.plane_status == "Available":
temp_lst.append(item.name)
print_airplane_name_and_models(all_planes)
print("The listed airplanes are available for the given date and time")
airplane_str = self.__llapi.get_voyage_airplane(temp_lst)
print("")
man_voyage = input(
"Would you like to man the voyage at this time? (Y/N): ").lower()
while man_voyage != "y" and man_voyage != "n":
print("Wrong input. Please choose Y or N")
man_voyage = input(
"Would you like to man the voyage at this time? (Y/N): "
).lower()
if man_voyage == "y":
self.__man_voyage(destination_str, new_departure_time,
airplane_str, year_str, month_str, day_str,
hour_str, minutes_str)
else:
new_voyage = Voyage(destination_str, new_departure_time,
airplane_str)
self.__llapi.add_voyage(new_voyage)
def __man_voyage(self, destination_str, new_departure_time, airplane_str,
year_str, month_str, day_str, hour_str, minutes_str):
'''Takes in name of airport (destination), departure time, airplane name
and date and time to create a new voyage. If all input is valid it
vill write the voyage to the voyage.csv file.'''
''' Prenta lausa flugstjóra'''
airplanes = self.__llapi.get_airplane(int(year_str), int(month_str),
int(day_str), int(hour_str),
int(minutes_str))
for item in airplanes:
if airplane_str == item.name:
model = item.model
pilots_model = self.__llapi.get_available_pilots(
int(year_str), int(month_str), int(day_str), model)
print_pilots_by_model(pilots_model)
captain_str = self.__llapi.get_crew("captain")
while not self.__llapi.check_occupation("C", captain_str,
pilots_model):
print(not_licensed())
captain_str = self.__llapi.get_crew("captain")
pilot_str = self.__llapi.get_crew("pilot")
while not self.__llapi.check_occupation("P", pilot_str, pilots_model):
print(not_licensed())
pilot_str = self.__llapi.get_crew("pilot")
flight_attendants = self.__llapi.get_available_crew(
int(year_str), int(month_str), int(day_str))
print_flight_attendants(flight_attendants)
fsm_str = self.__llapi.get_crew("flight service manager")
while not self.__llapi.check_occupation("FSM", fsm_str,
flight_attendants):
print(not_licensed())
fsm_str = self.__llapi.get_crew("flight service manager")
fa_on_voyage_str = input(
"Would you like to add a Flight Attendant on this voyage? (Y/N): "
).lower()
while fa_on_voyage_str != "y" and fa_on_voyage_str != "n":
print("Wrong input. Please choose Y or N")
fa_on_voyage_str = input(
"Would you like to add a Flight Attendant on this voyage? (Y/N): "
).lower()
if fa_on_voyage_str == "y":
fa_str = self.__llapi.get_crew("flight attendant")
while not self.__llapi.check_occupation("FA", fa_str,
flight_attendants):
print(not_licensed())
fa_str = self.__llapi.get_crew("flight attendant")
else:
fa_str = "N/A"
if is_correct():
print(header_string("SUCCESS!", 50))
new_voyage = Voyage(destination_str, new_departure_time,
airplane_str, captain_str, pilot_str, fsm_str,
fa_str)
self.__llapi.add_voyage(new_voyage)
press_enter()
else:
self.__create_voyage(destination_str)
class Update_Menu:
def __init__(self):
self.__llapi = LLAPI()
def update_menu(self):
'''Prints the update menu on the screen and asks the user
to choose an action.'''
action = ""
while (action != "b"):
print(header_string("UPDATE", 50))
print("1: Update employee")
print("2: Update airport contact info")
print("3: Update voyage")
print("b: Back")
print("")
action = input("Choose an option: ").lower()
if action == "1":
self.__update_employee()
elif action == "2":
self.__update_destination()
elif action == "3":
pass
elif action != 'b':
print(
header_string('WRONG INPUT, please select a valid input!',
50))
try_again()
elif action == "3":
self.__update_voyage()
def __update_employee(self):
'''Takes no input. Prints on the screen and asks for input to update/change
the employee informations that can be changed. There is no output
printed only saved in the employee.csv data file'''
new_employee = []
print(header_string("UPDATE EMPLOYEE", 50))
emp_to_update = self.__llapi.get_employee()
print_possible_employee_for_update(emp_to_update)
employee = input("Social Security Number: ")
while not (self.__llapi.is_ssn_valid(employee)):
print("Please insert a valid 10-digit social security number.")
employee = input("Social Security Number: ")
if not self.__llapi.check_if_ssn_unique(employee):
employee_information = self.__llapi.get_employee_information(
employee)
print_employee(employee_information)
new_occupation = self.__llapi.choose_occupation()
print("Occupation: ", new_occupation)
print("")
print("------------------------------------------")
print("To leave information unchanged press enter")
print("------------------------------------------")
new_address = get_address()
new_home_phone = self.__llapi.get_phone("Home")
new_cell_phone = self.__llapi.get_phone("Cell")
new_email = get_email("update")
if new_occupation in ["C", "P"]:
print("")
print('List of airplane models')
airplanes = self.__llapi.get_airplanes()
print_airplane_models(airplanes)
new_licence = self.__llapi.get_airplane_model("update")
if is_correct():
new_employee.extend([
new_occupation, new_address, new_home_phone,
new_cell_phone, new_email, new_licence
])
self.__llapi.update_employee(employee, new_employee)
print(header_string("SUCCESS!", 50))
press_enter()
else:
self.__update_employee()
else:
print("Employee doesn't exist")
press_enter()
def __update_destination(self):
'''Takes no input. Prints on the screen and asks for input to update/change
the contact informations for a selected destination. There is no output
printed only saved in the destination.csv data file'''
print(header_string("UPDATE DESTINATION", 50))
new_contact = []
action2 = ""
airport = self.__llapi.get_destination()
print_airport(airport)
destination = self.__llapi.get_voyage_airport()
print("")
print("What information would you like to update?")
print("1: Contact name")
print("2: Contact phone")
print("3: Both")
print("")
action2 = input("Choose an option: ").lower()
print("")
if action2 == "1":
new_name = get_string("New contact name")
new_contact.append(new_name)
new_contact.append("")
elif action2 == "2":
new_phone = self.__llapi.get_phone("New contact")
new_contact.append("")
new_contact.append(new_phone)
elif action2 == "3":
new_name = get_string("New contact name")
new_phone = self.__llapi.get_phone("New contact")
new_contact.append(new_name)
new_contact.append(new_phone)
if is_correct():
print(header_string("SUCCESS!", 50))
self.__llapi.update_destination(destination, new_contact)
press_enter()
else:
self.__update_destination()
def __update_voyage(self):
'''Takes no input. Prints on the screen and asks for input to man a voyage.
There is no output printed only saved in the voyage.csv data file'''
print(header_string("MAN A VOYAGE", 50))
'''Lista upp Destination'''
airport = self.__llapi.get_destination()
print_airport(airport)
voyage_destination = self.__llapi.get_voyage_airport()
'''Velja dagsetningu'''
print("What date are you looking for? (only use numbers)")
year_str, month_str, day_str = self.__llapi.get_voyage_date()
'''Lista upp ómannaðar voyages'''
voyages = self.__llapi.get_voyage_destination(voyage_destination,
int(year_str),
int(month_str),
int(day_str))
'''Láta velja voyage'''
if print_voyages_destination(voyages, voyage_destination):
the_voyage_lst = self.__llapi.get_flight_number(
voyage_destination, year_str, month_str, day_str)
the_voyage = the_voyage_lst[0]
airplanes = self.__llapi.get_airplanes()
for item in airplanes:
if the_voyage.airplane == item.name:
model = item.model
pilots_model = self.__llapi.get_available_pilots(
int(year_str), int(month_str), int(day_str), model)
print_pilots_by_model(pilots_model)
captain_str = self.__llapi.get_crew("captain")
while not self.__llapi.check_occupation("C", captain_str,
pilots_model):
print(not_licensed())
captain_str = self.__llapi.get_crew("captain")
pilot_str = self.__llapi.get_crew("pilot")
while not self.__llapi.check_occupation("P", pilot_str,
pilots_model):
print(not_licensed())
pilot_str = self.__llapi.get_crew("pilot")
flight_attendants = self.__llapi.get_available_crew(
int(year_str), int(month_str), int(day_str))
print_flight_attendants(flight_attendants)
fsm_str = self.__llapi.get_crew("flight service manager")
while not self.__llapi.check_occupation("FSM", fsm_str,
flight_attendants):
print(not_licensed())
fsm_str = self.__llapi.get_crew("flight service manager")
fa_on_voyage_str = input(
"Would you like to add a Flight Attendant on this voyage? (Y/N): "
).lower()
while fa_on_voyage_str != "y" and fa_on_voyage_str != "n":
print("Wrong input. Please choose Y or N")
fa_on_voyage_str = input(
"Would you like to add a Flight Attendant on this voyage? (Y/N): "
).lower()
if fa_on_voyage_str == "y":
fa_str = self.__llapi.get_crew("flight attendant")
while not self.__llapi.check_occupation(
"FA", fa_str, flight_attendants):
print(not_licensed())
fa_str = self.__llapi.get_crew("flight attendant")
else:
fa_str = "N/A"
if is_correct():
print(header_string("SUCCESS!", 50))
self.__llapi.update_voyage(the_voyage, captain_str, pilot_str,
fsm_str, fa_str)
else:
self.__update_voyage()
press_enter()
class Get_Menu:
'''Get Option - Menu
This class allows the user to choose what to he can see/get:
------------------------------------------------------------
-Get_employee = if chosen get a list of employee information
-Get_destination = if chosen get a list of destinations
-Get_airplane = if chosen get airplane information
-Get_Voyage = if chosen get voyage information
-back = go back to main menu
'''
def __init__(self):
self.__llapi = LLAPI()
def get_menu(self):
'''Prints the get menu on the screen and asks the user
to choose an action.'''
action = ""
while (action != "b"):
print(header_string("GET", 50))
print_get_menu()
action = input("Choose an option: ").lower()
if action == "1":
self.__get_employee()
elif action == "2":
self.__get_destination()
elif action == "3":
self.__get_airplane_information()
elif action == "4":
self.__get_voyage()
elif action != 'b':
print(
header_string('WRONG INPUT, please select a valid input!',
50))
try_again()
def __get_employee(self):
'''Takes no input. Prints on the screen and asks for what employee information
the user wants to see. Based on more input the output is printed on the
screen after it has been fetched from the data files'''
print(header_string("GET EMPLOYEE INFORMATION", 50))
get_employee_information()
action = input("Choose an option: ").lower()
if action == "1":
print(header_string("GET ALL EMPLOYEES", 50))
employees = self.__llapi.get_employee()
print_employee(employees)
press_enter()
elif action == "2":
print(header_string("GET EMPLOYEE INFORMATION", 50))
SO_str = input("Social Security Number for the employee: ")
while not (self.__llapi.is_ssn_valid(SO_str)):
print("Please insert a valid 10-digit social security number.")
SO_str = input("Social Security Number: ")
if not self.__llapi.check_if_ssn_unique(SO_str):
employee_information = self.__llapi.get_employee_information(
SO_str)
print_employee(employee_information)
else:
print("Employee doesn't exist")
press_enter()
elif action == "3":
print(header_string("GET EMPLOYEES BY OCCUPATION", 50))
print_employee_by_occupation()
occupation = input(
"What occupation would you like to get? ").upper()
while occupation not in ["C", "P", "FA", "FSM"]:
print("Invalid input. Please choose an option from the list")
occupation = input(
"What occupation would you like to get? ").upper()
employee_by_occupation = self.__llapi.get_employee_by_occupation(
occupation)
print_employee(employee_by_occupation)
press_enter()
elif action == "4":
print(header_string("GET EMPLOYEES BY STATUS", 50))
print_employee_by_status()
employee_status = input(
"What status would you like to get? ").upper()
while employee_status not in ["A", "B"]:
print("Invalid input. Please choose employee status.")
employee_status = input(
"What status would you like to get? ").upper()
employee_by_status = self.__llapi.get_employee_by_status(
employee_status)
print_employee(employee_by_status)
press_enter()
elif action == "5":
check = 0
print(header_string("GET WEEK SCHEDULE FOR AN EMPLOYEE", 50))
employee = input("Social Security Number: ")
while not (self.__llapi.is_ssn_valid(employee)):
print("Please insert a valid 10-digit social security number.")
employee = input("Social Security Number: ")
if not (self.__llapi.check_if_ssn_unique(employee)):
while check == 0:
print(
"\nPlease insert date and you will get the week schedule that includes that date."
)
input_year, input_month, input_day = self.__llapi.get_voyage_date(
)
employee_information = self.__llapi.get_employee_information(
employee)
week_dates = self.__llapi.get_week_lst(
input_year, input_month, input_day)
week_schedule_lst = self.__llapi.get_week_schedule(
employee, input_year, input_month, input_day)
if week_schedule_lst:
print_employee_schedule(employee_information,
week_dates, week_schedule_lst)
check += 1
press_enter()
else:
print("Date out of bounds. Please try again.")
else:
print("Employee does not exist")
press_enter()
elif action == "6":
print(header_string("GET ALL PILOTS BY AIRPLANE MODEL", 50))
pilots = self.__llapi.get_pilots_by_airplane()
print_pilots_by_airplane(pilots)
press_enter()
elif action == "7":
print(header_string("GET PILOTS BY AN AIRPLANE MODEL", 50))
airplanes = self.__llapi.get_airplanes()
print_airplane_models(airplanes)
model_to_find = input("What airplane would you like to get? ")
pilots_model = self.__llapi.get_pilots_by_model(model_to_find)
if pilots_model:
print_pilots_by_model(pilots_model)
else:
print("\n" + "No pilots are licenced for that airplane")
press_enter()
def __get_destination(self):
'''Takes no input. Prints on the screen information about all destinations that
NaN air has on its books.'''
print(header_string("GET DESTINATIONS", 50))
destination = self.__llapi.get_destination()
print_destination(destination)
press_enter()
def __get_airplane_information(self):
'''Takes no input. Prints on the screen information about all airplanes that
NaN air owns and if the are available or not.'''
print(header_string("GET AIRPLANE INFORMATION", 50))
today = datetime.datetime.now()
airplanes = self.__llapi.get_airplane(today.year, today.month,
today.day, today.hour,
today.minute)
voyages = self.__llapi.get_all_voyage_at_date(today.year, today.month,
today.day)
voyages_lst = []
for item in airplanes:
if item.plane_status == 'Available':
voyages_lst.append(" ")
if item.plane_status != 'Available':
for voyage in voyages:
if item.name == voyage.airplane:
voyages_lst.append(voyage)
print_airplanes(airplanes, voyages_lst)
press_enter()
def __get_voyage(self):
'''Takes no input. Prints on the screen and asks for what voyage the
user wants to see information for. Based on user selection all
information about the selected voyage is printed on the screen.'''
print(header_string("GET VOYAGE INFORMATION", 50))
airport = self.__llapi.get_destination()
print_airport(airport)
voyage_destination = self.__llapi.get_voyage_airport()
print("What date are you looking for? (only use numbers)")
year_str, month_str, day_str = self.__llapi.get_voyage_date()
voyages = self.__llapi.get_voyage_destination(voyage_destination,
int(year_str),
int(month_str),
int(day_str))
if print_voyages_destination(voyages, voyage_destination):
voyage = self.__llapi.get_flight_number(voyage_destination,
year_str, month_str,
day_str)
print_the_voyage(voyage)
press_enter()