graph_module.py 文件源码

def main(args):
    '''Graph module's main function'''
    abs_path = ''
    if args.database is None:
        abs_path = os.getcwd()
    else:
        abs_path = os.path.abspath(args.database)
    if not database.is_database(abs_path, verbose=True):
        sys.exit()

    db = database.Database(abs_path)

    for graph_type in args.types:
        if graph_type == 'fitness':
            graph.fitness_graph(db, args.objectives)
        elif graph_type == 'pareto':
            graph.pareto_graph(db, args.objectives)