csv_analyzer.py

import csv


class CSVAnalyzer:

    def __init__(self, file_name):
        self.file_name = file_name
        self.employees = []

    def load_data(self):

        with open(
            self.file_name,
            "r",
            newline=""
        ) as file:

            reader = csv.DictReader(file)

            for row in reader:
                self.employees.append(row)

    def get_average_salary(self):

        total_salary = sum(
            int(employee["Salary"])
            for employee in self.employees
        )

        return total_salary / len(self.employees)

    def get_department_summary(self):

        summary = {}

        for employee in self.employees:

            department = employee["Department"]

            if department not in summary:
                summary[department] = 0

            summary[department] += 1

        return summary

    def get_highest_salary(self):

        return max(
            self.employees,
            key=lambda employee: int(employee["Salary"])
        )