INNER CODE UNIT · Python
__save_html_snapshot
handrew/browserpilot · browserpilot/agents/gpt_selenium_agent.py:281
def __save_html_snapshot(self):
"""Helpful for debugging."""
# Check if the folder exists, and if not, create it.
if not os.path.exists(self.debug_html_folder):
os.makedirs(self.debug_html_folder)
# Save an HTML of the entire page.
debug_name = "debug.html"
debug_name = os.path.join(self.debug_html_folder, debug_name)
html = self.driver.page_source
with open(debug_name, "w+") as f:
f.write(html)
# Save a screenshot of the entire page.
screenshot_name = "debug.png"
screenshot_name = os.path.join(self.debug_html_folder, screenshot_name)
self.driver.save_screenshot(screenshot_name)