executor.py

# @Time   : 2023.03.03
# @Author : Darrius Lei
# @Email  : darrius.lei@outlook.com

import argparse, sys, logging, os
from lib import glb_var, json_util
from lib.callback import Logger, CustomException

if __name__ == '__main__':
    if not os.path.exists('./cache/logger/'):
        os.makedirs('./cache/logger/')
    glb_var.__init__();
    log = Logger(
        level = logging.DEBUG,
        filename = './cache/logger/logger.log',
    ).get_log()
    glb_var.set_value('logger', log);
    parse = argparse.ArgumentParser();
    parse.add_argument('--data_process', '-dp', type = bool, default = False, help = 'Whether to process data(True/False)');
    parse.add_argument('--config', '-cfg', type = str, default = None, help = 'config for run');
    parse.add_argument('--saved_config', '-sc', type = str, default = None, help = 'path for saved config to test')
    parse.add_argument('--mode', type = str, default = 'train', help = 'train/test/train_and_test')

    args = parse.parse_args();

    #execute date process command
    if args.data_process:
        from data.generator import run_pcr
        dp_cfg = json_util.jsonload('./config/data_process_cfg.json');
        run_pcr(cfg = dp_cfg);
        sys.exit(0);

    #execute work command
    from Room.work import run_work
    if args.config is not None:
        if args.mode not in ['train', 'train_and_test']:
            log.error(f'The using of the configuration file does not support the {args.mode} mode. \n'
                      'Prompt: first use the [train] mode and then use the command [-- saved_config] to run the [test] mode, \n'
                      'or directly use the [train_and_test] mode');
            raise CustomException('ModeError');
        run_work(args.config, args.mode);
    elif args.saved_config is not None:
        if args.mode in ['train', 'train_and_test']:
            log.error('The saved model file only supports test mode');
            raise CustomException('ModeError');
        run_work(args.saved_config, args.mode);