import tkinter as tk
from tkinter import messagebox, END, RIDGE
import jieba

root = tk.Tk()
root.title("單詞檢索統(tǒng)計(jì)系統(tǒng)")
root.resizable(False, False)
root.geometry('450x400')
root.configure(bg='white')


# 新建文件
def fun1():
    content1 = text1.get()
    content2 = text2.get('1.0', END)
    if content1 == '':
        messagebox.showerror(title='Error', message='文件名不為空!')
    else:
        file = open(r'D:\圖庫\183\pythonProject\com\mao\\' + content1 + '.txt', 'w', encoding='GBK')
        file.writelines(content2)
        file.write('\n')
        messagebox.showinfo(title='Success', message='新建成功!')
        file.close()


# 單詞的統(tǒng)計(jì)
def fun2():
    content1 = text1.get()
    content3 = text3.get()
    if content1 == '':
        messagebox.showerror(title='Error', message='文件名不為空!')
    else:
        try:
            file = open(r'D:\圖庫\183\pythonProject\com\mao\\' + content1 + '.txt', 'r')
            if content3 != '':
                count = 0
                # 計(jì)數(shù)器初始化為0
                for line in file.readlines():
                    line = line.strip()
                    # 依次讀取每行,  去掉每行頭尾空白
                    count += line.count(content3)
                    # 逐行累加計(jì)數(shù)
                if count == 0:
                    messagebox.showerror(title='Error', message='沒有此單詞!')
                else:
                    text4.insert(tk.END, '共有 ' + str(count) + ' 個 ' + content3 + '\n')
                    with open(r"D:\圖庫\183\pythonProject\com\mao\text.txt", "a") as output:
                        output.write('共有 ' + str(count) + ' 個 ' + content3 + '\n')
                file.close()
            else:
                messagebox.showerror(title='Error', message='查詢目標(biāo)單詞不為空!')
        except Exception as e:
            messagebox.showerror(title='Error', message=e)


# 單詞的定位
def fun3():
    content1 = text1.get()
    content3 = text3.get()
    if content1 == '':
        messagebox.showerror(title='Error', message='文件名不為空!')
    else:
        try:
            file = open(r'D:\圖庫\183\pythonProject\com\mao\\' + content1 + '.txt', 'r')
            row = 1
            # 行初始化為1
            for line in file.readlines():
                # 檢索每一行
                if content3 == '':
                    messagebox.showerror(title='Error', message='查詢目標(biāo)單詞不為空!')
                    break
                else:
                    if line.count(content3):
                        # 每一行目標(biāo)單詞出現(xiàn)的次數(shù)
                        text4.insert(tk.END, '\n第' + str(row) + '行，' + str(line.count(content3)) + '次')
                        with open(r"D:\圖庫\183\pythonProject\com\mao\text.txt", "a") as output:
                            output.write('\n第' + str(row) + '行，' + str(line.count(content3)) + '次')
                        index_list = []
                        # 建立索引列表
                        index = line.find(content3)
                        # 找到單詞所在的位置：索引號
                        while index != -1:
                            # 判斷索引是否結(jié)束，并將查找到的索引號添加到索引列表中
                            index_list.append(index)
                            index = line.find(content3, index + 1)
                            # 下標(biāo)下一位開始，找到word之后添加到列表中
                        text4.insert(tk.END, '\n相應(yīng)位置在：')
                        for index in index_list:
                            # 遍歷索引列表并輸出
                            text4.insert(tk.END, str(index + 1) + ' ')
                        text4.insert(tk.END, '\n')
                        row += 1
                    else:
                        # 無法查詢到單詞,沒有目標(biāo)單詞的行輸出此行沒有找到，進(jìn)入下一行繼續(xù)查找
                        row += 1
                        continue
                fun2()
        except Exception as e:
            messagebox.showerror(title='Error', message=e)


# 分詞統(tǒng)計(jì)
def sign4():
    content1 = text1.get()

    def read(path):
        if path == '':
            messagebox.showerror(title='Error', message='文件名不為空!')
        else:
            try:
                text = open(r'D:\圖庫\183\pythonProject\com\mao\\' + path + '.txt', 'r')
                content = text.read()
                words = jieba.lcut(content)
                counts = {}
                for word in words:
                    if len(word) == 1:
                        continue
                    else:
                        counts[word] = counts.get(word, 0) + 1
                items = list(counts.items())
                items.sort(key=lambda x: x[1], reverse=True)
                counts = dict(items)
                text4.insert(tk.END, counts)
                with open(r"D:\圖庫\183\pythonProject\com\mao\text.txt", "a") as output:
                    output.write(str(counts))
                text.close()
            except Exception as e:
                messagebox.showerror(title='Error', message=e)

    read(content1)


# 頁面布局
button1 = tk.Button(root, text='新建文件', bg='pink', font=('微軟雅黑', 15), relief=RIDGE, command=fun1)
button1.place(x=0, y=0, width=100, height=80)
label1 = tk.Label(root, text='文件名:', font=('微軟雅黑', 15), )
label1.grid(padx=100)
text1 = tk.Entry(root, font=('MingLiU', 13))
text1.place(x=200, width=200, height=40)

label2 = tk.Label(root, text='文件內(nèi)容:', font=('微軟雅黑', 15))
label2.grid(pady=25)
text2 = tk.Text(root)
text2.place(x=200, y=45, width=200, height=40)

button2 = tk.Button(root, text='單詞計(jì)數(shù)', bg='pink', font=('微軟雅黑', 15), relief=RIDGE, command=fun2)
button2.place(x=0, y=80, width=100, height=80)

button3 = tk.Button(root, text='單詞定位', bg='pink', font=('微軟雅黑', 15), relief=RIDGE, command=fun3)
button3.place(x=0, y=160, width=100, height=80)

button4 = tk.Button(root, text='分詞統(tǒng)計(jì)', bg='pink', font=('微軟雅黑', 15), relief=RIDGE, command=sign4)
button4.place(x=0, y=240, width=100, height=80)

label3 = tk.Label(root, text='查詢目標(biāo):', font=('微軟雅黑', 15))
label3.grid(padx=100)
text3 = tk.Entry(root, font=('MingLiU', 13))
text3.place(x=200, y=90, width=200, height=58)

label4 = tk.Label(root, text='查詢結(jié)果:', font=('微軟雅黑', 15))
label4.grid(pady=75)
text4 = tk.Text(root, font=('楷體', 13))
text4.place(x=200, y=160, width=200, height=200)

button5 = tk.Button(root, text='退出程序', bg='pink', font=('微軟雅黑', 15), relief=RIDGE, command=root.destroy)
button5.place(x=0, y=320, width=100, height=80)
root.mainloop()