csv_reader.py

import csv


class CSVReader:
    def __init__(self, filename, skip_first_row=True):
        self.filename = filename
        self.skip_first_row = skip_first_row

    def read_records(self):
        records = []
        with open(self.filename, newline='', encoding='utf-8-sig') as csvfile:
            reader = csv.reader(csvfile, delimiter=',')
            if self.skip_first_row:
                next(reader)
            for row in reader:
                record = {
                    "Column1": row[0],
                    "Column2": row[1],
                    "company": row[2],
                    "line": row[3],
                    "departure_time": row[4],
                    "arrival_time": row[5],
                    "start_stop": row[6],
                    "end_stop": row[7],
                    "start_stop_lat": row[8],
                    "start_stop_lon": row[9],
                    "end_stop_lat": row[10],
                    "end_stop_lon": row[11],
                }
                records.append(record)
        return records