#encoding=utf-8
import os,sys,time
from collections import defaultdict
from tkinter import *
import tkinter.messagebox
from tkinter import ttk
from tkinter import scrolledtext

root= Tk()
root.title("有效代碼統(tǒng)計工具") #界面的title

def code_count(path,file_types):
 if os.path.exists(path):
  os.chdir(path)
 else:
  #messagebox.showwarning("您輸入的路徑不存在！")
  print("您輸入的路徑不存在！")
  #sys.exit()
 
 files_path=[]
 file_types=file_types.split()
 line_count=0
 space_count=0
 annotation_count=0
 file_lines_dict=dict()
 for root,dirs,files in os.walk(path):
  for f in files:
   files_path.append(os.path.join(root,f))

 for file_path in files_path:
  #print(os.path.splitext(file_path)[1][1:])
  file_type=os.path.splitext(file_path)[1][1:]
  if file_type in file_types:
   if file_type.lower()=="java":
    line_num,space_num,annotation_num=count_javafile_lines(file_path)
    line_count+=line_num
    space_count+=space_num
    annotation_count+=annotation_num
    file_lines_dict[file_path]=line_num,space_num,annotation_num
   if file_type.lower()=="py":
    line_num,space_num,annotation_num=count_py_lines(file_path)
    line_count+=line_num
    space_count+=space_num
    annotation_count+=annotation_num
    file_lines_dict[file_path]=line_num,space_num,annotation_num
    #file_info=file_show(line_num,space_num,annotation_num)
    #print(file_info[0])
 return line_count,file_lines_dict,space_count,annotation_count
 
def count_py_lines(file_path):
 line_count = 0
 space_count=0
 annotation_count=0
 flag =True
 try:
  fp = open(file_path,"r",encoding="utf-8")
  encoding_type="utf-8"
  for i in fp:
   pass
  fp.close()
 except:
  #print(file_path)
  encoding_type="gbk"

 with open(file_path,"r",encoding=encoding_type,errors="ignore") as fp:
  #print(file_path)
  """try:
   fp.read()
  except:
   fp.close()"""
  for line in fp:   
   if line.strip() == "":
    space_count+=1
   else:
    if line.strip().endswith("'''") and flag == False:
     annotation_count+=1
     #print(line)
     flag = True
     continue
    if line.strip().endswith('"""') and flag == False:
     annotation_count+=1
     #print('結(jié)尾雙引',line)
     flag = True
     continue
    if flag == False:
     annotation_count+=1
     #print("z",line)
     continue 
    """if flag == False:
     annotation_count+=1
     print("z",line)"""
    if line.strip().startswith("#encoding") \
      or line.strip().startswith("#-*-"):
     line_count += 1
    elif line.strip().startswith('"""') and line.strip().endswith('"""') and line.strip() != '"""':
     annotation_count+=1
     #print(line)
    elif line.strip().startswith("'''") and line.strip().endswith("'''") and line.strip() != "'''":
     annotation_count+=1
     #print(line)
    elif line.strip().startswith("#"):
     annotation_count+=1
     #print(line)
    elif line.strip().startswith("'''") and flag == True:
     flag = False
     annotation_count+=1
     #print(line)
    elif line.strip().startswith('"""') and flag == True:
     flag = False
     annotation_count+=1
     #print('開頭雙引',line)
    else:
     line_count += 1
 return line_count,space_count,annotation_count

#path=input("請輸入您要統(tǒng)計的絕對路徑：")
#file_types=input("請輸入您要統(tǒng)計的文件類型：")

#print("整個%s有%s類型文件%d個，共有%d行代碼"%(path,file_types,len(code_dict),codes))
#print("代碼最多的是%s，有%d行代碼"%(max_code[1],max_code[0]))

def count_javafile_lines(file_path):
 line_count = 0
 space_count=0
 annotation_count=0
 flag =True
 #read_type=''
 try:
  fp = open(file_path,"r",encoding="utf-8")
  encoding_type="utf-8"
  for i in fp:
   pass
  fp.close()
 except:
  #print(file_path)
  encoding_type="gbk"

 with open(file_path,"r",encoding=encoding_type) as fp:
  #print(file_path)
  for line in fp:   
   if line.strip() == "":
    space_count+=1
   else:
    if line.strip().endswith("*/") and flag == False:
     flag = True
     annotation_count+=1
     continue
    if flag == False:
     annotation_count+=1
     continue
    elif line.strip().startswith('/*') and line.strip().endswith('*/'):
     annotation_count+=1
    elif line.strip().startswith('/**') and line.strip().endswith('*/'):
     annotation_count+=1    
    elif line.strip().startswith("http://") and flag == True:
     flag = False
     continue
    else:
     line_count += 1
 return line_count,space_count,annotation_count

