import os
from pathlib import Path


class DatasetPaths:
    """
    Класс для хранения путей к файлам и директориям, задействованным при формировании датасета.

    root_path: os.PathLike - корневая директория, в которой будет храниться датасет
    save_intermediate_files: bool - флаг, указывающий, нужно ли сохранять промежуточные файлы, \
        которые используются только при формировании датасета
    """

    def __init__(
        self,
        root_path: os.PathLike,
        save_intermediate_files: bool = False,
    ):
        self.root_path = Path(root_path)
        self.abbreviations = self.root_path / 'abbreviations.csv'
        self.xml_info = self.root_path / 'xml_info.csv'
        self.dataset = self.root_path / 'dataset.pkl'

        self.save_intermediate_files = save_intermediate_files
        self.txt_path = self.root_path / 'txt' if save_intermediate_files else None
        self.txt_abbr = self.root_path / 'txt_abbr' if save_intermediate_files else None
        self.jsons_path = self.root_path / 'jsons' if save_intermediate_files else None