Sobes.tech
Junior — Senior
48

Варианты оптимизации и исправления представленного кода

Условие задачи

В этом примере показан класс, отвечающий за формирование отчётов. В конструкторе создаётся словарь data, в котором хранятся финансовые и нефинансовые отчёты. Для нефинансовых используется defaultdict с шаблоном, возвращающим структуру записи. Затем вызываются вспомогательные методы, которые заполняют данные, а при выполнении определённого условия – дополнительно обновляют их.

class GenerateReport:
    def __init__(self, *args, **kwargs):
        self.data = {
            'fin_reports': dict(),
            'non_fin_reports': defaultdict(self._non_fin_record_template),
        }
        self.setup_some_data(*args, **kwargs)
        if some_condition:
            self.update_some_data(*args, **kwargs)
            # several more such method calls
            ...

    def _non_fin_record_template(self):
        return {
            'sections': [],
            'totals': [],
        }

    def setup_some_data(self, *args, **kwargs):
        # make some calculations
        # ...
        # then set the data
        self.data['non_fin_reports']['sections']['section']['sub_section'] = some_data

    def update_some_data(self, *args, **kwargs):
        # make some calculations
        # ...
        # then update the data
        self.data['non_fin_reports']['sections']['section'] = updated_data

    @property
    def data(self):
        return self._data

    # usage example
    generate_report(*args, **kwargs)
    service = GenerateReport(*args, **kwargs)
    return Response(data=service.data)

В описании указаны типичные проблемы, которые могут возникнуть в таком коде: конфликт имён свойства и атрибута, отсутствие импорта defaultdict, некорректные обращения к вложенным структурам словаря и т.п. Предлагается проанализировать эти моменты и предложить варианты их исправления.