15. Задача на взаимодействие классов. Требуется создать систему, названную Касса железнодорожной станции . Клиент

python

from datetime import datetime



class Train:

    def __init__(self, number, source_station, destination_station, intermediate_stations, ticket_price):

        self.number = number

        self.source_station = source_station

        self.destination_station = destination_station

        self.intermediate_stations = intermediate_stations

        self.ticket_price = ticket_price



class Station:

    def __init__(self, name):

        self.name = name



class RailwayStationTicketOffice:

    def __init__(self):

        self.client_requests = []

        self.train_list = []



    def register_request(self, destination_station, travel_date_time):

        self.client_requests.append((destination_station, travel_date_time))



    def find_suitable_train(self, destination_station, travel_date_time):

        for train in self.train_list:

            if destination_station in [train.destination_station] + train.intermediate_stations:

                if travel_date_time >= datetime.now():

                    return train

        return "Подходящий поезд не найден."



    def generate_invoice(self, train):

        invoice = """Счет на оплату:

        Номер поезда: {}

        Станция отправления: {}

        Станция назначения: {}

        Цена билета: {}

        Дата и время поездки: {}""".format(train.number, train.source_station, train.destination_station, train.ticket_price, datetime.now())

        return invoice



# Пример использования



station1 = Station("Москва")

station2 = Station("Санкт-Петербург")

station3 = Station("Тверь")

station4 = Station("Псков")



train1 = Train(1, "Москва", "Санкт-Петербург", ["Тверь"], 1000.0)

train2 = Train(2, "Москва", "Псков", ["Тверь"], 1500.0)

train3 = Train(3, "Санкт-Петербург", "Москва", [], 800.0)



office = RailwayStationTicketOffice()

office.train_list = [train1, train2, train3]



office.register_request("Санкт-Петербург", datetime(2022, 12, 1, 10, 0, 0))

office.register_request("Тверь", datetime(2022, 12, 2, 12, 30, 0))

office.register_request("Псков", datetime(2022, 12, 3, 15, 45, 0))



request1 = office.client_requests[0]

request2 = office.client_requests[1]

request3 = office.client_requests[2]



selected_train1 = office.find_suitable_train(request1[0], request1[1])

selected_train2 = office.find_suitable_train(request2[0], request2[1])

selected_train3 = office.find_suitable_train(request3[0], request3[1])



if isinstance(selected_train1, Train):

    invoice1 = office.generate_invoice(selected_train1)

else:

    invoice1 = selected_train1



if isinstance(selected_train2, Train):

    invoice2 = office.generate_invoice(selected_train2)

else:

    invoice2 = selected_train2



if isinstance(selected_train3, Train):

    invoice3 = office.generate_invoice(selected_train3)

else:

    invoice3 = selected_train3



print(invoice1)

print(invoice2)

print(invoice3)