from object_classes.flight_creator import Flight
import datetime

class Voyage:

    def __init__(self, flight_out=None, flight_in=None):
        self.__flight_out = flight_out or Flight()
        self.__flight_in = flight_in or Flight()
        self.valid_voyage = None

    def is_valid(self):
        '''Function to check if the voyage is valid'''
        self.__flight_out.is_valid()
        valid, invalid_list = self.__flight_in.is_valid()  # saves if the flight is valid or not, if it is not valid, saves what is wrong
        return (valid, invalid_list)


    def edit(self, object_type, item):
        '''Function used by the ui to add pilots, or crew to the flight'''
        for element in item:
            object_type = object_type.lower()  # saves the type to edit
            worked1 = self.__flight_out.edit_staff(object_type, element)  # saves if it worked
            worked2 = self.__flight_in.edit_staff(object_type, element)
        return worked1 and worked2  # returns true if both worked

    def edit_options(self):
        '''Used by the ui to know what he can edit'''
        return ["Pilot", "Crew"]


    def edit_airplane_ID(self, new_ID):
        '''Takes in an airplane id and replaces it'''
        self.__flight_out.edit_airplane(new_ID)
        self.__flight_in.edit_airplane(new_ID)


    def edit_destination(self, new_destination):
        '''Used by the ui to edit destinations in a flight'''
        where, rkv = new_destination
        self.__flight_out.edit_destination(where)  # adds in where the airplane is going
        self.__flight_in.edit_destination(rkv)
        self.__flight_out.edit_departure(rkv)
        self.__flight_in.edit_departure(where)

    def edit_departure_time(self, new_time):
        '''Takes in a departure time from the user, error checks it, then calculates the arrival time'''
        if len(new_time) == 16:
            try:
                datetime.datetime.strptime(new_time,"%Y-%m-%d,%H:%M")  # checks if the time is valid
            except ValueError:
                return False
            arrival, time = self.__flight_out.edit_departure_time(new_time)  # edits the departure time, returns the arrival
            departure_time = arrival + datetime.timedelta(hours=1)  # adds one hour
            self.__flight_in.set_departure_time(departure_time)  # sets the departure time for the flight home
            self.__flight_in.set_arrival_time(departure_time + datetime.timedelta(hours=time))  # calculates the time to go home
            return True
        return False

    def get_staff(self, what_type):
        '''Returns the staff members of the flights'''
        return self.__flight_in.get_staff(what_type)

    def get_date(self):
        '''Used to get the flight out from iceland date'''
        date1 = self.__flight_out.get_date()
        if type(date1) is str:  # checks if the object is a string, if it is we need to add the seconds
            if len(date1) != 19:
                date1 += ":00"  # add them
            date1 = datetime.datetime.strptime(date1,"%Y-%m-%dT%H:%M:%S")  # creates the date time object
        return date1  # returns it


    def get_both_dates(self):
        '''Returns the flight in date and flight out date as objects'''
        date1 = self.__flight_out.get_date()
        date2 = self.__flight_in.get_date()
        datelist = []
        ret_datelist = []
        if str(date1)[0:10] == str(date2)[0:10]:  # removes the time from the dates, and checks if the dates are the same
            datelist.append(date1)  # if the dates are the same, returns only one date
        else:
            datelist.append(date1)  # if the dates are diffrent return both dates
            datelist.append(date2)
        for date in datelist: # turns the dates into datetime objects
            if type(date) is str:
                if len(date) != 19: # if the date is missing seconds add them
                    date += ":00"
                date = datetime.datetime.strptime(date,"%Y-%m-%dT%H:%M:%S")  # creates the datetime object
            ret_datelist.append(date)
        return ret_datelist  # returns the date/s

    def set_id(self, flight_id):
        '''Takes in an id and adds it to the out and in flight'''
        if len(flight_id) != 6:
            flight_id = flight_id[:-1] + "0" + flight_id[-1]  # adding the missing 0 if the last number is in the range 0-9
        self.__flight_out.set_id(flight_id)
        in_id = flight_id[:-1] + str(int(flight_id[-2:])+ 1)  # adding one to the end of the id
        self.__flight_in.set_id(in_id)



    def get_destination(self):
        ''' Returns the flight destination'''
        return self.__flight_out.get_destination()

    def get_type(self):
        ''' Returns the class type'''
        return "voyage"

    def save(self):
        ''' Gets a save string the data layer uses to save the flights'''
        result_str = self.__flight_out.save() + "\n" + self.__flight_in.save()
        return result_str

    def display_str(self):
        '''Display string that the ui uses to display the flights'''
        return '{:15} {} {:15}  {}'.format(self.__flight_out.display_str(), "|---|",self.__flight_in.display_str(), self.__flight_in.is_valid()[0])

    def display_str_edit(self):
        '''Used by ui edit'''
        return '{:20}'.format(self.__flight_out.display_str())

    def get_search_reference(self):
        '''Returns a serch refrence for the flight'''
        return self.__flight_out.get_search_reference()