You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
455 lines
18 KiB
455 lines
18 KiB
import tkinter as tk
|
|
from tkinter import messagebox, ttk
|
|
import pymysql
|
|
import datetime
|
|
import time
|
|
import 评论总结
|
|
import 招生信息
|
|
import pyperclip
|
|
#import ttkbootstrap as ttk
|
|
from test import QuestionnaireGUI
|
|
|
|
class MasterChargeSystem(tk.Tk):
|
|
current_username=None
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.login_frame = None
|
|
self.register_frame = None
|
|
self.test_frame = None
|
|
self.detail_window = None
|
|
self.summary_window = None
|
|
self.create_login_window()
|
|
|
|
def register_database(self, username, password):
|
|
with pymysql.connect(host='localhost', user='root', password='123456', database='mynewdb') as connection:
|
|
cursor = connection.cursor()
|
|
sql = "INSERT INTO user1(username, password) VALUES (%s, %s)"
|
|
cursor.execute(sql, (username, password))
|
|
connection.commit()
|
|
|
|
def login(self):
|
|
global current_username
|
|
username = self.username_entry.get().strip()
|
|
password = self.password_entry.get().strip()
|
|
|
|
with pymysql.connect(host='localhost', user='root', password='123456', database='mynewdb') as connection:
|
|
cursor = connection.cursor()
|
|
current_username=username
|
|
sql = "SELECT password FROM user1 WHERE username = %s"
|
|
cursor.execute(sql, (username,))
|
|
result = cursor.fetchone()
|
|
|
|
if result:
|
|
stored_password = result[0]
|
|
if stored_password == password:
|
|
messagebox.showinfo("登录", "登陆成功")
|
|
self.show_main_window()
|
|
else:
|
|
messagebox.showerror("登录", "用户名或密码错误")
|
|
else:
|
|
messagebox.showerror("登录", "用户名不存在")
|
|
|
|
def register(self):
|
|
username = self.username_entry.get().strip()
|
|
password = self.password_entry.get().strip()
|
|
confirm_password = self.confirm_password_entry.get().strip()
|
|
|
|
if username and password:
|
|
if password == confirm_password:
|
|
self.register_database(username, password)
|
|
messagebox.showinfo("注册", "注册成功")
|
|
self.create_login_window()
|
|
else:
|
|
messagebox.showerror("注册", "两次输入的密码不匹配")
|
|
else:
|
|
messagebox.showerror("注册", "用户名或密码为空")
|
|
|
|
def show_main_window(self):
|
|
if self.login_frame:
|
|
self.login_frame.pack_forget()
|
|
if self.detail_window:
|
|
self.detail_window.pack_forget()
|
|
if self.test_frame:
|
|
self.test_frame.pack_forget()
|
|
|
|
self.title("考研信息系统")
|
|
self.attributes('-fullscreen',True)
|
|
|
|
self.main_frame = tk.Frame(self)
|
|
self.main_frame.pack(fill=tk.BOTH, expand=True, padx=50, pady=50)
|
|
|
|
main_label = tk.Label(self.main_frame, text="欢迎来到考研信息系统！")
|
|
main_label.pack(pady=10)
|
|
|
|
|
|
try:
|
|
with pymysql.connect(host='localhost', user='root', password='123456', database='mynewdb') as connection:
|
|
cursor = connection.cursor()
|
|
|
|
# 添加搜索框
|
|
search_frame = tk.Frame(self.main_frame)
|
|
search_frame.pack(pady=10)
|
|
|
|
search_label = tk.Label(search_frame, text="搜索：地区")
|
|
search_label.pack(side=tk.LEFT)
|
|
|
|
cursor.execute("select distinct 省份 from universities_beijing")
|
|
location = cursor.fetchall()
|
|
|
|
self.search2_entry = ttk.Combobox(search_frame, values=location)
|
|
self.search2_entry.pack(side=tk.LEFT, padx=5)
|
|
|
|
cursor.execute("select distinct 招生年份 from universities_beijing")
|
|
year = cursor.fetchall()
|
|
|
|
search_label = tk.Label(search_frame, text=" 年份")
|
|
search_label.pack(side=tk.LEFT)
|
|
|
|
self.search3_entry = ttk.Combobox(search_frame, values=year)
|
|
self.search3_entry.pack(side=tk.LEFT, padx=5)
|
|
|
|
cursor.execute("select distinct 学校类型 from universities_beijing")
|
|
time = cursor.fetchall()
|
|
|
|
search_label = tk.Label(search_frame, text=" 学校类型")
|
|
search_label.pack(side=tk.LEFT)
|
|
|
|
self.search4_entry = ttk.Combobox(search_frame, values=time)
|
|
self.search4_entry.pack(side=tk.LEFT, padx=5)
|
|
|
|
search_label = tk.Label(search_frame, text=" 院校")
|
|
search_label.pack(side=tk.LEFT)
|
|
|
|
self.search1_entry = tk.Entry(search_frame)
|
|
self.search1_entry.pack(side=tk.LEFT, padx=5)
|
|
|
|
search_label = tk.Label(search_frame, text=" 评论总结")
|
|
search_label.pack(side=tk.LEFT)
|
|
|
|
self.search5_entry = tk.Entry(search_frame)
|
|
self.search5_entry.pack(side=tk.LEFT, padx=5)
|
|
|
|
affirm_button = tk.Button(search_frame, text="确认", command=self.search_data)
|
|
affirm_button.pack(side=tk.RIGHT)
|
|
|
|
|
|
# 创建一个用于 Treeview 和滚动条的框架
|
|
tree_frame = tk.Frame(self.main_frame)
|
|
tree_frame.pack(fill=tk.BOTH, expand=True)
|
|
|
|
self.tree = ttk.Treeview(tree_frame)
|
|
|
|
self.vertical_scrollbar = ttk.Scrollbar(tree_frame, orient=tk.VERTICAL, command=self.tree.yview)
|
|
|
|
self.tree.configure(yscrollcommand=self.vertical_scrollbar.set)
|
|
|
|
# 获取列名
|
|
cursor.execute("SELECT * FROM universities_beijing LIMIT 0") # 替换为您的表名
|
|
self.columns = [description[0] for description in cursor.description]
|
|
|
|
# 获取数据
|
|
cursor.execute("SELECT * FROM universities_beijing") # 获取所有数据
|
|
self.data = cursor.fetchall()
|
|
|
|
# 设置列
|
|
self.tree['columns'] = self.columns
|
|
self.tree.column("#0", width=0, stretch=tk.NO) # 隐藏第一个空列
|
|
|
|
# 动态设置列和列名
|
|
for column in self.columns:
|
|
self.tree.column(column, anchor=tk.W, width=50, stretch=tk.YES) # 设置为自适应宽度
|
|
self.tree.heading(column, text=column, anchor=tk.W)
|
|
|
|
# 将 treeview 和滚动条放置到 tree_frame
|
|
self.tree.pack(side=tk.LEFT, fill=tk.BOTH, expand=True)
|
|
self.vertical_scrollbar.pack(side=tk.RIGHT, fill=tk.Y)
|
|
|
|
# 插入数据
|
|
for row in self.data:
|
|
self.tree.insert("", tk.END, values=row)
|
|
|
|
# 绑定点击事件
|
|
self.tree.bind("<Double-1>", self.show_university_details)
|
|
|
|
except pymysql.Error as err:
|
|
messagebox.showerror("数据库错误", str(err))
|
|
|
|
update_button=tk.Button(self.main_frame,text="更新数据",command=self.update)
|
|
update_button.pack(pady=10,padx=10)
|
|
|
|
exit_button = tk.Button(self.main_frame, text="退出", command=self.quit)
|
|
exit_button.pack(pady=10,padx=20)
|
|
|
|
comment_button = tk.Button(self.main_frame, text="自测", command=self.testYouself)
|
|
comment_button.pack(pady=10,padx=30)
|
|
|
|
def update(self):
|
|
response = messagebox.askyesno("确认", "您确定要更新吗？")
|
|
if response == True:
|
|
with pymysql.connect(host='localhost', user='root', password='123456', database='mynewdb') as connection:
|
|
cursor = connection.cursor()
|
|
招生信息.star()
|
|
sql="""LOAD DATA INFILE "D:\\software\\test2\\pythonProject(1)\\student1.csv"
|
|
INTO TABLE mynewdb.universities_beijing
|
|
FIELDS TERMINATED BY ','
|
|
ENCLOSED BY '"'
|
|
LINES TERMINATED BY '\r\n'
|
|
IGNORE 1 ROWS;"""
|
|
cursor.execute(sql)
|
|
|
|
else:
|
|
messagebox.showinfo("更新取消","已取消")
|
|
|
|
|
|
|
|
def show_university_details(self, event):
|
|
selected_item = self.tree.selection()[0] # 获取选中的行
|
|
university_data = self.tree.item(selected_item, 'values') # 获取行数据
|
|
|
|
with pymysql.connect(host='localhost', user='root', password='123456', database='mynewdb')as connection:
|
|
cursor= connection.cursor()
|
|
|
|
|
|
if university_data:
|
|
self.detail_window = tk.Toplevel(self) # 创建新窗口
|
|
self.detail_window.title("大学详细信息")
|
|
self.detail_window.attributes('-fullscreen', True)
|
|
|
|
cursor.execute("SELECT * FROM universities_beijing LIMIT 0") # 替换为您的表名
|
|
columns = [desc[0] for desc in cursor.description]
|
|
tree_detail = ttk.Treeview(self.detail_window, columns=[1,2], show='headings')
|
|
# 显示大学的其他信息
|
|
i=0
|
|
while i < len(columns):
|
|
for detail in university_data:
|
|
tree_detail.insert("","end",values=(columns[i]+":",detail))
|
|
i+=1
|
|
tree_detail.pack(expand=True, fill='both')
|
|
#添加评论搜索框
|
|
search_label = tk.Label(self.detail_window, text="搜索评论内容:")
|
|
search_label.pack(side=tk.LEFT)
|
|
|
|
self.search1_comment_entry = tk.Entry(self.detail_window)
|
|
self.search1_comment_entry.pack(side=tk.LEFT, padx=5)
|
|
|
|
affirm_button = tk.Button(self.detail_window, text="确认", command=self.search_comment)
|
|
affirm_button.pack(side=tk.LEFT)
|
|
# 添加评论框
|
|
self.tree_comment=ttk.Treeview(self.detail_window,columns=[1,2,3,4],show='headings')
|
|
self.tree_comment.column('1',width=300)
|
|
self.tree_comment.column('2', width=300)
|
|
self.tree_comment.column('3', width=800)
|
|
self.tree_comment.column('4',width=200)
|
|
self.tree_comment.pack(expand=True,fill='both')
|
|
sql="select * FROM comments where university_name=%s Order by comment_time desc"
|
|
cursor.execute(sql,(university_data[0],))
|
|
self.comment_result = cursor.fetchall()
|
|
for row in self.comment_result:
|
|
self.tree_comment.insert("", tk.END, values=row)
|
|
comment_label = tk.Label(self.detail_window, text="添加评论:")
|
|
comment_label.pack(pady=10)
|
|
|
|
self.comment_entry = tk.Entry(self.detail_window)
|
|
self.comment_entry.pack(pady=5)
|
|
|
|
submit_comment_button = tk.Button(self.detail_window, text="提交评论",command=lambda: self.submit_comment(university_data[0]))
|
|
submit_comment_button.pack(pady=10)
|
|
|
|
delete_button = tk.Button(self.detail_window,text="删除评论",command=self.delete_comment)
|
|
delete_button.pack(pady=10)
|
|
|
|
summary_button = tk.Button(self.detail_window, text="总结词条",command=self.show_summary(university_data[0]))
|
|
summary_button.pack(pady=10)
|
|
|
|
back_button = tk.Button(self.detail_window, text="返回", command=self.detail_window.destroy)
|
|
back_button.pack(pady=10)
|
|
|
|
cursor.close()
|
|
|
|
def search_comment(self):
|
|
search1_term = self.search1_comment_entry.get()
|
|
|
|
|
|
# 清空当前Treeview
|
|
self.tree_comment.delete(*self.tree_comment.get_children())
|
|
|
|
# 重新插入符合搜索条件的数据
|
|
for row in self.comment_result:
|
|
if any((search1_term in str(value).lower() for value in row)):
|
|
self.tree_comment.insert("", tk.END, values=row)
|
|
|
|
|
|
def show_summary(self,university_name):
|
|
if self.detail_window:
|
|
self.detail_window.destroy
|
|
|
|
self.tree_summary = tk.Toplevel(self.detail_window)
|
|
|
|
|
|
with pymysql.connect(host='localhost', user='root', password='123456', database='mynewdb') as connection:
|
|
cursor = connection.cursor()
|
|
sql="select comment from comments where university_name=%s"
|
|
cursor.execute(sql,(university_name,))
|
|
result=cursor.fetchall()
|
|
|
|
file1=open("评论内容.txt","w+",encoding="utf-8")
|
|
for line in result:
|
|
file1.writelines(line)
|
|
file1.write("\n")
|
|
file1.close()
|
|
|
|
评论总结.main()
|
|
|
|
file2 = open(r"keys_TFIDF.csv","r",encoding="utf-8")
|
|
datas = file2.readlines()
|
|
datas = str(datas)
|
|
wenben_text_area = tk.Text(self.tree_summary, height=5, width=73)
|
|
wenben_text_area.grid(row=3, columnspan=3, sticky="w")
|
|
original_text = datas.replace('\n', '')
|
|
for data in original_text:
|
|
wenben_text_area.insert(tk.END, data)
|
|
file2.close()
|
|
|
|
|
|
|
|
def submit_comment(self, university_name):
|
|
global current_username
|
|
comment = self.comment_entry.get()
|
|
if comment:
|
|
# Store comment in the database (add your SQL query here)
|
|
with pymysql.connect(host='localhost', user='root', password='123456', database='mynewdb') as connection:
|
|
cursor = connection.cursor()
|
|
comment_time=datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
|
sql = "INSERT INTO comments (username,university_name, comment,comment_time) VALUES (%s,%s, %s, %s)"
|
|
cursor.execute(sql, (current_username,university_name, comment,comment_time))
|
|
connection.commit()
|
|
|
|
messagebox.showinfo("评论", "评论提交成功")
|
|
self.comment_entry.delete(0, tk.END) # 清空输入框
|
|
else:
|
|
messagebox.showerror("评论", "评论不能为空")
|
|
|
|
def delete_comment(self):
|
|
with pymysql.connect(host='localhost', user='root', password='123456', database='mynewdb') as connection:
|
|
cursor = connection.cursor()
|
|
sql="""
|
|
"""
|
|
|
|
def testYouself(self):
|
|
if self.main_frame:
|
|
self.main_frame.pack_forget()
|
|
|
|
self.title("自测界面")
|
|
sw = self.winfo_screenwidth()
|
|
sh = self.winfo_screenheight()
|
|
Width = 400
|
|
Height = 300
|
|
cen_x = (sw - Width) / 2
|
|
cen_y = (sh - Height) / 2
|
|
self.geometry('%dx%d+%d+%d' % (Width, Height, cen_x, cen_y))
|
|
self.test_frame = tk.Frame(self)
|
|
self.test_frame.pack(padx=20, pady=20)
|
|
app = QuestionnaireGUI(self.test_frame)
|
|
app.generate_next_question()
|
|
|
|
self.back_button = tk.Button(self.test_frame, text="退出", command=self.show_main_window)
|
|
self.back_button.pack(pady=10)
|
|
|
|
|
|
def search_data(self):
|
|
search1_term = self.search1_entry.get()
|
|
search2_term = self.search2_entry.get()
|
|
search3_term = self.search3_entry.get()
|
|
search4_term = self.search4_entry.get()
|
|
search5_term = self.search5_entry.get()
|
|
|
|
# 清空当前Treeview
|
|
self.tree.delete(*self.tree.get_children())
|
|
|
|
# 重新插入符合搜索条件的数据
|
|
for row in self.data:
|
|
if any((search1_term in str(value).lower() for value in row)):
|
|
if any((search2_term in str(value).lower() for value in row)):
|
|
if any((search3_term in str(value).lower() for value in row)):
|
|
if any((search4_term in str(value).lower() for value in row)):
|
|
if any((search5_term in str(value).lower() for value in row)):
|
|
self.tree.insert("", tk.END, values=row)
|
|
|
|
|
|
def show_register_window(self):
|
|
if self.login_frame:
|
|
self.login_frame.pack_forget()
|
|
self.title("用户注册界面")
|
|
sw = self.winfo_screenwidth()
|
|
sh = self.winfo_screenheight()
|
|
Width = 400
|
|
Height = 300
|
|
cen_x = (sw - Width) / 2
|
|
cen_y = (sh - Height) / 2
|
|
self.geometry('%dx%d+%d+%d' % (Width, Height, cen_x, cen_y))
|
|
|
|
self.register_frame = tk.Frame(self)
|
|
self.register_frame.pack(padx=20, pady=20)
|
|
|
|
self.username_label = tk.Label(self.register_frame, text="用户名:")
|
|
self.username_label.grid(row=0, column=0)
|
|
self.username_entry = tk.Entry(self.register_frame)
|
|
self.username_entry.grid(row=0, column=1)
|
|
|
|
self.password_label = tk.Label(self.register_frame, text="密码:")
|
|
self.password_label.grid(row=1, column=0)
|
|
self.password_entry = tk.Entry(self.register_frame, show="*")
|
|
self.password_entry.grid(row=1, column=1)
|
|
|
|
self.confirm_password_label = tk.Label(self.register_frame, text="再次输入密码:")
|
|
self.confirm_password_label.grid(row=2, column=0)
|
|
self.confirm_password_entry = tk.Entry(self.register_frame, show="*")
|
|
self.confirm_password_entry.grid(row=2, column=1)
|
|
|
|
self.register_button = tk.Button(self.register_frame, text="注册", command=self.register)
|
|
self.register_button.grid(row=3, column=0)
|
|
|
|
self.back_button = tk.Button(self.register_frame, text="返回", command=self.create_login_window)
|
|
self.back_button.grid(row=3, column=1)
|
|
|
|
def create_login_window(self):
|
|
if self.register_frame:
|
|
self.register_frame.pack_forget()
|
|
self.title("登录界面")
|
|
sw = self.winfo_screenwidth()
|
|
sh = self.winfo_screenheight()
|
|
Width = 400
|
|
Height = 300
|
|
cen_x = (sw - Width) / 2
|
|
cen_y = (sh - Height) / 2
|
|
self.geometry('%dx%d+%d+%d' % (Width, Height, cen_x, cen_y))
|
|
|
|
self.login_frame = tk.Frame(self)
|
|
self.login_frame.pack(padx=20, pady=20)
|
|
|
|
self.username_label = tk.Label(self.login_frame, text="用户名:")
|
|
self.username_label.grid(row=0, column=0)
|
|
self.username_entry = tk.Entry(self.login_frame)
|
|
self.username_entry.grid(row=0, column=1)
|
|
|
|
self.password_label = tk.Label(self.login_frame, text="密码:")
|
|
self.password_label.grid(row=1, column=0)
|
|
self.password_entry = tk.Entry(self.login_frame, show="*")
|
|
self.password_entry.grid(row=1, column=1)
|
|
|
|
self.login_button = tk.Button(self.login_frame, text="登录", command=self.login)
|
|
self.login_button.grid(row=2, column=0)
|
|
|
|
self.back_button = tk.Button(self.login_frame, text="退出", command=self.quit)
|
|
self.back_button.grid(row=2, column=1)
|
|
|
|
self.register_button = tk.Button(self.login_frame, text="注册", command=self.show_register_window)
|
|
self.register_button.grid(row=3, column=0)
|
|
|
|
self.deiconify()
|
|
|
|
|
|
# Start the application
|
|
if __name__ == "__main__":
|
|
app = MasterChargeSystem()
|
|
app.mainloop() |