processing.py

# Copyright (c) 2020 Anastasiia Birillo, Elena Lyulina

import sys
import logging

sys.path.append('.')
from src.main.util import consts
from src.main.cli.util import ICli
from src.main.cli.configs import PROCESSING_LEVEL, PROCESSING_PARAMS
from src.main.util.log_util import configure_logger, add_console_stream

log = logging.getLogger(consts.LOGGER_NAME)


class PreprocessingCli(ICli):

    def __init__(self):
        super().__init__()
        self._path = None
        self._level = None

    def configure_args(self) -> None:
        self._parser.add_argument(PROCESSING_PARAMS.PATH.value, type=str, nargs=1, help='data path')
        self._parser.add_argument(PROCESSING_PARAMS.LEVEL.value, nargs='?', const=3, default=3,
                                  help=PROCESSING_LEVEL.description())

    def parse_args(self) -> None:
        args = self._parser.parse_args()
        self._path = self.handle_path(args.path[0])
        self._level = self.str_to_preprocessing_level(args.level)

    def main(self) -> None:
        self.parse_args()
        path = self._path
        for level_index in range(0, self._level.value + 1):
            current_level = PROCESSING_LEVEL(level_index)
            self._log.info(f'Current action is {current_level.level_handler()}')
            path = current_level.level_handler()(path)
        self._log.info(f'Folder with data: {path}')
        print(f'Folder with data: {path}')


if __name__ == '__main__':
    configure_logger(to_delete_previous_logs=True)
    add_console_stream(log)

    preprocessing_cli = PreprocessingCli()
    preprocessing_cli.main()