srz-zumix / iutest

tools/python/eval_line.py

#!/usr/bin/env python
#
# eval_line.py
#
# Copyright (C) 2014-2020, Takazumi Shirayanagi
# This software is released under the new BSD License,
# see LICENSE
#

import re
import codecs

from argparse import ArgumentParser
from argparse import Action


# EvalIntAction
class EvalIntAction(Action):
    def __call__(self, parser, namespace, values, options_string=None):
        setattr(namespace, self.dest, int(eval(values[0])))


# command line option
def parse_command_line():

This code seems to be duplicated in your project.

    parser = ArgumentParser()
    parser.add_argument(
        '-v',
        '--version',
        action='version',
        version=u'%(prog)s version 0.2'
    )
    parser.add_argument(
        '-r',
        '--repeat',
        action=EvalIntAction,
        nargs=1,
        default=1,
        help='Set eval repeat count.'
    )
    parser.add_argument(
        '-s',
        '--start',
        action=EvalIntAction,
        nargs=1,
        default=0,
        help='Set eval repeat start no.'
    )
    parser.add_argument(
        '-o',
        '--output',
        default=None,
        help='output file path.'
    )
    parser.add_argument(
        '--encoding',
        default=None,
        help='output file encoding.'
    )
    parser.add_argument(
        'expression',
        metavar='EXP',
        nargs='+',
        help='eval expressions'
    )
    options = parser.parse_args()
    return options


def eval_line(r, define, line):
    m = r.finditer(define)
    if m is None:
        return define
    else:
        out = define
        for e in m:
            out = out.replace(e.group(0), str(eval(e.group(1))))
        return out


def main():
    re_eval = re.compile(r'\${([^}]+)}')
    options = parse_command_line()
    define = options.expression[0]
    for a in options.expression[1:]:
        define += ' ' + a

    output = options.output
    if output is None:
        for line in range(options.repeat):
            print(eval_line(re_eval, define, line + options.start))
    else:
        if options.encoding:
            file = codecs.open(output, 'w', options.encoding)
        else:
            file = open(output, 'w')
        for line in range(options.repeat):
            file.write(eval_line(re_eval, define, line + options.start))
            file.write('\n')
        file.close()


if __name__ == '__main__':
    main()