grattoir/__init__.py

@@ -1,93 +1,40 @@
+from selenium.webdriver import firefox
 from fake_useragent import UserAgent
 from pathlib import Path
-from selenium.webdriver import firefox
-import importlib
 import os
-import tempfile
-import types
-import uuid
 
 
-def unique_name(prefix):
+def tor_profile():
-    return prefix + "-" + str(uuid.uuid4())
+    tor_firefox_path = str(Path("~/.tor project/firefox").expanduser())
+    ini_path = tor_firefox_path + "/profiles.ini"
+    with open(ini_path) as ini:
+        for line in ini:
+            if "Path=" in line:
+                return tor_firefox_path + '/' + line.split('=')[1][:-1]
+
+
+def tor_browser_binary_path():
+    for pe in os.environ["PATH"].split(os.pathsep):
+        checkname = os.path.join(pe, "tor-browser")
+        if os.access(checkname, os.X_OK) and not os.path.isdir(checkname):
+            return checkname
+    return None
 
 
 class Grattoir(object):
-    def __init__(self, name, with_har=False):
+    def __init__(self):
-        self.name = name
+        self._driver = None
-        self.with_har = with_har
-        if self.with_har:
-            self.har_filename = unique_name(self.name)
-            self.har_export_path = tempfile.TemporaryDirectory()
-            self.export_path = (
-                self.har_export_path.name + "/" + self.har_filename
-            )
-
-    def tor_browser_binary_path(self):
-        for pe in os.environ["PATH"].split(os.pathsep):
-            checkname = os.path.join(pe, "tor-browser")
-            if os.access(checkname, os.X_OK) and not os.path.isdir(checkname):
-                return checkname
-        return None
-
-    def install_har_export_trigger(self):
-        with importlib.resources.path(
-            __package__ + ".assets",
-            "har_export_trigger-0.6.2resigned1.xpi",
-        ) as p:
-            self.driver.install_addon(p.as_uri()[7:])
-
-    def tor_profile_path(self):
-        tor_firefox_path = str(Path("~/.tor project/firefox").expanduser())
-        ini_path = tor_firefox_path + "/profiles.ini"
-        with open(ini_path) as ini:
-            for line in ini:
-                if "Path=" in line:
-                    return tor_firefox_path + '/' + line.split('=')[1][:-1]
-
-    def har_enabled_profile(self, profile):
-        prefs = {
-            "extensions.netmonitor.har.enableAutomation": True,
-            "extensions.netmonitor.har.contentAPIToken": "test",
-            "extensions.netmonitor.har.autoConnect": True,
-            "devtools.netmonitor.har.compress": False,
-            "devtools.netmonitor.har.defaultFileName": self.har_filename,
-            "devtools.netmonitor.har.defaultLogDir": self.har_export_path.name,
-            "devtools.netmonitor.har.enableAutoExportToFile": True,
-            "devtools.netmonitor.har.forceExport": True,
-            "devtools.netmonitor.har.includeResponseBodies": True,
-            "devtools.netmonitor.har.jsonp": True,
-            "devtools.netmonitor.har.jsonpCallback": True,
-            "devtools.netmonitor.har.pageLoadedTimeout": "2500",
-            "devtools.toolbox.selectedTool": "netmonitor",
-            "devtools.netmonitor.persistlog": True,
-        }
-        for k, v in prefs.items():
-            profile.set_preference(k, v)
-        return profile
-
-    def firefox_profile(self):
-        profile = firefox.firefox_profile.FirefoxProfile(profile_directory=self.tor_profile_path())
-        if self.with_har:
-            profile = self.har_enabled_profile(profile)
-        return profile
-
-    def init_driver(self):
-        self.driver = None
-        b_path = self.tor_browser_binary_path()
-        binary = firefox.firefox_binary.FirefoxBinary(firefox_path=b_path)
-        profile = self.firefox_profile()
-        options = firefox.options.Options()
-        options.profile = profile
-        options.binary = binary
-        self.driver = firefox.webdriver.WebDriver(options=options)
-        self.install_har_export_trigger()
 
     def __enter__(self):
-        self.init_driver()
+        p_path = tor_profile()
-        return self
+        profile = firefox.firefox_profile.FirefoxProfile(profile_directory=p_path)
+        b_path = tor_browser_binary_path()
+        binary = firefox.firefox_binary.FirefoxBinary(firefox_path=b_path)
+        o = firefox.options.Options()
+        o.profile = profile
+        o.binary = binary
+        self._driver = firefox.webdriver.WebDriver(options=o)
+        return self._driver
 
     def __exit__(self, exc_type, exc_value, exc_tb):
-        self.driver.quit()
+        self._driver.quit()
-        #if self.with_har:
-            #self.har_export_path.cleanup()