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!'
