From d7bfe4ed40200e6d4e3e6406087296cc80a108a9 Mon Sep 17 00:00:00 2001
From: mmalter <dev@michaelmalter.fr>
Date: Wed, 12 Feb 2025 05:09:40 +0100
Subject: [PATCH] add method to retrieve har

---
 grattoir/__init__.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/grattoir/__init__.py b/grattoir/__init__.py
index 8b4bb60..e019c4c 100644
--- a/grattoir/__init__.py
+++ b/grattoir/__init__.py
@@ -3,6 +3,7 @@ from fake_useragent import UserAgent
 from pathlib import Path
 import importlib
 import os
+import types
 
 
 def tor_profile():
@@ -31,6 +32,13 @@ def install_har_export_trigger(driver):
     return driver
 
 
+def har_data(self):
+    return self.execute_async_script(
+    "HAR.triggerExport().then(arguments[0]);"
+)
+
+
+
 class Grattoir(object):
     def __enter__(self):
         self._driver = None
@@ -45,6 +53,7 @@ class Grattoir(object):
         o.binary = binary
         self._driver = firefox.webdriver.WebDriver(options=o)
         self._driver = install_har_export_trigger(self._driver)
+        self._driver.har_data = types.MethodType(har_data, self._driver)
         return self._driver
 
     def __exit__(self, exc_type, exc_value, exc_tb):