Skip to content

Commit

Permalink
Add a separate Browser._get_browser_download_dir
Browse files Browse the repository at this point in the history
This avoids duplicating this branch everywhere, and stop WKTR from
installing into browsers/main/browsers/main.
  • Loading branch information
gsnedders committed May 23, 2023
1 parent 6a978c2 commit 2255458
Showing 1 changed file with 11 additions and 9 deletions.
20 changes: 11 additions & 9 deletions tools/wpt/browser.py
Original file line number Diff line number Diff line change
Expand Up @@ -81,6 +81,12 @@ class Browser:
def __init__(self, logger):
self.logger = logger

def _get_browser_download_dir(self, dest, channel):
if dest is None:
return self._get_browser_binary_dir(dest, channel)

return dest

def _get_browser_binary_dir(self, dest, channel):
if dest is None:
# os.getcwd() doesn't include the venv path
Expand Down Expand Up @@ -108,7 +114,7 @@ def download_from_url(
"""
self.logger.info("Downloading from %s" % url)

dest = self._get_browser_binary_dir(dest, channel)
dest = self._get_browser_download_dir(dest, channel)

resp = get(url)
filename = get_download_filename(resp, default_name)
Expand Down Expand Up @@ -235,8 +241,7 @@ def download(self, dest=None, channel="nightly", rename=None):
}
os_key = (self.platform, uname[4])

if dest is None:
dest = self._get_browser_binary_dir(None, channel)
dest = self._get_browser_download_dir(dest, channel)

if channel not in product:
raise ValueError("Unrecognised release channel: %s" % channel)
Expand Down Expand Up @@ -848,8 +853,7 @@ def _get_webdriver_url(self, version, revision=None):
return self._build_snapshots_url(revision, filename)

def download(self, dest=None, channel=None, rename=None, version=None, revision=None):
if dest is None:
dest = self._get_browser_binary_dir(None, channel)
dest = self._get_browser_download_dir(dest, channel)

filename = f"{self._chromium_package_name}.zip"

Expand Down Expand Up @@ -1739,8 +1743,7 @@ def download(self, dest=None, channel="preview", rename=None, system_version=Non
if channel != "preview":
raise ValueError(f"can only install 'preview', not '{channel}'")

if dest is None:
dest = self._get_browser_binary_dir(None, channel)
dest = self._get_browser_download_dir(dest, channel)

stp_downloads = self._find_downloads()

Expand Down Expand Up @@ -2057,8 +2060,7 @@ def download(self, dest=None, channel=None, rename=None):
bundle_filename = response.text.strip()
bundle_url = base_download_dir + bundle_filename

if dest is None:
dest = self._get_browser_binary_dir(None, channel)
dest = self._get_browser_download_dir(dest, channel)
bundle_file_path = os.path.join(dest, bundle_filename)

self.logger.info("Downloading WebKitGTK MiniBrowser bundle from %s" % bundle_url)
Expand Down

0 comments on commit 2255458

Please sign in to comment.