export_constant Namespace Reference

Functions
	main ()

Function Documentation

main()

export_constant.main

(

void

)

Definition at line 23 of file export_constant.py.

def main():
    parser = argparse.ArgumentParser(
        description='Export CONSTANT value with given file format.')
    parser.add_argument('-c',
                        '--constant',
                        type=str,
                        required=True,
                        help='Constant name to export')
    parser.add_argument(
        '-f',
        '--format',
        type=str,
        required=True,
        choices=['cfg', 'txt'],
        help=
        'File format to export. The created cfg file contains CONSTANT under the one-optimize section.'
    )
    parser.add_argument(
        '--exclusive',
        action='store_true',
        help='Exports the rest of the options except for the given constant')
    parser.add_argument('-o',
                        '--output_path',
                        type=str,
                        required=True,
                        help='Path to output')
 
    args = parser.parse_args()
 
    if not hasattr(CONSTANT, args.constant):
        raise NameError('Not found given constant name')
 
    if args.exclusive:
        constant_to_exclude = getattr(CONSTANT, args.constant)
        constant_to_export = []
        for opt in CONSTANT.OPTIMIZATION_OPTS:
            if opt[0] in constant_to_exclude:
                continue
            constant_to_export.append(opt[0])
    else:
        constant_to_export = getattr(CONSTANT, args.constant)
 
    if args.format == 'cfg':
        SECTION_TO_EXPORT = 'one-optimize'
        config = configparser.ConfigParser()
        config[SECTION_TO_EXPORT] = dict()
        for constant in constant_to_export:
            config[SECTION_TO_EXPORT][constant] = 'True'
 
        with open(args.output_path, 'w') as f:
            config.write(f)
 
    if args.format == 'txt':
        with open(args.output_path, 'w') as f:
            for constant in constant_to_export:
                f.write(f"{constant}\n")
 
 

References main().

Referenced by main().