piecharts.py 文件源码

def generateImages(picklefile, pickledir, filehash, imagedir, pietype):

    leaf_file = open(os.path.join(pickledir, picklefile), 'rb')
    (piedata, pielabels) = cPickle.load(leaf_file)
    leaf_file.close()

    pylab.figure(1, figsize=(6.5,6.5))
    ax = pylab.axes([0.2, 0.15, 0.6, 0.6])

    pylab.pie(piedata, labels=pielabels)

    pylab.savefig(os.path.join(imagedir, '%s-%s.png' % (filehash, pietype)))
    pylab.gcf().clear()
    os.unlink(os.path.join(pickledir, picklefile))