better install procedure

mmalter 2025-02-11 04:43:20 +01:00
parent 88511b62aa
commit 32011e4701
1 changed files with 11 additions and 1 deletions

View File

@ -1,6 +1,7 @@
from selenium.webdriver import firefox
from fake_useragent import UserAgent
from pathlib import Path
import importlib
import os
@ -21,6 +22,15 @@ def tor_browser_binary_path():
return None
def install_har_export_trigger(driver):
with importlib.resources.path(
__package__.assets,
"har_export_trigger-0.6.2resigned1.xpi",
) as p:
driver.install_addon(p.as_uri())
return driver
class Grattoir(object):
def __enter__(self):
self._driver = None
@ -34,7 +44,7 @@ class Grattoir(object):
o.profile = profile
o.binary = binary
self._driver = firefox.webdriver.WebDriver(options=o)
self._driver.install_addon("assets/har_export_trigger-0.6.2resigned1.xpi")
self._driver = install_har_export_trigger(self._driver)
return self._driver
def __exit__(self, exc_type, exc_value, exc_tb):