From eab6cb8fba96407f2b7d5f91f0f07b5d1e5f318b Mon Sep 17 00:00:00 2001
From: Diaoxiaozhang <zjrdmczh@outlook.com>
Date: Tue, 26 Mar 2024 19:04:52 +0800
Subject: [PATCH] debug

---
 main.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/main.py b/main.py
index cd7bdec..3a64ba3 100644
--- a/main.py
+++ b/main.py
@@ -196,8 +196,14 @@ def select_directory():
         if choice == "1" or choice == "2":
             if choice == "1":
                 files_to_decrypt = [select_file()]
+                if files_to_decrypt == [""]:
+                    print("检测到文件选择窗口被关闭")
+                    continue
             elif choice == "2":
                 dir_to_decrypt = select_directory()
+                if dir_to_decrypt == "":
+                    print("检测到目录选择窗口被关闭")
+                    continue
                 files_to_decrypt = glob.glob(os.path.join(dir_to_decrypt, "*.xm"))
             print("请选择是否需要设置输出路径：（不设置默认为本程序目录下的output文件夹）")
             print("1. 设置输出路径")
@@ -205,6 +211,9 @@ def select_directory():
             choice = input()
             if choice == "1":
                 output_path = select_directory()
+                if output_path == "":
+                    print("检测到目录选择窗口被关闭")
+                    continue
             elif choice == "2":
                 output_path = ""
             for file in files_to_decrypt: