def draw(json_parser, if_print): palette = json_parser.get_palette() screen = json_parser.get_screen() shapes = json_parser.get_shapes() fig = plt.figure(figsize=(screen["width"], screen["height"]), dpi=1) fig.artists.append(plt.Rectangle((0, 0), screen["width"], screen["height"], facecolor=JSONParser.return_color(palette, screen["bg_color"]), zorder=0)) for i in shapes: i.add_shape(fig) if if_print == 1: fig.savefig(argv[3]) plt.close(fig) else: plt.show()