def clear_old(rt_path, branch): branch_dir = os.path.join(rt_path, "email_src", branch) date = chart.read_chart(rt_path, branch)[0][0] for f in os.listdir(branch_dir): if f.split(".")[1] == "xls" or f.split(".")[1] == "txt": pass elif f.split(".")[1] == "html": if f.split(".")[0].split("_")[-1] == date: pass # print f else: template_path = os.path.join(branch_dir, f) os.remove(template_path) print "Html delete -", f elif f.split(".")[1] == "png": if f.split(".")[0].split("_")[-1] == date: pass # print f else: template_path = os.path.join(branch_dir, f) os.remove(template_path) print "Image delete -", f else: other_path = os.path.join(branch_dir, f) os.remove(other_path) print "delete ", f print "Files remained:" for f in os.listdir(branch_dir): print "---", f return 0
y_range = top - bottom y_step = math.ceil(y_range/6) y_axis =[int(top)] for n in range(1, 7): y_axis.append(int(top - n*y_step)) y_axis_str = [str(y)+'%' for y in y_axis] plt.yticks(y_axis, y_axis_str, fontsize=font_size) plt.legend(bbox_to_anchor=(0.7, -0.2), frameon=False, ncol=3, fontsize=font_size, handlelength=handlelength) graph_path = os.path.join(graph_dir, branch+'_rate_'+date_list[0]+'.png') if os.path.exists(graph_path): change = 0 else: change = 1 fig.savefig(graph_path, dpi=dpi) if user == read_para.owner and change == 1: os.chmod(graph_path, stat.S_IRWXG+stat.S_IRWXU) fig.clf() return branch+'_'+date_list[0]+'.png', branch+'_rate_'+date_list[0]+'.png' if __name__ == '__main__': path = os.path.split(os.path.realpath(__file__))[0] argument = read_para.parse_args() branches = argument.branch for branch in branches: branch = branch.upper() datelist, datadict = chart.read_chart(path, branch) # for date in datelist: # print date, datadict[date] pic1, pic2 = draw_graph(path, branch, datelist, datadict, 30) print 'Graph '+pic1+' and '+pic2+' completed!'