modules/PartField/partfield/config/__init__.py

import argparse
import os.path as osp
from datetime import datetime
import pytz

def default_argument_parser(add_help=True, default_config_file=""):
    parser = argparse.ArgumentParser(add_help=add_help)
    parser.add_argument("--config-file", '-c', default=default_config_file, metavar="FILE", help="path to config file")
    parser.add_argument(
        "--opts",
        help="Modify config options using the command-line",
        default=None,
        nargs=argparse.REMAINDER,
    )
    return parser

def setup(args, freeze=True):
    from .defaults import _C as cfg
    cfg = cfg.clone()
    cfg.merge_from_file(args.config_file)
    cfg.merge_from_list(args.opts)
    dt = datetime.now(pytz.timezone('America/Los_Angeles')).strftime('%y%m%d-%H%M%S')
    cfg.output_dir = osp.join(cfg.output_dir, cfg.name, dt)
    if freeze:
        cfg.freeze()
    return cfg