def show(): #當(dāng)按鈕被點擊，就調(diào)用這個方法
 pathlist=e1.get() #調(diào)用get()方法得到在文本框中輸入的內(nèi)容
 file_types=e2.get().lower()

 file_types_list=["py","java"]
 
 if not pathlist:
  tkinter.messagebox.showwarning('提示',"請輸入文件路徑！")
  return None
 if not file_types:
  tkinter.messagebox.showwarning('提示',"請輸入要統(tǒng)計的類型！")
  return None
 #print(type(file_types),file_types)
 if '\u4e00'<=file_types<='\u9fa5' or not file_types in file_types_list: #判斷文件類型輸入的是否是中文
  tkinter.messagebox.showwarning('錯誤',"輸入統(tǒng)計類型有誤！")
  return None

 text.delete(1.0,END) #刪除顯示文本框中，原有的內(nèi)容
 
 for path in pathlist.split(";"):
  path=path.strip()
  codes,code_dict,space,annotation=code_count(path,file_types) #將函數(shù)返回的結(jié)果賦值給變量，方便輸出
  max_code=max(zip(code_dict.values(),code_dict.keys()))
  #print(codes,code_dict)
  #print("整個%s有%s類型文件%d個，共有%d行代碼"%(path,file_types,len(code_dict),codes))
  #print("代碼最多的是%s，有%d行代碼"%(max_code[1],max_code[0]))
  for k,v in code_dict.items():
   text.insert(INSERT,"文件%s 有效代碼數(shù)%s\n"%(k,v[0])) #將文件名和有效代碼輸出到文本框中
  
  text.insert(INSERT,"整個%s下有%s類型文件%d個，共有%d行有效代碼\n"%(path,file_types,len(code_dict),codes)) #將結(jié)果輸出到文本框中
  text.insert(INSERT,"共有%d行注釋\n"%(annotation))
  text.insert(INSERT,"共有%d行空行\(zhòng)n"%(space))
  text.insert(INSERT,"代碼最多的是%s，有%s行有效代碼\n\n"%(max_code[1],max_code[0][0]))
 
frame= Frame(root) #使用Frame增加一層容器
frame.pack(padx=50,pady=40) #設(shè)置區(qū)域
label= Label(frame,text="路徑：",font=("宋體",15),fg="blue").grid(row=0,padx=10,pady=5,sticky=N) #創(chuàng)建標(biāo)簽
label= Label(frame,text="類型：",font=("宋體",15),fg="blue").grid(row=1,padx=10,pady=5)
e1= Entry(frame,foreground = 'blue',font = ('Helvetica', '12')) #創(chuàng)建文本輸入框
e2= Entry(frame,font = ('Helvetica', '12', 'bold'))
e1.grid(row=0,column=1,sticky=W) #布置文本輸入框
e2.grid(row=1,column=1,sticky=W,)
labeltitle=Label(frame,text="輸入多個文件路徑請使用';'分割",font=("宋體",10,'bold'),fg="red")
labeltitle.grid(row=2,column=1,sticky=NW)
frame.bind_all("<F1>",lambda event:helpinf())
frame.bind_all("<Return>",lambda event:show())
frame.bind_all("<Alt-F4>",lambda event:sys.exit())
frame.bind_all("<Control-s>",lambda event:save())

#print(path,file_types)

hi_there= Button(frame ,text=" 提交 ",font=("宋體",13),width=10,command=show).grid(row=3,column=0,padx=15,pady=5) #創(chuàng)建按鈕
hi_there= Button(frame ,text=" 退出 ",font=("宋體",13),width=10,command=root.quit).grid(row=3,column=1,padx=15,pady=5)
#self.hi_there.pack()
text = scrolledtext.ScrolledText(frame,width=40,height=10,font=("宋體",15)) #創(chuàng)建可滾動的文本顯示框
text.grid(row=4,column=0,padx=40,pady=15,columnspan=2) #放置文本顯示框

def save():
 #print(text.get("0.0","end"))
 if not text.get("0.0","end").strip(): #獲取文本框內(nèi)容，從開始到結(jié)束
  tkinter.messagebox.showwarning('提示',"還沒有統(tǒng)計數(shù)據(jù)！")
  return None
 savecount=''
 nowtime=time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) #獲取當(dāng)前時間并格式化輸出
 savecount=nowtime+"\n"+text.get("0.0","end")
 with open("e:\\save.txt",'w') as fp:
  fp.write(savecount)
 tkinter.messagebox.showinfo('提示',"結(jié)果已保存")

def history():
 if os.path.exists("e:\\save.txt"):
  with open("e:\\save.txt",'r') as fp: 
   historytxt=fp.read()
 tkinter.messagebox.showinfo('歷史',historytxt)

def helpinf():
 tkinter.messagebox.showinfo('幫助',"""1.輸入您要統(tǒng)計的代碼文件路徑
2.輸入您要統(tǒng)計的代碼文件類型
3.保存功能只能保存上次查詢的結(jié)果
快捷鍵：
F1    查看幫助
ENTE   提交
Alt-F4   退出
Control-s 保存
            """)

def aboutinf():
 tkinter.messagebox.showinfo('關(guān)于',"您現(xiàn)在正在使用的是測試版本 by:田川")

menu=Menu(root)
submenu1=Menu(menu,tearoff=0)
menu.add_cascade(label='查看',menu=submenu1)
submenu1.add_command(label='歷史',command=history)
submenu1.add_command(label='保存',command=save)
submenu1.add_separator()
submenu1.add_command(label='退出', command=root.quit)
submenu2=Menu(menu,tearoff=0)
menu.add_cascade(label='幫助',menu=submenu2)
submenu2.add_command(label='查看幫助',command=helpinf)
submenu2.add_command(label='關(guān)于',command=aboutinf)
root.config(menu=menu)
#以上都是菜單欄的設(shè)置
"""
def caidan(root):
 menu=tkinter.Menu(root)
 submenu1=tkinter.Menu(menu,tearoff=0)
 menu.add_cascade(label='查看',menu=submenu1)
 submenu2 = tkinter.Menu(menu, tearoff=0)
 submenu2.add_command(label='復(fù)制')
 submenu2.add_command(label='粘貼')
 menu.add_cascade(label='編輯',menu=submenu2)
 submenu = tkinter.Menu(menu, tearoff=0)
 submenu.add_command(
 ='查看幫助')
 submenu.add_separator()
 submenu.add_command(label='關(guān)于計算機')
 menu.add_cascade(label='幫助',menu=submenu)
 root.config(menu=menu)

caidan(root)"""
root.mainloop() #執(zhí)行tk!