import pathlib


def _get_csv_file_path(data_dir: str, three_sec_song: bool) -> pathlib.Path:
    """Gets CSV file path from data directory.

    Parameters
    ----------
    data_dir : str
        directory where datafiles are stored
    three_sec_song : bool
        if should use the three second song csv (as opposed to thirty second songs)

    Returns
    -------
    pathlib.Path
        path to csv file
    """
    if three_sec_song:
        return pathlib.Path(data_dir).joinpath(pathlib.Path("Data/features_3_sec.csv"))
    else:
        return pathlib.Path(data_dir).joinpath(pathlib.Path("Data/features_30_sec.csv"))