From e4c5a6ad0c903f64a6b52d46a84c98a76640f0b8 Mon Sep 17 00:00:00 2001
From: g1879 <g1879@qq.com>
Date: Mon, 3 Aug 2020 23:40:47 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=8A=A5=E9=94=99=E6=8F=90?=
 =?UTF-8?q?=E7=A4=BA=EF=BC=9Bdownload()=E5=87=BD=E6=95=B0=E5=87=BA?=
 =?UTF-8?q?=E9=94=99=E6=97=B6=E5=8F=AF=E6=98=BE=E7=A4=BA=E8=AF=A6=E7=BB=86?=
 =?UTF-8?q?=E4=BF=A1=E6=81=AF=EF=BC=9B=5Fmake=5Fresponse()=E6=94=B9?=
 =?UTF-8?q?=E4=B8=BA=E8=BF=94=E5=9B=9Etuple?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 DrissionPage/session_page.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/DrissionPage/session_page.py b/DrissionPage/session_page.py
index 9a1fe9f..8b3b11f 100644
--- a/DrissionPage/session_page.py
+++ b/DrissionPage/session_page.py
@@ -152,7 +152,9 @@ class SessionPage(object):
         else:
             file_name = f'untitled_{time()}_{randint(0, 100)}'
 
+        file_name = re.sub(r'[\\/*:|<>?"]', '', file_name).strip()
         if rename:  # 重命名文件，不改变扩展名
+            rename = re.sub(r'[\\/*:|<>?"]', '', rename).strip()
             ext_name = file_name.split('.')[-1]
             if rename.lower().endswith(f'.{ext_name}'.lower()) or ext_name == file_name:
                 full_name = rename
@@ -161,7 +163,6 @@ class SessionPage(object):
         else:
             full_name = file_name
 
-        full_name = re.sub(r'[\\/*:|<>?"]', '', full_name).strip()
         goal_Path = Path(goal_path)
         goal_path = ''
         for key, i in enumerate(goal_Path.parts):  # 去除路径中的非法字符