chromedriver之"此类型文件可能会损害您的计算机"提示的处理

背景

最近在使用selenium进行自动文件下载时,突然出现了一个报错:

下载进行不下去了。

思路

经过各种谷歌、百度,均告诉我在要增加params,关闭浏览器安全选项,配置如下:

chromeOptions = webdriver.ChromeOptions()
prefs = {"profile.default_content_settings.popups": 0,
          "download.default_directory": path,
          "download.prompt_for_download": False,
          # "download.directory_upgrade": 'true',
          "safebrowsing.enabled": True}
chromeOptions.add_experimental_option("prefs", prefs)

事实证明,可能以前的版本是可行的,现在的真心不行。

上面配置重点是"safebrowsing.enabled": True。在MacOS的环境下,哪怕不配也是没有问题的,Windows就不行了。

最后在谷歌上找到一篇相关文章,大意是说这个是无解的,可能是windows系统安全的问题,

对于这个解释我还是比较认可的,所以在mac上就不会提示。

Let’s start frankly: you can’t disable this feature. You can merely tweak the download settings in order to avoid it.

https://windowsreport.com/typ...

那么问题来了,既然这样,有什么曲线救国的办法呢?

当chromedriver弹出这个提示的时候,其实文件已经下载完成,如下图:

我们只需要将文件名修改为正确的名字和后缀即可(比如test.txt),直接无视警告提醒。思路如下:

  • 找到最新下载的文件:通过对下载目录的文件按照创建时间排序,找到最新的
  • 判断是否该文件是否已下载完成:通过判断时间间隔前后该文件是否有大小的变化

结论

根据上面思路,实现的关键代码如下:

def sort_file():
    global path
    dir_lists = os.listdir(path)
    dir_lists.sort(key=lambda fn: os.path.getmtime(os.path.join(path, fn)))
    return dir_lists[-1]
  
def changeName(path, oldname, newname):
    old_path = os.path.join(path, oldname)
    new_path = os.path.join(path, newname + '.txt')
    if os.path.exists(old_path):
        if os.path.exists(new_path):
            os.remove(new_path)
        os.rename(old_path, new_path)
        print ('rename done!' + newname)
    else:
        print ('no file found!')
  
def download():
  ...
  temp_filename = sort_file()
      if u'未确认' in temp_filename:
          temp_filesize_old = os.path.getsize(os.path.join(path, temp_filename))
          while True:
              time.sleep(1)
              temp_filesize_new = os.path.getsize(os.path.join(path, temp_filename))
              if temp_filesize_old == temp_filesize_new:
                  changeName(path, temp_filename, ip)
                  return
              else:
                  temp_filesize_old = temp_filesize_new
      else:
          print(u'下载失败')

需要注意的是,在文件重命名的时候,先检查下文件是否已经存在,先删除,在创建。

以上。

如果有更好的思路,欢迎分享。

你可能感兴趣的:(selenium,python)