feat: 还原之前另存的配置文件 (#185)

* feat: 还原之前另存的配置文件

* Apply suggestions from code review

Co-authored-by: 宁静致远 <[email protected]>

---------

Co-authored-by: 宁静致远 <[email protected]>

Author: DuckDuckStudio

Diff for: src/config/备份.py

@@ -0,0 +1,86 @@
+import os
+import sys
+import shutil
+import tkinter as tk
+from tkinter import filedialog
+from tkinter import messagebox
+from colorama import init, Fore
+
+def 选择目标文件夹():
+    while True:
+        root = tk.Tk()
+        root.withdraw() # 隐藏主窗口
+
+        目标文件夹 = filedialog.askdirectory(title="你要将配置文件备份到哪里？")
+
+        if 检查文件夹(目标文件夹):
+            return 目标文件夹
+
+def 检查文件夹(目标文件夹):
+    # 检测是否选择了文件夹
+    if not 目标文件夹:
+        # 当用户点击选择文件窗口的“取消”按钮时
+        messagebox.showerror("错误", "未选择文件夹，操作取消。")
+        sys.exit(3)
+        # return False
+    # 检查目标文件夹是否存在，如果不存在则创建
+    if not os.path.exists(目标文件夹):
+        messagebox.showwarning("警告", "目标文件夹不存在，将新建一个同名文件夹。")
+        os.makedirs(目标文件夹)
+    # 检查目标文件夹是否为空
+    if len(os.listdir(目标文件夹)) != 0:
+        messagebox.showerror("错误", "目标文件夹不为空，请选择一个空文件夹。")
+        return False
+    return True # 一切正常
+
+def 复制配置文件(目标文件夹):
+    # 获取当前运行脚本的文件夹路径的上级目录作为源文件夹
+    当前文件夹 = os.path.dirname(os.path.abspath(sys.argv[0]))
+    源文件夹 = os.path.dirname(当前文件夹)
+
+    已复制的文件 = [] # 用于存储已经复制的文件路径
+
+    try:
+        # 遍历源文件夹中的所有文件和子文件夹
+        for root, dirs, files in os.walk(源文件夹):
+            for file in files:
+                if file == 'config.ini':
+                    源路径 = os.path.join(root, file)
+                    if root == 源文件夹:
+                        # 忽略源文件夹中的 config.ini
+                        continue
+                    目标路径 = os.path.join(目标文件夹, os.path.relpath(源路径, 源文件夹))
+                    目标目录 = os.path.dirname(目标路径)
+
+                    # 创建目标文件夹中的子文件夹（如果需要）
+                    if not os.path.exists(目标目录):
+                        os.makedirs(目标目录)
+                    
+                    # 复制文件
+                    shutil.copyfile(源路径, 目标路径)
+                    已复制的文件.append(目标路径) # 记录已复制的文件
+
+        # 如果成功完成所有文件复制
+        messagebox.showinfo("恭喜", "配置文件备份成功！")
+    except KeyboardInterrupt:
+        # 用户按下 Ctrl+C 取消操作
+        messagebox.showerror("错误", "用户已取消操作。")
+        
+        for file_path in 已复制的文件:
+            os.remove(file_path) # 删除已复制的文件
+    except Exception as e:
+        # 出现异常时取消操作并清空已复制的文件
+        messagebox.showerror("错误", f"备份过程中出现错误:\n{e}")
+
+        for file_path in 已复制的文件:
+            os.remove(file_path) # 删除已复制的文件
+
+
+def main():
+    init(autoreset=True)
+    # 让用户选择目标文件夹
+    目标文件夹 = 选择目标文件夹()
+    复制配置文件(目标文件夹)
+
+if __name__ == "__main__":
+    main()

Diff for: src/config/还原.py

@@ -0,0 +1,88 @@
+import os
+import sys
+import configparser
+import tkinter as tk
+from tkinter import filedialog
+from tkinter import messagebox
+from colorama import init, Fore
+import shutil
+
+def 选择备份文件夹():
+    root = tk.Tk()
+    root.withdraw()
+    备份文件夹 = filedialog.askdirectory(title="你之前将配置文件备份到了哪里？")
+    if not os.path.exists(备份文件夹):
+        messagebox.showerror("错误", "备份文件夹不存在")
+    return 备份文件夹
+
+def 比较配置文件键(文件1, 文件2):
+    config1 = configparser.ConfigParser()
+    config2 = configparser.ConfigParser()
+    
+    try:
+        config1.read(文件1)
+        config2.read(文件2)
+    except Exception as e:
+        return False, f"配置文件解析失败: {str(e)}"
+
+    # 比较所有节
+    sections1 = set(config1.sections())
+    sections2 = set(config2.sections())
+    if sections1 != sections2:
+        return False, f"节不一致: 备份文件节={sections1}，目标文件节={sections2}"
+
+    # 比较每个节内的键
+    for section in sections1:
+        keys1 = set(config1[section].keys())
+        keys2 = set(config2[section].keys())
+        if keys1 != keys2:
+            return False, f"节 [{section}] 中: 备份文件键={keys1}，目标文件键={keys2}"
+
+    return True, "键完全一致"
+
+def 还原配置文件(备份文件夹, 目标根目录):
+    init(autoreset=True)
+    已处理文件 = 0
+    跳过文件 = 0
+
+    for root, dirs, files in os.walk(备份文件夹):
+        for file in files:
+            if file == 'config.ini':
+                备份路径 = os.path.join(root, file)
+                相对路径 = os.path.relpath(备份路径, 备份文件夹)
+                目标路径 = os.path.join(目标根目录, 相对路径)
+
+                # 检查目标文件是否存在
+                if not os.path.exists(目标路径):
+                    messagebox.showwarning("警告", f"您使用的版本中不存在配置文件 {相对路径}，跳过还原。")
+                    跳过文件 += 1
+                    continue
+
+                # 比较键
+                键一致, 信息 = 比较配置文件键(备份路径, 目标路径)
+                if not 键一致:
+                    messagebox.showwarning("警告", f"您使用的版本中的配置文件 {相对路径} 与您备份的配置文件的键不一致，跳过还原。\n{信息}")
+                    跳过文件 += 1
+                    continue
+
+                # 覆盖文件
+                try:
+                    shutil.copyfile(备份路径, 目标路径)
+                    已处理文件 += 1
+                except Exception as e:
+                    messagebox.showerror("错误", f"还原配置文件 {相对路径} 失败:\n{str(e)}")
+                    跳过文件 += 1
+
+    messagebox.showinfo("恭喜", f"成功还原配置文件！\n我们一共成功还原了 {已处理文件} 个配置文件，跳过了 {跳过文件} 个配置文件。")
+
+def main():
+    init(autoreset=True)
+    备份文件夹 = 选择备份文件夹()
+    
+    脚本路径 = os.path.abspath(sys.argv[0])
+    目标根目录 = os.path.dirname(os.path.dirname(脚本路径))
+
+    还原配置文件(备份文件夹, 目标根目录)
+
+if __name__ == "__main__":
+    main()

Diff for: src/temporarily/config-save.py

@@ -219,7 +219,8 @@ categories = {
     },
     "关于芙芙工具箱": {
         "检查版本": ".\\Show_version.exe",
-        "另存配置文件": ".\\temporarily\\config-save.exe",
+        "备份配置": ".\\config\\备份.exe",
+        "还原配置": ".\\config\\还原.exe",
         "安装文件夹": os.getcwd(),
         "程序文档": "https://duckduckstudio.github.io/yazicbs.github.io/Tools/Fufu_Tools/wiki/",
         "程序官网": "https://duckduckstudio.github.io/yazicbs.github.io/Tools/Fufu_Tools/",

Diff for: src/芙芙工具箱(for安装程序).pyw

@@ -230,7 +230,8 @@ categories = {
     "关于芙芙工具箱": {
         "检查版本": ".\\Show_version.exe",
         "修改开始菜单图标": "link",
-        "另存配置文件": ".\\temporarily\\config-save.exe",
+        "备份配置": ".\\config\\备份.exe",
+        "还原配置": ".\\config\\还原.exe",
         "安装文件夹": os.getcwd(),
         "程序文档": "https://duckduckstudio.github.io/yazicbs.github.io/Tools/Fufu_Tools/wiki/",
         "程序官网": "https://duckduckstudio.github.io/yazicbs.github.io/Tools/Fufu_Tools/",

Diff for: src/芙芙工具箱(for打包).pyw

@@ -233,7 +233,8 @@ categories = {
     "关于芙芙工具箱": {
         "检查版本": ".\\Show_version.py",
         "修改开始菜单图标": "link",
-        "另存配置文件": ".\\temporarily\\config-save.py",
+        "备份配置": ".\\config\\备份.py",
+        "还原配置": ".\\config\\还原.py",
         "安装文件夹": os.path.dirname(os.path.abspath(__file__)),
         "程序文档": "https://duckduckstudio.github.io/yazicbs.github.io/Tools/Fufu_Tools/wiki/",
         "程序官网": "https://duckduckstudio.github.io/yazicbs.github.io/Tools/Fufu_Tools/",