kyfx
/
SeleniumBase
mirror of https://github.com/seleniumbase/SeleniumBase.git
5
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fixes download_file() method not being able to store filename with question marks in Windows.

This commit is contained in:
Dmitriy Sintsov 2025-04-08 22:06:11 +03:00
parent 990701d9ee
commit 930763c198
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
seleniumbase/fixtures/page_utils.py
View File

@ -2,11 +2,13 @@
import codecs import codecs
import fasteners import fasteners
import os import os
import platform
import re import re
import requests import requests
from selenium.webdriver.common.by import By from selenium.webdriver.common.by import By
from seleniumbase.fixtures import constants from seleniumbase.fixtures import constants
from seleniumbase.fixtures import css_to_xpath from seleniumbase.fixtures import css_to_xpath
from urllib.parse import urlparse
def get_domain_url(url): def get_domain_url(url):
@ -297,6 +299,10 @@ def _print_unique_links_with_status_codes(page_url, soup):
def _download_file_to(file_url, destination_folder, new_file_name=None): def _download_file_to(file_url, destination_folder, new_file_name=None):
if new_file_name: if new_file_name:
file_name = new_file_name file_name = new_file_name
else:
if platform.system() == "Windows":
url_parts = urlparse(file_url)
file_name = url_parts.path.split("/")[-1]
else: else:
file_name = file_url.split("/")[-1] file_name = file_url.split("/")[-1]
r = requests.get(file_url, timeout=5) r = requests.get(file_url, timeout=5)