import tkinter as tk
|
from tkinter import messagebox, simpledialog, ttk
|
import pymysql
|
from Crypto.Cipher import AES
|
import base64
|
import json
|
import os
|
|
VUE_APP_EncryptPrivateValue = "12345678abcdefgh"
|
HISTORY_FILE = "username_history.json"
|
|
class PasswordQueryApp:
|
def __init__(self, root):
|
self.root = root
|
self.root.title("密码查询工具")
|
|
self.database_info_file = "database_info.json"
|
self.database_info = self.load_database_info()
|
|
self.create_maintenance_page()
|
self.create_query_page()
|
self.update_username_menu()
|
|
def load_database_info(self):
|
try:
|
with open(self.database_info_file, 'r') as f:
|
return json.load(f)
|
except FileNotFoundError:
|
return {}
|
|
def save_database_info(self):
|
with open(self.database_info_file, 'w') as f:
|
json.dump(self.database_info, f, indent=4)
|
|
def create_maintenance_page(self):
|
maintenance_frame = tk.LabelFrame(self.root, text="数据库信息维护")
|
maintenance_frame.pack(fill="both", expand="yes", padx=10, pady=10)
|
|
tk.Label(maintenance_frame, text="单位名称:").grid(row=0, column=0, sticky="w")
|
self.unit_name_entry = tk.Entry(maintenance_frame)
|
self.unit_name_entry.grid(row=0, column=1, padx=5, pady=5)
|
|
tk.Label(maintenance_frame, text="MySQL 主机:").grid(row=1, column=0, sticky="w")
|
self.host_entry = tk.Entry(maintenance_frame)
|
self.host_entry.grid(row=1, column=1, padx=5, pady=5)
|
|
tk.Label(maintenance_frame, text="MySQL 端口:").grid(row=2, column=0, sticky="w")
|
self.port_entry = tk.Entry(maintenance_frame)
|
self.port_entry.grid(row=2, column=1, padx=5, pady=5)
|
self.port_entry.insert(0, "3306")
|
|
tk.Label(maintenance_frame, text="MySQL 用户名:").grid(row=3, column=0, sticky="w")
|
self.user_entry = tk.Entry(maintenance_frame)
|
self.user_entry.grid(row=3, column=1, padx=5, pady=5)
|
|
tk.Label(maintenance_frame, text="MySQL 密码:").grid(row=4, column=0, sticky="w")
|
self.password_entry = tk.Entry(maintenance_frame, show="*")
|
self.password_entry.grid(row=4, column=1, padx=5, pady=5)
|
|
tk.Label(maintenance_frame, text="数据库名:").grid(row=5, column=0, sticky="w")
|
self.db_name_entry = tk.Entry(maintenance_frame)
|
self.db_name_entry.grid(row=5, column=1, padx=5, pady=5)
|
|
tk.Label(maintenance_frame, text="用户表名:").grid(row=6, column=0, sticky="w")
|
self.table_name_entry = tk.Entry(maintenance_frame)
|
self.table_name_entry.grid(row=6, column=1, padx=5, pady=5)
|
self.table_name_entry.insert(0, "sys_user")
|
|
tk.Label(maintenance_frame, text="用户名列名:").grid(row=7, column=0, sticky="w")
|
self.username_col_entry = tk.Entry(maintenance_frame)
|
self.username_col_entry.grid(row=7, column=1, padx=5, pady=5)
|
self.username_col_entry.insert(0, "login_name")
|
|
tk.Label(maintenance_frame, text="密码列名:").grid(row=8, column=0, sticky="w")
|
self.password_col_entry = tk.Entry(maintenance_frame)
|
self.password_col_entry.grid(row=8, column=1, padx=5, pady=5)
|
self.password_col_entry.insert(0, "plain_text")
|
|
save_button = tk.Button(maintenance_frame, text="保存信息", command=self.save_database_info_action)
|
save_button.grid(row=9, columnspan=2, pady=10)
|
|
def update_username_menu(self):
|
history = self.load_history()
|
menu = self.username_menu["menu"]
|
menu.delete(0, "end")
|
for name in history:
|
menu.add_command(label=name, command=lambda value=name: self.username_entry.delete(0, tk.END) or self.username_entry.insert(0, value))
|
|
def save_database_info_action(self):
|
unit_name = self.unit_name_entry.get().strip()
|
if unit_name == "":
|
messagebox.showerror("错误", "单位名称为必填项")
|
return
|
|
self.database_info[unit_name] = {
|
"host": self.host_entry.get().strip(),
|
"port": self.port_entry.get().strip(),
|
"user": self.user_entry.get().strip(),
|
"password": self.password_entry.get().strip() or "",
|
"db_name": self.db_name_entry.get().strip(),
|
"table_name": self.table_name_entry.get().strip(),
|
"username_col": self.username_col_entry.get().strip(),
|
"password_col": self.password_col_entry.get().strip()
|
}
|
|
self.save_database_info()
|
self.update_username_menu()
|
messagebox.showinfo("成功", "数据库信息保存成功")
|
|
def create_query_page(self):
|
query_frame = tk.LabelFrame(self.root, text="密码查询")
|
query_frame.pack(fill="both", expand="yes", padx=10, pady=10)
|
|
tk.Label(query_frame, text="选择单位名称:").grid(row=0, column=0, sticky="w")
|
self.unit_name_var = tk.StringVar(value="请选择单位")
|
self.unit_name_dropdown = tk.OptionMenu(query_frame, self.unit_name_var, *self.database_info.keys())
|
self.unit_name_dropdown.grid(row=0, column=1, padx=5, pady=5)
|
|
tk.Label(query_frame, text="输入用户名:").grid(row=1, column=0, sticky="w")
|
self.username_entry = tk.Entry(query_frame)
|
self.username_entry.grid(row=1, column=1, padx=5, pady=5)
|
|
query_button = tk.Button(query_frame, text="查询密码", command=self.query_password_action)
|
query_button.grid(row=2, columnspan=2, pady=10)
|
|
self.password_var = tk.StringVar()
|
tk.Label(query_frame, text="密码:").grid(row=3, column=0, sticky="w")
|
self.password_entry_login = tk.Entry(query_frame, textvariable=self.password_var, state="readonly")
|
self.password_entry_login.grid(row=3, column=1, padx=5, pady=5)
|
|
copy_button = tk.Button(query_frame, text="复制密码", command=self.copy_password)
|
copy_button.grid(row=4, column=0, pady=10)
|
|
reset_button = tk.Button(query_frame, text="重置", command=self.reset_fields)
|
reset_button.grid(row=4, column=1, pady=10)
|
|
# 用户名下拉菜单
|
tk.Label(query_frame, text="选择用户名:").grid(row=5, column=0, sticky="w")
|
self.username_menu_var = tk.StringVar()
|
self.username_menu_var.set("选择用户名")
|
self.username_menu = tk.OptionMenu(query_frame, self.username_menu_var, *self.load_history())
|
self.username_menu.grid(row=5, column=1, padx=5, pady=5)
|
|
def query_password_action(self):
|
unit_name = self.unit_name_var.get()
|
username = self.username_entry.get().strip()
|
|
if unit_name not in self.database_info:
|
messagebox.showerror("错误", "请选择有效的单位名称")
|
return
|
|
db_info = self.database_info[unit_name]
|
|
try:
|
conn = pymysql.connect(
|
host=db_info["host"],
|
port=int(db_info["port"]),
|
user=db_info["user"],
|
password=db_info["password"],
|
database=db_info["db_name"]
|
)
|
cursor = conn.cursor()
|
|
query = f"SELECT {db_info['username_col']}, {db_info['password_col']} FROM {db_info['table_name']} WHERE {db_info['username_col']} = %s"
|
cursor.execute(query, (username,))
|
result = cursor.fetchone()
|
|
if result:
|
username_db, password = result
|
if password is not None:
|
decrypted_password = self.decrypt_password(username_db, password)
|
self.password_var.set(decrypted_password)
|
else:
|
messagebox.showinfo("未找到", "未找到该用户名对应的密码")
|
else:
|
messagebox.showinfo("未找到", "未找到该用户名对应的密码")
|
|
cursor.close()
|
conn.close()
|
except pymysql.MySQLError as err:
|
messagebox.showerror("数据库错误", f"错误: {err}")
|
|
def decrypt_password(self, username, password):
|
try:
|
user_info = self.get_user_info(username)
|
if user_info:
|
encrypted_password, encryption_key = user_info
|
return self.decrypt(encrypted_password, encryption_key)
|
else:
|
return "用户信息不完整或不存在"
|
except Exception as e:
|
return f"解密错误: {e}"
|
|
def decrypt(self, encrypted_password, encryption_key):
|
try:
|
key = encryption_key.encode('utf-8')
|
iv = VUE_APP_EncryptPrivateValue.encode('utf-8')
|
encrypted_password = base64.b64decode(encrypted_password)
|
cipher = AES.new(key, AES.MODE_CBC, iv)
|
decrypted = cipher.decrypt(encrypted_password)
|
unpadded_text = decrypted.rstrip(b'\0').decode('utf-8')
|
return unpadded_text
|
except Exception as e:
|
return f"解密错误: {e}"
|
|
def get_user_info(self, username):
|
unit_name = self.unit_name_var.get()
|
if unit_name not in self.database_info:
|
return None
|
|
db_info = self.database_info[unit_name]
|
try:
|
conn = pymysql.connect(
|
host=db_info["host"],
|
port=int(db_info["port"]),
|
user=db_info["user"],
|
password=db_info["password"],
|
database=db_info["db_name"]
|
)
|
cursor = conn.cursor()
|
cursor.execute(f"SELECT {db_info['username_col']}, {db_info['password_col']} FROM {db_info['table_name']} WHERE {db_info['username_col']} = %s", (username,))
|
result = cursor.fetchone()
|
conn.close()
|
return result
|
except pymysql.MySQLError as err:
|
messagebox.showerror("数据库错误", f"错误: {err}")
|
return None
|
|
def copy_password(self):
|
decrypted_password = self.password_var.get()
|
if decrypted_password:
|
self.root.clipboard_clear()
|
self.root.clipboard_append(decrypted_password)
|
messagebox.showinfo("复制成功", "密码已复制到剪贴板")
|
|
def reset_fields(self):
|
self.username_entry.delete(0, tk.END)
|
self.password_var.set("")
|
self.unit_name_var.set("请选择单位")
|
|
def load_history(self):
|
if os.path.exists(HISTORY_FILE):
|
with open(HISTORY_FILE, 'r') as f:
|
history = json.load(f)
|
else:
|
history = []
|
return history
|
|
def save_history(self, history):
|
with open(HISTORY_FILE, 'w') as f:
|
json.dump(history, f)
|
|
if __name__ == "__main__":
|
root = tk.Tk()
|
app = PasswordQueryApp(root)
|
root.mainloop()
|
