chromedriver之"此型別檔案可能會損害您的計算機"提示的處理

摘要： 背景 最近在使用selenium進行自動檔案下載時，突然出現了一個報錯： 下載進行不下去了。 思路 經過各種谷歌、百度，均告訴我在要增加params，關閉瀏覽器安全選項，配置如下： chromeOptions = webdriver.ChromeOptions...

背景

最近在使用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上就不會提示。

那麼問題來了，既然這樣，有什麼曲線救國的辦法呢？

當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'下載失敗')

需要注意的是，在檔案重新命名的時候，先檢查下檔案是否已經存在，先刪除，在建立。

以上。

如果有更好的思路，歡迎分享。