import os
# from matplotlib.pyplot import *
import matplotlib.pyplot as plt
import pandas as pd
# import numpy as np
# import xlsxwriter
from openpyxl.drawing.image import Image
from openpyxl import load_workbookxxxx
os.chdir('C:/Users/xxxx/Desktop/CommitLog/')
output_dir = 'C:/Users/xxxx/Desktop/CommitLog/'
def excel_draw(csv_file, excel_file):
# pass
df=pd.read_csv(csv_file,encoding='utf-8',index_col=0 , usecols = [0,1])
csv_name =csv_file.split('.')[0].split('/')[-1]
print csv_name
# print df_x
# # df = np.array(df_x)
# x = df_x.values.tolist()
# print x
# df_y=pd.read_csv("./Edk2 - Copy.csv",encoding='utf-8', usecols = [0,1])#index_col=0 ,
# train_data = np.array(df_y)
# print train_data
# y = train_data.tolist()
# print y
df_plot = df.plot(kind='bar', rot=0, color='green', width=0.35)
plt.title('%s Comprehensive' % csv_name, fontproperties='SimHei')
plt.legend(loc=1)
plt.xlabel('modules', fontproperties='SimHei', fontsize=10)
plt.xticks(fontproperties='SimHei',fontsize=10,color='black',rotation=30)
# for a, b in zip(x, y):
# plt.text(a, b + 0.05, '%.0f' % b, ha='center', va='bottom', fontsize=17)
# plt.xlim(0,11)
plt.ylabel('values', fontproperties='SimHei')
plt.savefig('./%s.jpg' % csv_name,dpi=100, bbox_inches = 'tight')
img = Image('./%s.jpg' % csv_name)
workbook1 = load_workbook(excel_file)
worksheet = workbook1[csv_name]
worksheet.add_image(img, 'D2')
workbook1.save(excel_file)
# wb = xlsxwriter.Workbook('all1.xlsx')
# worksheet1.insert_image('D2', './Edk2.jpg')
plt.show()
csv_file = os.path.join(output_dir, "Edk2.csv")
print csv_file
excel_file =os.path.join(output_dir, "WW46.xlsx")
excel_draw(csv_file, excel_file)
# f.figure()
# f.subplot(331)
# bar()