import argparse
from problem_generator import ProblemGenerator


if __name__ == '__main__':
    parser = argparse.ArgumentParser()

    parser.add_argument('type', type=str)
    parser.add_argument('name', type=str)

    args = parser.parse_args()

    pg = ProblemGenerator(
        problem_type=args.type,
        problem_name=args.name,
        problem_description=''
    )

    if pg.check_problem_solved():
        print 'Problem {} solved'.format(pg)
        pg.save_db()