769 lines
32 KiB
Python
769 lines
32 KiB
Python
from __future__ import annotations
|
||
|
||
import tkinter as tk
|
||
from tkinter import ttk, messagebox, scrolledtext
|
||
import threading
|
||
import sys
|
||
import os
|
||
|
||
# 添加当前目录到路径
|
||
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
||
|
||
from config_loader import get_config, Config
|
||
from step1 import Step1FeastCoding
|
||
from step2 import Step2Converter
|
||
from editor_gui import open_editor
|
||
from markdown_editor import ThemeManager
|
||
from db_manager import get_db_manager, reset_db_manager, ProjectOrder
|
||
|
||
|
||
class YidaimaGUI:
|
||
def __init__(self, root):
|
||
self.root = root
|
||
self.root.title("自动化工具")
|
||
self.root.geometry("1200x900")
|
||
self.root.resizable(True, True)
|
||
|
||
# 设置样式
|
||
self.style = ttk.Style()
|
||
self.style.configure("TButton", padding=6, relief="flat", background="#007aff")
|
||
self.style.configure("TLabel", padding=6, font=("Microsoft YaHei", 10))
|
||
self.style.configure("TEntry", padding=6)
|
||
|
||
# 设置 Treeview 样式，增加行高
|
||
self.style.configure("Treeview", rowheight=30)
|
||
|
||
# 加载配置
|
||
self.config = get_config()
|
||
|
||
# 初始化管理器
|
||
self.theme_manager = ThemeManager()
|
||
self._init_db_manager()
|
||
|
||
# 运行状态
|
||
self.is_running = False
|
||
|
||
# 文章管理分页状态
|
||
self.current_page = 1
|
||
self.page_size = 20
|
||
self.total_items = 0
|
||
self.search_keyword = ""
|
||
|
||
# 创建界面
|
||
self.create_widgets()
|
||
|
||
def _init_db_manager(self):
|
||
"""初始化数据库管理器"""
|
||
try:
|
||
db_config = self.config.get("database", {})
|
||
self.db_manager = get_db_manager(
|
||
host=db_config.get("host", "localhost"),
|
||
database=db_config.get("database", "test"),
|
||
user=db_config.get("user", "root"),
|
||
password=db_config.get("password", "123456"),
|
||
port=db_config.get("port", 3306)
|
||
)
|
||
except Exception as e:
|
||
print(f"[GUI] 初始化数据库管理器失败: {e}")
|
||
self.db_manager = None
|
||
|
||
def create_widgets(self):
|
||
"""创建主界面"""
|
||
# 主框架
|
||
main_frame = ttk.Frame(self.root, padding="10")
|
||
main_frame.pack(fill=tk.BOTH, expand=True)
|
||
|
||
# 创建 Notebook (Tab 控件)
|
||
self.notebook = ttk.Notebook(main_frame)
|
||
self.notebook.pack(fill=tk.BOTH, expand=True)
|
||
|
||
# === Tab 1: 发布微信公众号 ===
|
||
self.tab_publish = ttk.Frame(self.notebook, padding="10")
|
||
self.notebook.add(self.tab_publish, text="发布微信公众号")
|
||
self._create_publish_tab()
|
||
|
||
# === Tab 2: 文章发布管理 ===
|
||
self.tab_manage = ttk.Frame(self.notebook, padding="10")
|
||
self.notebook.add(self.tab_manage, text="文章发布管理")
|
||
self._create_manage_tab()
|
||
|
||
# === Tab 4: 参数设置 ===
|
||
self.tab_settings = ttk.Frame(self.notebook, padding="10")
|
||
self.notebook.add(self.tab_settings, text="参数设置")
|
||
self._create_settings_tab()
|
||
|
||
def _create_publish_tab(self):
|
||
"""创建发布微信公众号 Tab"""
|
||
# 配置网格
|
||
self.tab_publish.columnconfigure(0, weight=1)
|
||
self.tab_publish.rowconfigure(2, weight=1)
|
||
|
||
# === 配置区域 ===
|
||
config_frame = ttk.LabelFrame(self.tab_publish, text="微信公众号配置", padding="10")
|
||
config_frame.grid(row=0, column=0, sticky=(tk.W, tk.E), pady=(0, 10))
|
||
config_frame.columnconfigure(1, weight=1)
|
||
|
||
# 项目名称
|
||
ttk.Label(config_frame, text="项目名称:").grid(row=0, column=0, sticky=tk.W)
|
||
self.project_name_var = tk.StringVar(
|
||
value=self.config.get("step1.project_name", "【A173】基于Springboot + vue3实现的学生交流互助平台")
|
||
)
|
||
self.project_name_entry = ttk.Entry(config_frame, textvariable=self.project_name_var, width=50)
|
||
self.project_name_entry.grid(row=0, column=1, sticky=(tk.W, tk.E), padx=(10, 0))
|
||
|
||
# CSS 样式方案
|
||
ttk.Label(config_frame, text="CSS 样式方案:").grid(row=1, column=0, sticky=tk.W, pady=(10, 0))
|
||
self.css_scheme_var = tk.StringVar()
|
||
scheme_frame = ttk.Frame(config_frame)
|
||
scheme_frame.grid(row=1, column=1, sticky=(tk.W, tk.E), padx=(10, 0), pady=(10, 0))
|
||
scheme_frame.columnconfigure(0, weight=1)
|
||
|
||
self.css_scheme_combo = ttk.Combobox(
|
||
scheme_frame,
|
||
values=self._get_css_scheme_names(),
|
||
textvariable=self.css_scheme_var,
|
||
state="readonly",
|
||
width=50
|
||
)
|
||
self.css_scheme_combo.grid(row=0, column=0, sticky=(tk.W, tk.E))
|
||
self.css_scheme_combo.set("默认")
|
||
|
||
# 刷新按钮
|
||
ttk.Button(scheme_frame, text="刷新", command=self.refresh_css_schemes, width=8).grid(row=0, column=1, padx=(5, 0))
|
||
|
||
# === 操作按钮区域 ===
|
||
button_frame = ttk.Frame(self.tab_publish)
|
||
button_frame.grid(row=1, column=0, pady=(10, 10))
|
||
|
||
# 发布微信公众号按钮
|
||
self.publish_btn = ttk.Button(
|
||
button_frame,
|
||
text="发布微信公众号",
|
||
command=self.run_full_flow,
|
||
width=25
|
||
)
|
||
self.publish_btn.pack(side=tk.LEFT)
|
||
|
||
# Markdown 编辑器按钮
|
||
ttk.Button(
|
||
button_frame,
|
||
text="Markdown 编辑器",
|
||
command=self._open_markdown_editor,
|
||
width=15
|
||
).pack(side=tk.LEFT, padx=(10, 0))
|
||
|
||
|
||
|
||
# === 日志输出区域 ===
|
||
log_frame = ttk.LabelFrame(self.tab_publish, text="运行日志", padding="10")
|
||
log_frame.grid(row=2, column=0, sticky=(tk.W, tk.E, tk.N, tk.S), pady=(10, 0))
|
||
log_frame.columnconfigure(0, weight=1)
|
||
log_frame.rowconfigure(0, weight=1)
|
||
|
||
self.log_text = scrolledtext.ScrolledText(
|
||
log_frame,
|
||
wrap=tk.WORD,
|
||
width=80,
|
||
height=15,
|
||
font=("Consolas", 9)
|
||
)
|
||
self.log_text.grid(row=0, column=0, sticky=(tk.W, tk.E, tk.N, tk.S))
|
||
|
||
# 进度条
|
||
self.progress = ttk.Progressbar(
|
||
self.tab_publish,
|
||
mode='indeterminate',
|
||
length=400
|
||
)
|
||
self.progress.grid(row=3, column=0, pady=(10, 0), sticky=(tk.W, tk.E))
|
||
|
||
# 状态标签
|
||
self.status_var = tk.StringVar(value="就绪")
|
||
self.status_label = ttk.Label(
|
||
self.tab_publish,
|
||
textvariable=self.status_var,
|
||
foreground="#666"
|
||
)
|
||
self.status_label.grid(row=4, column=0, pady=(10, 0))
|
||
|
||
def _create_manage_tab(self):
|
||
"""创建文章发布管理 Tab"""
|
||
# 配置网格
|
||
self.tab_manage.columnconfigure(0, weight=1)
|
||
self.tab_manage.rowconfigure(2, weight=1)
|
||
|
||
# === 搜索区域 ===
|
||
search_frame = ttk.Frame(self.tab_manage)
|
||
search_frame.grid(row=0, column=0, sticky=(tk.W, tk.E), pady=(0, 10))
|
||
search_frame.columnconfigure(1, weight=1)
|
||
|
||
ttk.Label(search_frame, text="搜索名称:").grid(row=0, column=0, sticky=tk.W)
|
||
self.search_var = tk.StringVar()
|
||
self.search_entry = ttk.Entry(search_frame, textvariable=self.search_var, width=40)
|
||
self.search_entry.grid(row=0, column=1, sticky=(tk.W, tk.E), padx=(10, 0))
|
||
|
||
ttk.Button(search_frame, text="搜索", command=self._search_projects, width=10).grid(row=0, column=2, padx=(10, 0))
|
||
ttk.Button(search_frame, text="刷新", command=self._refresh_projects, width=10).grid(row=0, column=3, padx=(10, 0))
|
||
|
||
# === 数据表格区域 ===
|
||
table_frame = ttk.LabelFrame(self.tab_manage, text="项目列表", padding="5")
|
||
table_frame.grid(row=1, column=0, rowspan=2, sticky=(tk.W, tk.E, tk.N, tk.S), pady=(0, 10))
|
||
table_frame.columnconfigure(0, weight=1)
|
||
table_frame.rowconfigure(0, weight=1)
|
||
|
||
# 创建 Treeview
|
||
columns = ("name", "name2", "paths", "zlzt", "sfxxm", "bianhao", "actions")
|
||
self.tree = ttk.Treeview(
|
||
table_frame,
|
||
columns=columns,
|
||
show="headings",
|
||
height=35
|
||
)
|
||
|
||
# 设置列标题
|
||
self.tree.heading("name", text="项目名称")
|
||
self.tree.heading("name2", text="名称2")
|
||
self.tree.heading("paths", text="路径")
|
||
self.tree.heading("zlzt", text="整理状态")
|
||
self.tree.heading("sfxxm", text="是否新项目")
|
||
self.tree.heading("bianhao", text="编号")
|
||
self.tree.heading("actions", text="操作")
|
||
|
||
# 设置列宽
|
||
self.tree.column("name", width=150, anchor=tk.W)
|
||
self.tree.column("name2", width=100, anchor=tk.W)
|
||
self.tree.column("paths", width=120, anchor=tk.W)
|
||
self.tree.column("zlzt", width=80, anchor=tk.CENTER)
|
||
self.tree.column("sfxxm", width=80, anchor=tk.CENTER)
|
||
self.tree.column("bianhao", width=80, anchor=tk.CENTER)
|
||
self.tree.column("actions", width=200, anchor=tk.CENTER)
|
||
|
||
# 添加滚动条
|
||
scrollbar_y = ttk.Scrollbar(table_frame, orient=tk.VERTICAL, command=self.tree.yview)
|
||
scrollbar_x = ttk.Scrollbar(table_frame, orient=tk.HORIZONTAL, command=self.tree.xview)
|
||
self.tree.configure(yscrollcommand=scrollbar_y.set, xscrollcommand=scrollbar_x.set)
|
||
|
||
self.tree.grid(row=0, column=0, sticky=(tk.W, tk.E, tk.N, tk.S))
|
||
scrollbar_y.grid(row=0, column=1, sticky=(tk.N, tk.S))
|
||
scrollbar_x.grid(row=1, column=0, sticky=(tk.W, tk.E))
|
||
|
||
# 绑定右键点击事件
|
||
self.tree.bind("<Button-3>", self._on_tree_right_click)
|
||
|
||
# === 分页区域 ===
|
||
page_frame = ttk.Frame(self.tab_manage)
|
||
page_frame.grid(row=3, column=0, sticky=(tk.W, tk.E), pady=(5, 0))
|
||
|
||
self.page_info_var = tk.StringVar(value="第 1 页，共 0 页，共 0 条")
|
||
ttk.Label(page_frame, textvariable=self.page_info_var).pack(side=tk.LEFT)
|
||
|
||
ttk.Button(page_frame, text="上一页", command=self._prev_page).pack(side=tk.LEFT, padx=(20, 5))
|
||
ttk.Button(page_frame, text="下一页", command=self._next_page).pack(side=tk.LEFT, padx=(5, 0))
|
||
|
||
# 加载数据（延迟加载，避免初始化时出错）
|
||
# self._load_projects()
|
||
|
||
def _create_settings_tab(self):
|
||
"""创建参数设置 Tab"""
|
||
# 创建 Canvas 和 Scrollbar 实现滚动
|
||
canvas = tk.Canvas(self.tab_settings)
|
||
scrollbar = ttk.Scrollbar(self.tab_settings, orient="vertical", command=canvas.yview)
|
||
self.settings_frame = ttk.Frame(canvas, padding="10")
|
||
|
||
self.settings_frame.bind(
|
||
"<Configure>",
|
||
lambda e: canvas.configure(scrollregion=canvas.bbox("all"))
|
||
)
|
||
|
||
canvas.create_window((0, 0), window=self.settings_frame, anchor="nw")
|
||
canvas.configure(yscrollcommand=scrollbar.set)
|
||
|
||
canvas.pack(side="left", fill="both", expand=True)
|
||
scrollbar.pack(side="right", fill="y")
|
||
|
||
# 配置网格
|
||
self.settings_frame.columnconfigure(1, weight=1)
|
||
|
||
row = 0
|
||
|
||
# === 网站账号配置 ===
|
||
website_frame = ttk.LabelFrame(self.settings_frame, text="网站账号配置", padding="10")
|
||
website_frame.grid(row=row, column=0, columnspan=2, sticky=(tk.W, tk.E), pady=(0, 15))
|
||
website_frame.columnconfigure(1, weight=1)
|
||
|
||
ttk.Label(website_frame, text="网站用户名:").grid(row=0, column=0, sticky=tk.W)
|
||
self.setting_username_var = tk.StringVar(value=self.config.get("step1.username", "wangpeng"))
|
||
ttk.Entry(website_frame, textvariable=self.setting_username_var, width=40).grid(row=0, column=1, sticky=(tk.W, tk.E), padx=(10, 0))
|
||
|
||
ttk.Label(website_frame, text="网站密码:").grid(row=1, column=0, sticky=tk.W, pady=(10, 0))
|
||
self.setting_password_var = tk.StringVar(value=self.config.get("step1.password", ""))
|
||
ttk.Entry(website_frame, textvariable=self.setting_password_var, show="*", width=40).grid(row=1, column=1, sticky=(tk.W, tk.E), padx=(10, 0), pady=(10, 0))
|
||
|
||
row += 1
|
||
|
||
# === Chrome 配置 ===
|
||
chrome_frame = ttk.LabelFrame(self.settings_frame, text="Chrome 配置", padding="10")
|
||
chrome_frame.grid(row=row, column=0, columnspan=2, sticky=(tk.W, tk.E), pady=(0, 15))
|
||
chrome_frame.columnconfigure(1, weight=1)
|
||
|
||
ttk.Label(chrome_frame, text="Chrome 路径:").grid(row=0, column=0, sticky=tk.W)
|
||
self.setting_chrome_var = tk.StringVar(value=self.config.get("chrome.path", r"C:\Program Files\Google\Chrome\Application\chrome.exe"))
|
||
ttk.Entry(chrome_frame, textvariable=self.setting_chrome_var, width=50).grid(row=0, column=1, sticky=(tk.W, tk.E), padx=(10, 0))
|
||
|
||
row += 1
|
||
|
||
# === 微信公众号配置 ===
|
||
wechat_frame = ttk.LabelFrame(self.settings_frame, text="微信公众号配置", padding="10")
|
||
wechat_frame.grid(row=row, column=0, columnspan=2, sticky=(tk.W, tk.E), pady=(0, 15))
|
||
wechat_frame.columnconfigure(1, weight=1)
|
||
|
||
ttk.Label(wechat_frame, text="AppID:").grid(row=0, column=0, sticky=tk.W)
|
||
self.setting_appid_var = tk.StringVar(value=self.config.get("wechat.appid", ""))
|
||
ttk.Entry(wechat_frame, textvariable=self.setting_appid_var, width=40).grid(row=0, column=1, sticky=(tk.W, tk.E), padx=(10, 0))
|
||
|
||
ttk.Label(wechat_frame, text="AppSecret:").grid(row=1, column=0, sticky=tk.W, pady=(10, 0))
|
||
self.setting_appsecret_var = tk.StringVar(value=self.config.get("wechat.appsecret", ""))
|
||
ttk.Entry(wechat_frame, textvariable=self.setting_appsecret_var, show="*", width=40).grid(row=1, column=1, sticky=(tk.W, tk.E), padx=(10, 0), pady=(10, 0))
|
||
|
||
row += 1
|
||
|
||
# === 数据库配置 ===
|
||
db_frame = ttk.LabelFrame(self.settings_frame, text="数据库配置", padding="10")
|
||
db_frame.grid(row=row, column=0, columnspan=2, sticky=(tk.W, tk.E), pady=(0, 15))
|
||
db_frame.columnconfigure(1, weight=1)
|
||
|
||
ttk.Label(db_frame, text="主机地址:").grid(row=0, column=0, sticky=tk.W)
|
||
self.setting_db_host_var = tk.StringVar(value=self.config.get("database.host", "localhost"))
|
||
ttk.Entry(db_frame, textvariable=self.setting_db_host_var, width=30).grid(row=0, column=1, sticky=(tk.W, tk.E), padx=(10, 0))
|
||
|
||
ttk.Label(db_frame, text="端口:").grid(row=1, column=0, sticky=tk.W, pady=(10, 0))
|
||
self.setting_db_port_var = tk.StringVar(value=str(self.config.get("database.port", "3306")))
|
||
ttk.Entry(db_frame, textvariable=self.setting_db_port_var, width=30).grid(row=1, column=1, sticky=(tk.W, tk.E), padx=(10, 0), pady=(10, 0))
|
||
|
||
ttk.Label(db_frame, text="数据库名:").grid(row=2, column=0, sticky=tk.W, pady=(10, 0))
|
||
self.setting_db_name_var = tk.StringVar(value=self.config.get("database.database", "test"))
|
||
ttk.Entry(db_frame, textvariable=self.setting_db_name_var, width=30).grid(row=2, column=1, sticky=(tk.W, tk.E), padx=(10, 0), pady=(10, 0))
|
||
|
||
ttk.Label(db_frame, text="用户名:").grid(row=3, column=0, sticky=tk.W, pady=(10, 0))
|
||
self.setting_db_user_var = tk.StringVar(value=self.config.get("database.user", "root"))
|
||
ttk.Entry(db_frame, textvariable=self.setting_db_user_var, width=30).grid(row=3, column=1, sticky=(tk.W, tk.E), padx=(10, 0), pady=(10, 0))
|
||
|
||
ttk.Label(db_frame, text="密码:").grid(row=4, column=0, sticky=tk.W, pady=(10, 0))
|
||
self.setting_db_pass_var = tk.StringVar(value=self.config.get("database.password", "123456"))
|
||
ttk.Entry(db_frame, textvariable=self.setting_db_pass_var, show="*", width=30).grid(row=4, column=1, sticky=(tk.W, tk.E), padx=(10, 0), pady=(10, 0))
|
||
|
||
# 测试连接按钮
|
||
ttk.Button(db_frame, text="🧪 测试连接", command=self._test_db_connection).grid(row=5, column=1, sticky=tk.W, pady=(10, 0))
|
||
|
||
row += 1
|
||
|
||
# === 保存按钮 ===
|
||
btn_frame = ttk.Frame(self.settings_frame)
|
||
btn_frame.grid(row=row, column=0, columnspan=2, pady=(20, 0))
|
||
|
||
ttk.Button(btn_frame, text="💾 保存配置", command=self._save_settings, width=20).pack(side=tk.LEFT, padx=(0, 10))
|
||
ttk.Button(btn_frame, text="🔄 重置", command=self._reset_settings, width=15).pack(side=tk.LEFT)
|
||
|
||
# ==================== 发布公众号 Tab 方法 ====================
|
||
|
||
def _open_markdown_editor(self):
|
||
"""打开 Markdown 编辑器"""
|
||
try:
|
||
open_editor(parent=self.root)
|
||
except Exception as e:
|
||
messagebox.showerror("错误", f"打开 Markdown 编辑器失败: {e}")
|
||
|
||
def log(self, message: str):
|
||
"""添加日志"""
|
||
self.log_text.insert(tk.END, f"{message}\n")
|
||
self.log_text.see(tk.END)
|
||
self.root.update_idletasks()
|
||
|
||
def set_running(self, running: bool):
|
||
"""设置运行状态"""
|
||
self.is_running = running
|
||
state = tk.DISABLED if running else tk.NORMAL
|
||
self.publish_btn.config(state=state)
|
||
|
||
if running:
|
||
self.progress.start()
|
||
self.status_var.set("运行中...")
|
||
else:
|
||
self.progress.stop()
|
||
self.status_var.set("就绪")
|
||
|
||
def run_full_flow(self):
|
||
"""运行完整流程"""
|
||
if self.is_running:
|
||
return
|
||
|
||
thread = threading.Thread(target=self._run_full_flow_thread)
|
||
thread.daemon = True
|
||
thread.start()
|
||
|
||
def _run_full_flow_thread(self):
|
||
"""在后台线程中运行完整流程"""
|
||
try:
|
||
self.set_running(True)
|
||
self.root.after(0, lambda: self.log("=" * 50))
|
||
self.root.after(0, lambda: self.log("开始运行完整流程..."))
|
||
self.root.after(0, lambda: self.log("=" * 50))
|
||
|
||
# 获取配置（从配置文件读取）
|
||
chrome_path = self.config.get("chrome.path", "")
|
||
username = self.config.get("step1.username", "")
|
||
password = self.config.get("step1.password", "")
|
||
project_name = self.project_name_var.get()
|
||
|
||
# Step 1
|
||
self.root.after(0, lambda: self.log("\n[Step 1] 开始执行..."))
|
||
|
||
# 定义日志回调函数，将 step1 的日志输出到 GUI
|
||
def step1_log_callback(message: str):
|
||
self.root.after(0, lambda msg=message: self.log(msg))
|
||
|
||
step1 = Step1FeastCoding(chrome_path, username, password, log_callback=step1_log_callback)
|
||
step1.project_name = project_name # 设置项目名称
|
||
step1.run()
|
||
self.root.after(0, lambda: self.log("[Step 1] 执行完成！"))
|
||
|
||
# Step 2
|
||
self.root.after(0, lambda: self.log("\n[Step 2] 开始执行..."))
|
||
|
||
# 定义日志回调函数，将 step2 的日志输出到 GUI
|
||
def log_callback(message: str):
|
||
self.root.after(0, lambda msg=message: self.log(msg))
|
||
|
||
# 获取 CSS 方案 ID（根据名称查找）
|
||
scheme_name = self.css_scheme_var.get()
|
||
css_scheme_id = self._get_scheme_id_by_name(scheme_name)
|
||
|
||
step2 = Step2Converter(log_callback=log_callback, css_scheme_id=css_scheme_id)
|
||
step2.run()
|
||
self.root.after(0, lambda: self.log("[Step 2] 执行完成！"))
|
||
|
||
self.root.after(0, lambda: self.log("\n" + "=" * 50))
|
||
self.root.after(0, lambda: self.log("完整流程执行成功！"))
|
||
self.root.after(0, lambda: self.log("=" * 50))
|
||
self.root.after(0, lambda: messagebox.showinfo("成功", "完整流程执行成功！"))
|
||
|
||
except Exception as e:
|
||
error_msg = str(e)
|
||
self.root.after(0, lambda: self.log(f"\n[ERROR] {error_msg}"))
|
||
self.root.after(0, lambda: messagebox.showerror("错误", f"执行失败: {error_msg}"))
|
||
finally:
|
||
self.set_running(False)
|
||
|
||
def _get_css_scheme_names(self):
|
||
"""获取 CSS 方案名称列表（只显示名称）"""
|
||
schemes = self.theme_manager.get_all_css_schemes()
|
||
return ["默认"] + [s.name for s in schemes]
|
||
|
||
def _get_scheme_id_by_name(self, name: str):
|
||
"""根据名称获取方案 ID"""
|
||
if name == "默认":
|
||
return None
|
||
schemes = self.theme_manager.get_all_css_schemes()
|
||
for scheme in schemes:
|
||
if scheme.name == name:
|
||
return scheme.id
|
||
return None
|
||
|
||
def refresh_css_schemes(self):
|
||
"""刷新 CSS 方案列表"""
|
||
# 重新加载配置
|
||
self.theme_manager = ThemeManager()
|
||
# 更新下拉框
|
||
self.css_scheme_combo['values'] = self._get_css_scheme_names()
|
||
messagebox.showinfo("提示", "CSS 方案列表已刷新！")
|
||
|
||
# ==================== 文章管理 Tab 方法 ====================
|
||
|
||
def _load_projects(self):
|
||
"""加载项目列表"""
|
||
if not self.db_manager:
|
||
messagebox.showwarning("警告", "数据库未配置或连接失败")
|
||
return
|
||
|
||
try:
|
||
projects, total = self.db_manager.get_projects(
|
||
page=self.current_page,
|
||
page_size=self.page_size,
|
||
search_name=self.search_keyword if self.search_keyword else None
|
||
)
|
||
|
||
self.total_items = total
|
||
|
||
# 清空表格
|
||
for item in self.tree.get_children():
|
||
self.tree.delete(item)
|
||
|
||
# 插入数据
|
||
for project in projects:
|
||
self.tree.insert("", tk.END, values=(
|
||
project.name or "",
|
||
project.name2 or "",
|
||
project.paths or "",
|
||
project.zlzt or "",
|
||
project.sfxxm or "",
|
||
project.bianhao or "",
|
||
"右键操作"
|
||
))
|
||
|
||
# 更新分页信息
|
||
total_pages = (total + self.page_size - 1) // self.page_size if total > 0 else 1
|
||
self.page_info_var.set(f"第 {self.current_page} 页，共 {total_pages} 页，共 {total} 条")
|
||
|
||
except Exception as e:
|
||
messagebox.showerror("错误", f"加载数据失败: {e}")
|
||
|
||
def _search_projects(self):
|
||
"""搜索项目"""
|
||
try:
|
||
self.search_keyword = self.search_var.get().strip()
|
||
self.current_page = 1
|
||
self._load_projects()
|
||
except Exception as e:
|
||
messagebox.showerror("错误", f"搜索失败: {e}")
|
||
|
||
def _refresh_projects(self):
|
||
"""刷新项目列表"""
|
||
self.search_keyword = ""
|
||
self.search_var.set("")
|
||
self.current_page = 1
|
||
self._load_projects()
|
||
|
||
def _prev_page(self):
|
||
"""上一页"""
|
||
if self.current_page > 1:
|
||
self.current_page -= 1
|
||
self._load_projects()
|
||
|
||
def _next_page(self):
|
||
"""下一页"""
|
||
total_pages = (self.total_items + self.page_size - 1) // self.page_size if self.total_items > 0 else 1
|
||
if self.current_page < total_pages:
|
||
self.current_page += 1
|
||
self._load_projects()
|
||
|
||
def _on_tree_right_click(self, event):
|
||
"""处理表格右键点击事件"""
|
||
# 获取点击位置对应的项
|
||
item = self.tree.identify_row(event.y)
|
||
if not item:
|
||
return
|
||
|
||
# 选中当前项
|
||
self.tree.selection_set(item)
|
||
|
||
# 获取选中项的值
|
||
values = self.tree.item(item, "values")
|
||
if not values:
|
||
return
|
||
|
||
name = values[0]
|
||
name2 = values[1]
|
||
current_zlzt = values[3] # 整理状态现在在第4列（索引3）
|
||
|
||
# 创建操作菜单
|
||
menu = tk.Menu(self.root, tearoff=0)
|
||
|
||
# 变更已整理状态选项
|
||
new_zlzt = "已整理" if current_zlzt != "已整理" else "未整理"
|
||
menu.add_command(
|
||
label=f"变更状态: {new_zlzt}",
|
||
command=lambda: self._update_project_zlzt(name, new_zlzt)
|
||
)
|
||
|
||
menu.add_separator()
|
||
|
||
# 删除选项
|
||
menu.add_command(
|
||
label="删除",
|
||
command=lambda: self._delete_project(name)
|
||
)
|
||
|
||
menu.add_separator()
|
||
|
||
# 复制路径选项
|
||
menu.add_command(
|
||
label="复制路径",
|
||
command=lambda: self._copy_path(name)
|
||
)
|
||
|
||
menu.add_separator()
|
||
|
||
# 复制name2名称选项
|
||
menu.add_command(
|
||
label="复制名称2",
|
||
command=lambda: self._copy_name2(name2)
|
||
)
|
||
|
||
# 显示菜单
|
||
menu.post(event.x_root, event.y_root)
|
||
|
||
def _update_project_zlzt(self, name: str, zlzt: str):
|
||
"""更新项目整理状态"""
|
||
if not self.db_manager:
|
||
return
|
||
|
||
if messagebox.askyesno("确认", f"确定将 '{name}' 标记为 {zlzt} 吗？"):
|
||
if self.db_manager.update_zlzt(name, zlzt):
|
||
messagebox.showinfo("成功", f"已更新为 {zlzt}")
|
||
self._load_projects()
|
||
else:
|
||
messagebox.showerror("错误", "更新失败")
|
||
|
||
def _delete_project(self, name: str):
|
||
"""删除项目"""
|
||
if not self.db_manager:
|
||
return
|
||
|
||
if messagebox.askyesno("确认", f"确定删除 '{name}' 吗？\n此操作不可恢复！"):
|
||
if self.db_manager.delete_project(name):
|
||
messagebox.showinfo("成功", "删除成功")
|
||
self._load_projects()
|
||
else:
|
||
messagebox.showerror("错误", "删除失败")
|
||
|
||
def _copy_name2(self, name2: str):
|
||
"""复制name2名称到剪贴板"""
|
||
if not name2:
|
||
messagebox.showwarning("提示", "名称2为空，无法复制")
|
||
return
|
||
|
||
self.root.clipboard_clear()
|
||
self.root.clipboard_append(name2)
|
||
messagebox.showinfo("成功", f"已复制: {name2}")
|
||
|
||
def _copy_path(self, name: str):
|
||
"""复制项目路径到剪贴板"""
|
||
if not self.db_manager:
|
||
messagebox.showwarning("提示", "数据库未连接，无法获取路径")
|
||
return
|
||
|
||
try:
|
||
# 获取项目路径
|
||
project = self.db_manager.get_project_by_name(name)
|
||
if project and project.paths:
|
||
self.root.clipboard_clear()
|
||
self.root.clipboard_append(project.paths)
|
||
messagebox.showinfo("成功", f"已复制路径: {project.paths}")
|
||
else:
|
||
messagebox.showwarning("提示", "项目路径为空，无法复制")
|
||
except Exception as e:
|
||
messagebox.showerror("错误", f"复制路径失败: {e}")
|
||
|
||
# ==================== 参数设置 Tab 方法 ====================
|
||
|
||
def _test_db_connection(self):
|
||
"""测试数据库连接"""
|
||
def test_in_thread():
|
||
try:
|
||
from db_manager import DatabaseManager
|
||
|
||
host = self.setting_db_host_var.get()
|
||
port_str = self.setting_db_port_var.get()
|
||
port = int(port_str) if port_str else 3306
|
||
database = self.setting_db_name_var.get()
|
||
user = self.setting_db_user_var.get()
|
||
password = self.setting_db_pass_var.get()
|
||
|
||
print(f"[TestDB] 测试连接: {host}:{port}/{database} as {user}")
|
||
db = DatabaseManager(host, database, user, password, port)
|
||
success, error = db.test_connection()
|
||
|
||
if success:
|
||
self.root.after(0, lambda: messagebox.showinfo("成功", "数据库连接成功！"))
|
||
else:
|
||
self.root.after(0, lambda: messagebox.showerror("失败", f"连接失败: {error}"))
|
||
except Exception as e:
|
||
print(f"[TestDB] 测试连接异常: {e}")
|
||
self.root.after(0, lambda msg=str(e): messagebox.showerror("错误", f"测试连接出错: {msg}"))
|
||
|
||
# 在新线程中运行，避免阻塞 GUI
|
||
thread = threading.Thread(target=test_in_thread)
|
||
thread.daemon = True
|
||
thread.start()
|
||
|
||
def _save_settings(self):
|
||
"""保存配置"""
|
||
try:
|
||
# 构建新配置
|
||
new_config = {
|
||
"step1": {
|
||
"url": "https://feast.yidaima.cn/login",
|
||
"username": self.setting_username_var.get(),
|
||
"password": self.setting_password_var.get(),
|
||
"project_name": self.project_name_var.get(),
|
||
"feast_button": "FeastCoding"
|
||
},
|
||
"step2": {
|
||
"url": "http://yidaima.cn:6005/"
|
||
},
|
||
"chrome": {
|
||
"path": self.setting_chrome_var.get()
|
||
},
|
||
"wechat": {
|
||
"appid": self.setting_appid_var.get(),
|
||
"appsecret": self.setting_appsecret_var.get()
|
||
},
|
||
"database": {
|
||
"host": self.setting_db_host_var.get(),
|
||
"port": int(self.setting_db_port_var.get() or 3306),
|
||
"database": self.setting_db_name_var.get(),
|
||
"user": self.setting_db_user_var.get(),
|
||
"password": self.setting_db_pass_var.get()
|
||
}
|
||
}
|
||
|
||
# 保存到文件
|
||
import yaml
|
||
config_path = os.path.join(os.path.dirname(__file__), "config.yaml")
|
||
with open(config_path, 'w', encoding='utf-8') as f:
|
||
yaml.dump(new_config, f, allow_unicode=True, sort_keys=False)
|
||
|
||
# 重新加载配置
|
||
global _config_instance
|
||
_config_instance = None
|
||
self.config = get_config()
|
||
|
||
# 重置数据库管理器
|
||
reset_db_manager()
|
||
self._init_db_manager()
|
||
|
||
# 更新发布页面的值
|
||
self.username_var.set(self.config.get("step1.username", ""))
|
||
self.password_var.set(self.config.get("step1.password", ""))
|
||
self.chrome_path_var.set(self.config.get("chrome.path", ""))
|
||
|
||
messagebox.showinfo("成功", "配置已保存！")
|
||
|
||
except Exception as e:
|
||
messagebox.showerror("错误", f"保存失败: {e}")
|
||
|
||
def _reset_settings(self):
|
||
"""重置设置"""
|
||
if messagebox.askyesno("确认", "确定要重置所有设置吗？"):
|
||
self.setting_username_var.set(self.config.get("step1.username", "wangpeng"))
|
||
self.setting_password_var.set(self.config.get("step1.password", ""))
|
||
self.setting_chrome_var.set(self.config.get("chrome.path", r"C:\Program Files\Google\Chrome\Application\chrome.exe"))
|
||
self.setting_appid_var.set(self.config.get("wechat.appid", ""))
|
||
self.setting_appsecret_var.set(self.config.get("wechat.appsecret", ""))
|
||
self.setting_db_host_var.set(self.config.get("database.host", "localhost"))
|
||
self.setting_db_port_var.set(str(self.config.get("database.port", "3306")))
|
||
self.setting_db_name_var.set(self.config.get("database.database", "test"))
|
||
self.setting_db_user_var.set(self.config.get("database.user", "root"))
|
||
self.setting_db_pass_var.set(self.config.get("database.password", "123456"))
|
||
|
||
|
||
def main():
|
||
"""启动 GUI"""
|
||
root = tk.Tk()
|
||
app = YidaimaGUI(root)
|
||
root.mainloop()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|