main.py

from openpyxl import load_workbook  # type: ignore
from typing import List


class Worksheet:
    def __init__(self) -> None:
        self.sheet_items: List[dict] = []

    def __repr__(self) -> str:
        return str(self.sheet_items)

    def xlsx_to_dict(self, path, select_sheet=None):
        """
        Read a Worksheet and return it as array of dictionaries
        :param self: Worksheet Object
        :param path: Path to XLSX file
        :param select_sheet: Set active sheet by name
        :return: Array of rows as dictionaries
        """
        book = load_workbook(path)
        if select_sheet is None:
            sheet = book.active
        else:
            sheet = book[select_sheet]
        rows = sheet.max_row
        cols = sheet.max_column

        self.sheet_items = []

        def item(row, col):
            return (
                sheet.cell(row=1, column=col).value,
                str(sheet.cell(row=row, column=col).value),
            )

        for row in range(2, rows + 1):
            self.sheet_items.append(
                dict(item(row, col) for col in range(1, cols + 1))
            )

        return self

    def csv_to_dict(self, csv_file, delimiter=","):
        """
        Read a CSV and return it as array of dictionaries
        :param self: CSV Object
        :param csv_file: CSV file
        :param delimiter: CSV delimiter fe. ';', ','
        :return: Array of rows as dictionaries
        """
        import csv

        self.sheet_items = []

        dict_reader = csv.DictReader(csv_file, delimiter=delimiter)
        self.sheet_items = list(dict_reader)
        return self.sheet_items

    @property
    def header(self):
        """
        Return first row of data as dictionary
        :return: first row of data as dictionary
        """
        return self.sheet_items[0]

    @property
    def sanitize_sheet_items(self):
        """
        Remove None or empty keys from dictionaries
        :param self: XLSX or CSV Object
        :return: Array of rows as dictionaries
        """
        sanitized_items = self.sheet_items
        for item in sanitized_items:
            if None in item:
                del item[None]
            elif "" in item:
                del item[""]
        return sanitized_items