设置同名处理策略支持简写

This commit is contained in:
g1879 2023-10-16 22:50:12 +08:00
parent 122dd5e77a
commit 30f022fe71
2 changed files with 13 additions and 5 deletions

View File

@ -1,3 +1,3 @@
- 使用上的问题请先查看文档[使用文档](http://g1879.gitee.io/drissionpagedocs)
- 遇到bug请详细描述如何重现并附上代码
- 提问前先给本库打个星,谢谢
1. 使用上的问题请先查看文档[使用文档](http://g1879.gitee.io/drissionpagedocs)
2. 遇到bug请详细描述如何重现并附上代码
3. 提问前先给本库打个星,谢谢

View File

@ -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)