From 30f022fe71907f83bca79f33dab18b4a32bcfcc2 Mon Sep 17 00:00:00 2001
From: g1879 <g1879@qq.com>
Date: Mon, 16 Oct 2023 22:50:12 +0800
Subject: [PATCH] =?UTF-8?q?=E8=AE=BE=E7=BD=AE=E5=90=8C=E5=90=8D=E5=A4=84?=
=?UTF-8?q?=E7=90=86=E7=AD=96=E7=95=A5=E6=94=AF=E6=8C=81=E7=AE=80=E5=86=99?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
---
.gitee/ISSUE_TEMPLATE.zh-CN.md | 6 +++---
DrissionPage/setter.py | 12 ++++++++++--
2 files changed, 13 insertions(+), 5 deletions(-)
diff --git a/.gitee/ISSUE_TEMPLATE.zh-CN.md b/.gitee/ISSUE_TEMPLATE.zh-CN.md
index 1c9e53c..fc92e66 100644
--- a/.gitee/ISSUE_TEMPLATE.zh-CN.md
+++ b/.gitee/ISSUE_TEMPLATE.zh-CN.md
@@ -1,3 +1,3 @@
-- 使用上的问题请先查看文档[使用文档](http://g1879.gitee.io/drissionpagedocs)
-- 遇到bug请详细描述如何重现，并附上代码
-- 提问前先给本库打个星，谢谢
\ No newline at end of file
+1. 使用上的问题请先查看文档[使用文档](http://g1879.gitee.io/drissionpagedocs)
+2. 遇到bug请详细描述如何重现，并附上代码
+3. 提问前先给本库打个星，谢谢
\ No newline at end of file
diff --git a/DrissionPage/setter.py b/DrissionPage/setter.py
index 2298124..5677012 100644
--- a/DrissionPage/setter.py
+++ b/DrissionPage/setter.py
@@ -140,8 +140,16 @@ class TabSetter(ChromiumBaseSetter):
self._page.browser._dl_mgr.set_rename(self._page.tab_id, name)
def when_download_file_exists(self, mode):
- if mode not in ('rename', 'overwrite', 'skip'):
- raise ValueError(f"mode参数只能是'rename', 'overwrite', 'skip' 之一，现在是：{mode}")
+ """设置当存在同名文件时的处理方式
+ :param mode: 可在 'rename', 'overwrite', 'skip', 'r', 'o', 's'中选择
+ :return: None
+ """
+ types = {'rename': 'rename', 'overwrite': 'overwrite', 'skip': 'skip', 'r': 'rename', 'o': 'overwrite',
+ 's': 'skip'}
+ mode = types.get(mode, None)
+ if not mode:
+ raise ValueError(f"mode参数只能是'rename', 'overwrite', 'skip', 'r', 'o', 's' 之一，现在是：{mode}")
+
self._page.browser._dl_mgr.set_file_exists(self._page.tab_id, mode)