Perfomance improvement using slots

Slots_demo.py

import logging
import timeit
from dataclasses import dataclass
from functools import partial
from statistics import median

logging.basicConfig(level=logging.INFO)


@dataclass(slots=True)
class SchoolSlots:
    name: str
    rating: str
    students: int


@dataclass(slots=False)
class School:
    name: str
    rating: str
    students: int


def get_set(school: School | SchoolSlots):
    _ = school.name
    school.name = "No Name"
    _ = school.name


def main():
    wo_slots = School("Name 1", "A", 1000)
    w_slots = SchoolSlots("Name 2", "A+", 1000)

    no_slots_stat = median(timeit.repeat(partial(get_set, wo_slots), number=10 ** 6))
    with_slots_stat = median(timeit.repeat(partial(get_set, w_slots), number=10 ** 6))

    logging.info(f"Average improvement: {(no_slots_stat - with_slots_stat) * 100 / no_slots_stat}")


if __name__ == '__main__':
    main()