def _get_date_partitioned_paths()

in osci/datalake/local/staging.py [0:0]

• 12 lines of code
• 4 McCabe index (conditional complexity)

    def _get_date_partitioned_paths(self, dir_path: Union[str, Path], year: Union[str, int] = None,
                                    month: Union[str, int] = None, day: Union[str, int] = None,
                                    company: str = None,
                                    file_format='parquet'):

        if year is not None:
            dir_path = dir_path / str(year)
            if month is not None:
                dir_path = dir_path / str(month).zfill(2)
                if day is not None:
                    dir_path = dir_path / str(day).zfill(2)

        file_pattern = f'{normalize_company(name=company) + "-" if company else ""}*.{file_format}'
        return self._get_paths(dir_path=dir_path, file_pattern=file_pattern)