actual fix

pwncollege · Jan 28, 2025 · f508277 · f508277
1 parent 7e40681
commit f508277
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pwnshop/templates/base/selenium.py b/pwnshop/templates/base/selenium.py
@@ -1,5 +1,6 @@
 #!/opt/pwn.college/python
 
+import shutil
 import psutil
 import urllib
 import atexit
@@ -21,7 +22,7 @@
 
 options = FirefoxOptions()
 options.add_argument("--headless")
-service = FirefoxService(log_path="/dev/null", executable_path="/usr/local/bin/geckodriver")
+service = FirefoxService(log_path="/dev/null", executable_path=shutil.which("geckodriver"))
 browser = webdriver.Firefox(service=service, options=options)
 atexit.register(browser.quit)