def display_airline_flights(self):
if self.base_scrapper.check_code_exist(self.input_value):
url = '{}{}'.format(BASE_URL, self.input_value)
new_scrapper = AirlineScraper(url, "prettyTable")
return new_scrapper.display_airline_flights()
else:
return "Airline code doesn't exist"
def __init__(self, input_value):
self.input_value = input_value
self.base_scrapper = AirlineScraper(BASE_URL, "prettyTable")
def display_flight_history(self):
url = '{}{}'.format(FLIGHT_HISTORY, self.input_value)
new_scrapper = AirlineScraper(url, "prettyTable")
return new_scrapper.flight_history()
