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()