show.py

#!/usr/bin/env python3
import argparse
import sys
import glob

from dataparser import parse, nl, ni
from util import path, get_in_file_content

try:     import matplotlib.pyplot as plt
except:  pass

# ns is the namespace. 
# If needed write a simple parser which returns ns.
def test(tc_name, inp):
    print(tc_name)
    print(inp)
    print('==============')
    ns = parse(inp)
    # TODO: display inp in readable form!

def get_tasks():
    parser = argparse.ArgumentParser()
    parser.add_argument('testcase', nargs='?', default=None)
    parser.add_argument('-a', '--all', action='store_true')
    args = parser.parse_args()
    if not (args.testcase or args.all):
        print('provide either all: -a or a testcase.')
        exit()
    tasks = []
    if args.all:
        for f in sorted(glob.glob('in/*.in')):
            tc_name = path(f).name
            tasks.append(tc_name)
    else:
        tc_name = path(args.testcase).name
        tasks.append(tc_name)
    return tasks

if __name__=='__main__':
    tasks = get_tasks()
    
    for tc_name in tasks:
        inp = get_in_file_content(tc_name)
        test(tc_name, inp)