auto_max_all.py

from informat import *
from maxpoints import *
import pprint

def main():
    sit_lines = []
    for file_name in ['inex.txt', 'extra_inex.txt']:
        f = open(file_name)
        sit_lines.extend(get_sits_from_file(f))
    for sit_line in sit_lines:
        sit = parse_command_line(sit_line)
        print "\n Situation line: %s" % sit_line
        opts = get_options(sit)
        for i, option in enumerate(opts):
            print 'Option #%d' % i
            pprint.pprint(option)
            selected_fans = option_max_points(option)
            print 'Selected fans: '
            pprint.pprint(selected_fans)
            print 'Points: %d' % get_total_points(selected_fans)

def get_sits_from_file(f):
    sits = []
    try:
        for line in f:
            if not line.startswith('#'):
                line = line[:-1]
                sits.append(line)
    finally:
        f.close()
    return sits

if __name__ == "__main__":
    main()