read_xlsx.py

import xlrd

from utils.search_flights_data import SearchFlightsData
from utils.search_hotels_data import SearchHotelsData
from utils.search_tours_data import SearchToursData
from utils.search_transfers_data import SearchTransfersData


class XlsxReader:
    @staticmethod
    def get_xlsx_hotels_data():
        wb = xlrd.open_workbook(f"./utils/search_hotels_input_data.xlsx")
        sheet = wb.sheet_by_index(0)
        data = []

        for i in range(1, sheet.nrows):
            search_hotels_data = SearchHotelsData(sheet.cell(i, 0).value,  # destination
                                                  sheet.cell(i, 1).value,  # check in date
                                                  sheet.cell(i, 2).value,  # check out date
                                                  int(sheet.cell(i, 3).value),  # adults number
                                                  int(sheet.cell(i, 4).value))  # kids number
            data.append(search_hotels_data)
        return data

    @staticmethod
    def get_xlsx_flights_data():
        wb = xlrd.open_workbook(f"./utils/search_flights_input_data.xlsx")
        sheet = wb.sheet_by_index(0)
        data = []

        for i in range(1, sheet.nrows):
            search_flights_data = SearchFlightsData(sheet.cell(i, 0).value,  # cabin class
                                                    sheet.cell(i, 1).value,  # location from
                                                    sheet.cell(i, 2).value,  # location to
                                                    sheet.cell(i, 3).value,  # start year
                                                    sheet.cell(i, 4).value,  # start month
                                                    sheet.cell(i, 5).value,  # start day
                                                    sheet.cell(i, 6).value,  # end year
                                                    sheet.cell(i, 7).value,  # end month
                                                    sheet.cell(i, 8).value,  # end day
                                                    int(sheet.cell(i, 9).value),  # adults number
                                                    int(sheet.cell(i, 10).value),  # kids number
                                                    int(sheet.cell(i, 11).value))  # infants number
            data.append(search_flights_data)
        return data

    @staticmethod
    def get_xlsx_tours_data():
        wb = xlrd.open_workbook(f"./utils/search_tours_input_data.xlsx")
        sheet = wb.sheet_by_index(0)
        data = []

        for i in range(1, sheet.nrows):
            search_tours_data = SearchToursData(sheet.cell(i, 0).value,  # destination
                                                sheet.cell(i, 1).value,  # tour type
                                                sheet.cell(i, 2).value,  # start year
                                                sheet.cell(i, 3).value,  # start month
                                                sheet.cell(i, 4).value,  # start day
                                                int(sheet.cell(i, 5).value))  # adults number
            data.append(search_tours_data)
        return data

    @staticmethod
    def get_xlsx_transfers_data():
        wb = xlrd.open_workbook(f"./utils/search_transfers_input_data.xlsx")
        sheet = wb.sheet_by_index(0)
        data = []

        for i in range(1, sheet.nrows):
            search_transfers_data = SearchTransfersData(sheet.cell(i, 0).value,  # pick up location
                                                        sheet.cell(i, 1).value,  # drop off location
                                                        sheet.cell(i, 2).value,  # depart year
                                                        sheet.cell(i, 3).value,  # depart month
                                                        sheet.cell(i, 4).value,  # depart day
                                                        sheet.cell(i, 5).value,  # depart time
                                                        sheet.cell(i, 6).value,  # return year
                                                        sheet.cell(i, 7).value,  # return month
                                                        sheet.cell(i, 8).value,  # return day
                                                        sheet.cell(i, 9).value)  # return time
            data.append(search_transfers_data)
        return data