#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("有效代碼統計工具") #界面的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('結尾雙引',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("請輸入您要統計的絕對路徑:")
#file_types=input("請輸入您要統計的檔案類型:")
#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("//") and flag == True:
flag = False
continue
else:
line_count += 1
return line_count,space_count,annotation_count
def show(): #當按鈕被點選,就調用這個方法
pathlist=e1.get() #調用get()方法得到在文本框中輸入的内容
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('提示',"請輸入要統計的類型!")
return None
#print(type(file_types),file_types)
if '\u4e00'<=file_types<='\u9fa5' or not file_types in file_types_list: #判斷檔案類型輸入的是否是中文
tkinter.messagebox.showwarning('錯誤',"輸入統計類型有誤!")
return None
text.delete(1.0,END) #删除顯示文本框中,原有的内容
for path in pathlist.split(";"):
path=path.strip()
codes,code_dict,space,annotation=code_count(path,file_types) #将函數傳回的結果指派給變量,友善輸出
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 有效代碼數%s\n"%(k,v[0])) #将檔案名和有效代碼輸出到文本框中
text.insert(INSERT,"整個%s下有%s類型檔案%d個,共有%d行有效代碼\n"%(path,file_types,len(code_dict),codes)) #将結果輸出到文本框中
text.insert(INSERT,"共有%d行注釋\n"%(annotation))
text.insert(INSERT,"共有%d行空行\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) #設定區域
label= Label(frame,text="路徑:",font=("宋體",15),fg="blue").grid(row=0,padx=10,pady=5,sticky=N) #建立标簽
label= Label(frame,text="類型:",font=("宋體",15),fg="blue").grid(row=1,padx=10,pady=5)
e1= Entry(frame,foreground = 'blue',font = ('Helvetica', '12')) #建立文本輸入框
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) #建立按鈕
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)) #建立可滾動的文本顯示框
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(): #擷取文本框内容,從開始到結束
tkinter.messagebox.showwarning('提示',"還沒有統計資料!")
return None
savecount=''
nowtime=time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) #擷取目前時間并格式化輸出
savecount=nowtime+"\n"+text.get("0.0","end")
with open("e:\\save.txt",'w') as fp:
fp.write(savecount)
tkinter.messagebox.showinfo('提示',"結果已儲存")
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.輸入您要統計的代碼檔案路徑
2.輸入您要統計的代碼檔案類型
3.儲存功能隻能儲存上次查詢的結果
快捷鍵:
F1 檢視幫助
ENTE 送出
Alt-F4 退出
Control-s 儲存
""")
def aboutinf():
tkinter.messagebox.showinfo('關于',"您現在正在使用的是測試版本 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='關于',command=aboutinf)
root.config(menu=menu)
#以上都是菜單欄的設定
"""
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='複制')
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='關于計算機')
menu.add_cascade(label='幫助',menu=submenu)
root.config(menu=menu)
caidan(root)"""
root.mainloop() #執行tk!